Skip to content

Commit

Permalink
fix tests
Browse files Browse the repository at this point in the history
  • Loading branch information
@Benjoyo
Benjoyo committed Feb 20, 2024
1 parent 7fb73c9 commit 732df13
Show file tree
Hide file tree
Showing 17 changed files with 16 additions and 15 deletions.
Empty file removed bpm-ai-connectors-c8/tests/bpmn/__init__.py
View file
Empty file.
2 changes: 1 addition & 1 deletion bpm-ai-connectors-c8/tests/bpmn/test_compose.bpmn
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
<zeebe:ioMapping>
<zeebe:input source="gpt-3.5-turbo" target="llm" />
<zeebe:input source="none" target="asr" />
<zeebe:input source="tesseract" target="ocr" />
<zeebe:input source="none" target="ocr" />
<zeebe:input source="={ &#10; &#34;customerName&#34;: customer_name,&#10; &#34;companyName&#34;: company_name&#10;}" target="input_json" />
<zeebe:input source="letter" target="properties.type" />
<zeebe:input source="formal" target="properties.style" />
Expand Down
2 changes: 1 addition & 1 deletion bpm-ai-connectors-c8/tests/bpmn/test_decide_bool.bpmn
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
<zeebe:ioMapping>
<zeebe:input source="gpt-3.5-turbo" target="llm" />
<zeebe:input source="none" target="asr" />
<zeebe:input source="tesseract" target="ocr" />
<zeebe:input source="none" target="ocr" />
<zeebe:input source="={ &#10; &#34;text&#34;: text&#10;}" target="input_json" />
<zeebe:input source="direct" target="strategy" />
<zeebe:input source="=instruction" target="question" />
Expand Down
2 changes: 1 addition & 1 deletion bpm-ai-connectors-c8/tests/bpmn/test_decide_classifier_bool.bpmn
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
<zeebe:ioMapping>
<zeebe:input source="none" target="llm" />
<zeebe:input source="none" target="asr" />
<zeebe:input source="tesseract" target="ocr" />
<zeebe:input source="none" target="ocr" />
<zeebe:input source="MoritzLaurer/deberta-v3-xsmall-zeroshot-v1.1-all-33" target="classifier" />
<zeebe:input source="={ &#10; &#34;text&#34;: text&#10;}" target="input_json" />
<zeebe:input source="=instruction" target="question" />
Expand Down
2 changes: 1 addition & 1 deletion bpm-ai-connectors-c8/tests/bpmn/test_decide_classifier_string.bpmn
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
<zeebe:ioMapping>
<zeebe:input source="none" target="llm" />
<zeebe:input source="none" target="asr" />
<zeebe:input source="tesseract" target="ocr" />
<zeebe:input source="none" target="ocr" />
<zeebe:input source="MoritzLaurer/deberta-v3-base-zeroshot-v1.1-all-33" target="classifier" />
<zeebe:input source="={ &#10; &#34;text&#34;: text&#10;}" target="input_json" />
<zeebe:input source="=instruction" target="question" />
Expand Down
2 changes: 1 addition & 1 deletion bpm-ai-connectors-c8/tests/bpmn/test_decide_string.bpmn
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
<zeebe:ioMapping>
<zeebe:input source="gpt-3.5-turbo" target="llm" />
<zeebe:input source="none" target="asr" />
<zeebe:input source="tesseract" target="ocr" />
<zeebe:input source="none" target="ocr" />
<zeebe:input source="={ &#10; &#34;text&#34;: text&#10;}" target="input_json" />
<zeebe:input source="direct" target="strategy" />
<zeebe:input source="=instruction" target="question" />
Expand Down
2 changes: 1 addition & 1 deletion bpm-ai-connectors-c8/tests/bpmn/test_extract_faster_whisper.bpmn
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
<zeebe:ioMapping>
<zeebe:input source="gpt-3.5-turbo" target="llm" />
<zeebe:input source="faster_whisper-small" target="asr" />
<zeebe:input source="tesseract" target="ocr" />
<zeebe:input source="none" target="ocr" />
<zeebe:input source="={ &#10; &#34;quote&#34;: &#34;https://upload.wikimedia.org/wikipedia/commons/d/dd/Armstrong_Small_Step.ogg&#34;&#10;}" target="input_json" />
<zeebe:input source="=schema" target="output_schema" />
<zeebe:input source="SINGLE" target="mode" />
Expand Down
2 changes: 1 addition & 1 deletion bpm-ai-connectors-c8/tests/bpmn/test_extract_multiple.bpmn
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
<zeebe:ioMapping>
<zeebe:input source="gpt-3.5-turbo" target="llm" />
<zeebe:input source="none" target="asr" />
<zeebe:input source="tesseract" target="ocr" />
<zeebe:input source="none" target="ocr" />
<zeebe:input source="={ &#10; &#34;friends1&#34;: friends1,&#10; &#34;friends2&#34;: friends2&#10;}" target="input_json" />
<zeebe:input source="=schema" target="output_schema" />
<zeebe:input source="MULTIPLE" target="mode" />
Expand Down
2 changes: 1 addition & 1 deletion bpm-ai-connectors-c8/tests/bpmn/test_extract_qa_multiple.bpmn
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
<zeebe:ioMapping>
<zeebe:input source="none" target="llm" />
<zeebe:input source="none" target="asr" />
<zeebe:input source="tesseract" target="ocr" />
<zeebe:input source="none" target="ocr" />
<zeebe:input source="deepset/deberta-v3-base-squad2" target="qa" />
<zeebe:input source="={ &#10; &#34;text&#34;: text&#10;}" target="input_json" />
<zeebe:input source="=schema" target="output_schema" />
Expand Down
2 changes: 1 addition & 1 deletion bpm-ai-connectors-c8/tests/bpmn/test_extract_qa_single.bpmn
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
<zeebe:ioMapping>
<zeebe:input source="none" target="llm" />
<zeebe:input source="none" target="asr" />
<zeebe:input source="tesseract" target="ocr" />
<zeebe:input source="none" target="ocr" />
<zeebe:input source="deepset/deberta-v3-base-squad2" target="qa" />
<zeebe:input source="={ &#10; &#34;myVariable&#34;: text1,&#10; &#34;text&#34;: text2&#10;}" target="input_json" />
<zeebe:input source="=schema" target="output_schema" />
Expand Down
2 changes: 1 addition & 1 deletion bpm-ai-connectors-c8/tests/bpmn/test_extract_single.bpmn
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
<zeebe:ioMapping>
<zeebe:input source="gpt-3.5-turbo" target="llm" />
<zeebe:input source="none" target="asr" />
<zeebe:input source="tesseract" target="ocr" />
<zeebe:input source="none" target="ocr" />
<zeebe:input source="={ &#10; &#34;myVariable&#34;: text1,&#10; &#34;text&#34;: text2&#10;}" target="input_json" />
<zeebe:input source="=schema" target="output_schema" />
<zeebe:input source="SINGLE" target="mode" />
Expand Down
2 changes: 1 addition & 1 deletion bpm-ai-connectors-c8/tests/bpmn/test_extract_whisper.bpmn
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
<zeebe:ioMapping>
<zeebe:input source="gpt-3.5-turbo" target="llm" />
<zeebe:input source="openai-whisper-1" target="asr" />
<zeebe:input source="tesseract" target="ocr" />
<zeebe:input source="none" target="ocr" />
<zeebe:input source="={ &#10; &#34;quote&#34;: &#34;https://upload.wikimedia.org/wikipedia/commons/d/dd/Armstrong_Small_Step.ogg&#34;&#10;}" target="input_json" />
<zeebe:input source="=schema" target="output_schema" />
<zeebe:input source="SINGLE" target="mode" />
Expand Down
2 changes: 1 addition & 1 deletion bpm-ai-connectors-c8/tests/bpmn/test_generic.bpmn
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
<zeebe:ioMapping>
<zeebe:input source="gpt-3.5-turbo" target="llm" />
<zeebe:input source="none" target="asr" />
<zeebe:input source="tesseract" target="ocr" />
<zeebe:input source="none" target="ocr" />
<zeebe:input source="={ &#10; &#34;number&#34;: x&#10;}" target="input_json" />
<zeebe:input source="=task" target="task_description" />
<zeebe:input source="={&#10; result: { type: &#34;integer&#34; }&#10;}" target="output_schema" />
Expand Down
2 changes: 1 addition & 1 deletion bpm-ai-connectors-c8/tests/bpmn/test_translate.bpmn
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
<zeebe:ioMapping>
<zeebe:input source="gpt-3.5-turbo" target="llm" />
<zeebe:input source="none" target="asr" />
<zeebe:input source="tesseract" target="ocr" />
<zeebe:input source="none" target="ocr" />
<zeebe:input source="={ &#10; &#34;text1&#34;: text1,&#10; &#34;text2&#34;: text2&#10;}" target="input_json" />
<zeebe:input source="=target_lang" target="language" />
</zeebe:ioMapping>
Expand Down
2 changes: 1 addition & 1 deletion bpm-ai-connectors-c8/tests/bpmn/test_translate_nmt.bpmn
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
<zeebe:ioMapping>
<zeebe:input source="none" target="llm" />
<zeebe:input source="none" target="asr" />
<zeebe:input source="tesseract" target="ocr" />
<zeebe:input source="none" target="ocr" />
<zeebe:input source="Helsinki-NLP/opus-mt-{}-{}" target="nmt" />
<zeebe:input source="={ &#10; &#34;text1&#34;: text1,&#10; &#34;text2&#34;: text2&#10;}" target="input_json" />
<zeebe:input source="=target_lang" target="language" />
Expand Down
0 ...i-connectors-c8/tests/sample-invoice.webp → ...nectors-c8/tests/data/sample-invoice.webp
View file
File renamed without changes.
3 changes: 2 additions & 1 deletion bpm-ai-connectors-c8/tests/test_extract_ocr.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -8,10 +8,11 @@
logger = logging.getLogger(__name__)


@requires_inference()
def test_extract_ocr(runtime_selector, zeebe_test_client: ZeebeTestClient):
# given
variables = {
"invoice": "tests/sample-invoice.webp",
"invoice": "https://github.com/holunda-io/camunda-8-connector-gpt/raw/bpm-ai/bpm-ai-connectors-c8/tests/data/sample-invoice.webp",
"schema": {
"invoice_number": "What is the invoice number?",
"total": {
Expand Down

0 comments on commit 732df13

Please sign in to comment.