From 7d3d15a324905ff0688bad8ec5384f41df54cb6e Mon Sep 17 00:00:00 2001 From: EvensXia Date: Tue, 29 Oct 2024 16:19:41 +0800 Subject: [PATCH] format --- metagpt/provider/ollama_api.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) 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))