revert llm_vision

This commit is contained in:
EvensXia 2024-10-30 09:56:35 +08:00
parent 2bca5c9d06
commit 82dba4b9bc

View file

@ -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__":