revert llm_vision

geekan · Oct 30, 2024 · 82dba4b · 82dba4b
1 parent 2bca5c9
commit 82dba4b
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/examples/llm_vision.py b/examples/llm_vision.py
@@ -14,11 +14,9 @@ async def main():
 
     # check if the configured llm supports llm-vision capacity. If not, it will throw a error
     invoice_path = Path(__file__).parent.joinpath("..", "tests", "data", "invoices", "invoice-2.png")
-    encode_image(invoice_path)
-    # res = await llm.aask(msg="return `True` if this image might be a invoice, or return `False`", images=[img_base64])
-    res = await llm.aask(msg="hello")
-    print(res)
-    # assert ("true" in res.lower()) or ("invoice" in res.lower())
+    img_base64 = encode_image(invoice_path)
+    res = await llm.aask(msg="return `True` if this image might be a invoice, or return `False`", images=[img_base64])
+    assert ("true" in res.lower()) or ("invoice" in res.lower())
 
 
 if __name__ == "__main__":