Patched /tmp/tmpjf3dgkvy/examples/llm_vision.py

femto · Oct 9, 2024 · c8b7193 · c8b7193
1 parent 6f1cacc
commit c8b7193
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/llm_vision.py b/examples/llm_vision.py
@@ -16,7 +16,8 @@ async def main():
     invoice_path = Path(__file__).parent.joinpath("..", "tests", "data", "invoices", "invoice-2.png")
     img_base64 = encode_image(invoice_path)
     res = await llm.aask(msg="if this is a invoice, just return True else return False", images=[img_base64])
-    assert "true" in res.lower()
+    if "true" not in res.lower():
+        raise ValueError("Assertion failed: Expected response to contain 'true'.")
 
 
 if __name__ == "__main__":