apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-02 14:45:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix embed

Browse source
This commit is contained in:
EvensXia 2024-10-30 09:50:21 +08:00
parent f2aa4e3f9d
commit 062b13fc6c
2 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

3
examples/llm_vision.py
View file

@ -16,7 +16,8 @@ async def main():
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])
await llm.aask(msg="hello")
res = await llm.aask(msg="hello")
print(res)
# assert ("true" in res.lower()) or ("invoice" in res.lower())

4
metagpt/provider/ollama_api.py
View file

@ -49,7 +49,7 @@ class OllamaMessageBase:
if tpe == "text":
return msg["text"], None
elif tpe == "image_url":
return None, msg["image_url"]["url"][self._image_b64_rms :]
return None, msg["image_url"]["url"][self._image_b64_rms:]
else:
raise ValueError
else:
@ -317,7 +317,7 @@ class OllamaEmbeddings(OllamaLLM):
@register_provider(LLMType.OLLAMA_EMBED)
class OllamaEmbed(OllamaLLM):
class OllamaEmbed(OllamaEmbeddings):
@property
def _llama_api_inuse(self) -> OllamaMessageAPI:
return OllamaMessageAPI.EMBED