♻️ split sample code test scripts

Author: ianardee

.github/workflows/_test-code-samples.yml

@@ -4,6 +4,15 @@ on:
   workflow_call:
   workflow_dispatch:
 
+env:
+  MINDEE_API_KEY: ${{ secrets.MINDEE_API_KEY_SE_TESTS }}
+  MINDEE_V2_API_KEY: ${{ secrets.MINDEE_V2_SE_TESTS_API_KEY }}
+  MINDEE_V2_SE_TESTS_FINDOC_MODEL_ID: ${{ secrets.MINDEE_V2_SE_TESTS_FINDOC_MODEL_ID }}
+  MINDEE_V2_SE_TESTS_CLASSIFICATION_MODEL_ID: ${{ secrets.MINDEE_V2_SE_TESTS_CLASSIFICATION_MODEL_ID }}
+  MINDEE_V2_SE_TESTS_CROP_MODEL_ID: ${{ secrets.MINDEE_V2_SE_TESTS_CROP_MODEL_ID }}
+  MINDEE_V2_SE_TESTS_SPLIT_MODEL_ID: ${{ secrets.MINDEE_V2_SE_TESTS_SPLIT_MODEL_ID }}
+  MINDEE_V2_SE_TESTS_OCR_MODEL_ID: ${{ secrets.MINDEE_V2_SE_TESTS_OCR_MODEL_ID }}
+
 jobs:
   test:
     name: Run Code Samples
@@ -38,9 +47,13 @@ jobs:
           python -m pip install pip
           pip install -e .
 
-      - name: Tests code samples
+      - name: Tests v2 sample code
+        run: |
+          ./tests/test_v2_code_samples.sh
+
+      - name: Tests v1 sample code
         run: |
-            ./tests/test_code_samples.sh ${{ secrets.MINDEE_ACCOUNT_SE_TESTS }} ${{ secrets.MINDEE_ENDPOINT_SE_TESTS }} ${{ secrets.MINDEE_API_KEY_SE_TESTS }}  ${{ secrets.MINDEE_V2_SE_TESTS_API_KEY }}  ${{ secrets.MINDEE_V2_SE_TESTS_FINDOC_MODEL_ID }} ${{ secrets.MINDEE_V2_SE_TESTS_CLASSIFICATION_MODEL_ID }} ${{ secrets.MINDEE_V2_SE_TESTS_CROP_MODEL_ID }} ${{ secrets.MINDEE_V2_SE_TESTS_OCR_MODEL_ID }} ${{ secrets.MINDEE_V2_SE_TESTS_SPLIT_MODEL_ID }}
+          ./tests/test_v1_code_samples.sh ${{ secrets.MINDEE_ACCOUNT_SE_TESTS }} ${{ secrets.MINDEE_ENDPOINT_SE_TESTS }}
 
       - name: Notify Slack Action on Failure
         uses: ravsamhq/notify-slack-action@2.3.0

docs/extras/code_samples/v2_classification.txt

@@ -7,7 +7,7 @@ from mindee import (
 
 input_path = "/path/to/the/file.ext"
 api_key = "MY_API_KEY"
-model_id = "MY_CLASSIFICATION_MODEL_ID"
+model_id = "MY_MODEL_ID"
 
 # Init a new client
 mindee_client = ClientV2(api_key)

docs/extras/code_samples/v2_crop.txt

@@ -7,7 +7,7 @@ from mindee import (
 
 input_path = "/path/to/the/file.ext"
 api_key = "MY_API_KEY"
-model_id = "MY_CROP_MODEL_ID"
+model_id = "MY_MODEL_ID"
 
 # Init a new client
 mindee_client = ClientV2(api_key)

docs/extras/code_samples/v2_extraction.txt

@@ -42,3 +42,6 @@ response = mindee_client.enqueue_and_get_result(
 
 # Print a brief summary of the parsed data
 print(response.inference)
+
+# Access the result fields
+fields: dict = response.inference.result.fields

docs/extras/code_samples/v2_ocr.txt

@@ -7,7 +7,7 @@ from mindee import (
 
 input_path = "/path/to/the/file.ext"
 api_key = "MY_API_KEY"
-model_id = "MY_OCR_MODEL_ID"
+model_id = "MY_MODEL_ID"
 
 # Init a new client
 mindee_client = ClientV2(api_key)
@@ -31,5 +31,5 @@ response = mindee_client.enqueue_and_get_result(
 # Print a brief summary of the parsed data
 print(response.inference)
 
-# Access the ocr result
+# Access the OCR result
 pages: list = response.inference.result.pages

docs/extras/code_samples/v2_split.txt

@@ -7,7 +7,7 @@ from mindee import (
 
 input_path = "/path/to/the/file.ext"
 api_key = "MY_API_KEY"
-model_id = "MY_SPLIT_MODEL_ID"
+model_id = "MY_MODEL_ID"
 
 # Init a new client
 mindee_client = ClientV2(api_key)

tests/test_code_samples.sh tests/test_v1_code_samples.shtests/test_code_samples.sh renamed to tests/test_v1_code_samples.sh

@@ -1,18 +1,13 @@
 #! /bin/sh
 set -e
 
-OUTPUT_FILE='./_test.py'
+OUTPUT_FILE='./_test_v1.py'
 ACCOUNT=$1
 ENDPOINT=$2
-API_KEY=$3
-API_KEY_V2=$4
-MODEL_ID=$5
-CLASSIFICATION_MODEL_ID=$6
-CROP_MODEL_ID=$7
-OCR_MODEL_ID=$8
-SPLIT_MODEL_ID=$9
 
-for f in $(find ./docs/extras/code_samples -maxdepth 1 -name "*.txt" -not -name "workflow_*.txt" | sort -h)
+for f in $(
+  find ./docs/extras/code_samples -maxdepth 1 -name "*.txt" -not -name "workflow_*.txt" -not -name "v2_*.txt" | sort -h
+)
 do
   if echo "${f}" | grep -q "default_v2.txt"; then
     if [ -z "${API_KEY_V2}" ] || [ -z "${MODEL_ID}" ]; then
@@ -28,7 +23,7 @@ do
   echo "###############################################"
   echo
 
-  sed "s/my-api-key/${API_KEY}/" "${f}" > $OUTPUT_FILE
+  sed "s/my-api-key/${MINDEE_API_KEY}/" "${f}" > $OUTPUT_FILE
   sed -i 's/\/path\/to\/the\/file.ext/.\/tests\/data\/file_types\/pdf\/blank_1.pdf/' $OUTPUT_FILE
 
 

tests/test_v2_code_samples.sh

@@ -0,0 +1,46 @@
+#! /bin/sh
+set -e
+
+OUTPUT_FILE='./_test_v2.py'
+
+for f in $(
+  find ./docs/extras/code_samples -maxdepth 1 -name "v2_*.txt" | sort -h
+)
+do
+  echo
+  echo "###############################################"
+  echo "${f}"
+  echo "###############################################"
+  echo
+
+  sed "s/MY_API_KEY/${MINDEE_V2_API_KEY}/" "${f}" > $OUTPUT_FILE
+  sed -i 's/\/path\/to\/the\/file.ext/.\/tests\/data\/file_types\/pdf\/blank_1.pdf/' $OUTPUT_FILE
+
+  if echo "${f}" | grep -q "v2_extraction.txt"
+  then
+    sed -i "s/MY_MODEL_ID/${MINDEE_V2_SE_TESTS_FINDOC_MODEL_ID}/" $OUTPUT_FILE
+  fi
+
+  if echo "${f}" | grep -q "v2_classification.txt"
+  then
+    sed -i "s/MY_MODEL_ID/${MINDEE_V2_SE_TESTS_CLASSIFICATION_MODEL_ID}/" $OUTPUT_FILE
+  fi
+
+  if echo "${f}" | grep -q "v2_crop.txt"
+  then
+    sed -i "s/MY_MODEL_ID/${MINDEE_V2_SE_TESTS_CROP_MODEL_ID}/" $OUTPUT_FILE
+  fi
+
+  if echo "${f}" | grep -q "v2_split.txt"
+  then
+    sed -i "s/MY_MODEL_ID/${MINDEE_V2_SE_TESTS_SPLIT_MODEL_ID}/" $OUTPUT_FILE
+  fi
+
+  if echo "${f}" | grep -q "v2_ocr.txt"
+  then
+    sed -i "s/MY_MODEL_ID/${MINDEE_V2_SE_TESTS_OCR_MODEL_ID}/" $OUTPUT_FILE
+  fi
+
+  sleep 0.5  # avoid too many request errors
+  python $OUTPUT_FILE
+done