From c48d16d47504a3a100aa1850a8878843c77043af Mon Sep 17 00:00:00 2001 From: EvensXia Date: Tue, 29 Oct 2024 16:37:23 +0800 Subject: [PATCH] fix embedding --- metagpt/provider/ollama_api.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/metagpt/provider/ollama_api.py b/metagpt/provider/ollama_api.py index fdc6b41fb..178003742 100644 --- a/metagpt/provider/ollama_api.py +++ b/metagpt/provider/ollama_api.py @@ -48,7 +48,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: @@ -291,3 +291,6 @@ class OllamaEmbeddings(OllamaLLM): async def _achat_completion_stream(self, messages: list[dict], timeout: int = USE_CONFIG_TIMEOUT) -> str: return await self._achat_completion(messages, timeout=self.get_timeout(timeout)) + + def get_choice_text(self, rsp): + return rsp