mirror of
https://github.com/FoundationAgents/MetaGPT.git
synced 2026-07-05 16:02:14 +02:00
fix embed
This commit is contained in:
parent
f2aa4e3f9d
commit
062b13fc6c
2 changed files with 4 additions and 3 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue