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

fix: keep same return value in get_choice_function_arguments.

Browse source
This commit is contained in:
刘棒棒 2024-01-19 15:37:28 +08:00
parent c7d46d0b0f
commit c6695a3023
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
metagpt/provider/openai_api.py
View file

@ -217,7 +217,7 @@ class OpenAILLM(BaseLLM):
if code_value is None:
raise ValueError(f"Parse code error for {arguments}")
# arguments只有code的情况
return {"language": language_value, "code": code_value}
return {"language": language_value or "python", "code": code_value}
@handle_exception
def get_choice_function_arguments(self, rsp: ChatCompletion) -> dict:
@ -241,7 +241,7 @@ class OpenAILLM(BaseLLM):
f"Got JSONDecodeError for {message.tool_calls[0].function.arguments},\
we will use RegExp to parse code, \n {e}"
)
return {"language": "python", "code": self._parse_arguments(message.tool_calls[0].function.arguments)}
return self._parse_arguments(message.tool_calls[0].function.arguments)
elif message.tool_calls is None and message.content is not None:
# reponse is message
return {"language": "markdown", "code": self.get_choice_text(rsp)}