apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-04-25 00:36:55 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

format

Browse source
This commit is contained in:
EvensXia 2024-10-29 16:19:41 +08:00
parent 322974a917
commit 7d3d15a324
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
metagpt/provider/ollama_api.py
View file

@ -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))