apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-04 21:32:38 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge pull request #1544 from EvensXia/fix_metagpt_from_evensxia

Browse source 
fix ollama_api to add llava support
This commit is contained in:
Alexander Wu 2024-10-30 10:14:53 +08:00 • committed by GitHub
commit 9db0874102
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
6 changed files with 345 additions and 130 deletions
Download patch file Download diff file Expand all files Collapse all files

4
examples/llm_vision.py
View file

@ -15,8 +15,8 @@ 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")
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()
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__":