fix embed

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

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