apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-08 15:05:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fixup ollama_api chat prompt

Browse source
This commit is contained in:
EvensXia 2024-10-29 23:09:38 +08:00
parent 685d301a66
commit 187e512547
1 changed files with 5 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

16
metagpt/provider/ollama_api.py
View file

@ -63,15 +63,9 @@ class OllamaMessageMeta(type):
for base in bases:
if issubclass(base, OllamaMessageBase):
api_type = attrs["api_type"]
if isinstance(api_type, list):
for tpe in api_type:
assert tpe not in OllamaMessageMeta.registed_message, "api_type already exist"
assert isinstance(tpe, OllamaMessageAPI), "api_type not support"
OllamaMessageMeta.registed_message[tpe] = cls
else:
assert api_type not in OllamaMessageMeta.registed_message, "api_type already exist"
assert isinstance(api_type, OllamaMessageAPI), "api_type not support"
OllamaMessageMeta.registed_message[api_type] = cls
assert api_type not in OllamaMessageMeta.registed_message, "api_type already exist"
assert isinstance(api_type, OllamaMessageAPI), "api_type not support"
OllamaMessageMeta.registed_message[api_type] = cls
@classmethod
def get_message(cls, input_type: OllamaMessageAPI) -> type[OllamaMessageBase]:
@ -101,9 +95,9 @@ class OllamaMessageChat(OllamaMessageBase, metaclass=OllamaMessageMeta):
messes = []
for prompt in prompts:
if len(images) > 0:
messes.append({"role": "user", "content": "\n".join(prompts), "images": images})
messes.append({"role": "user", "content": prompt, "images": images})
else:
messes.append({"role": "user", "content": "\n".join(prompts)})
messes.append({"role": "user", "content": prompt})
sends = {"model": self.model, "messages": messes}
sends.update(self.additional_kwargs)
return sends