diff --git a/metagpt/provider/ollama_api.py b/metagpt/provider/ollama_api.py index 3f8467843..fdc6b41fb 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: @@ -211,7 +211,8 @@ class OllamaLLM(BaseLLM): else: raise ValueError - def get_choice_text(self, rsp): return self.ollama_message.get_choice(rsp) + def get_choice_text(self, rsp): + return self.ollama_message.get_choice(rsp) async def acompletion(self, messages: list[dict], timeout=USE_CONFIG_TIMEOUT) -> dict: return await self._achat_completion(messages, timeout=self.get_timeout(timeout))