From 73140529c7be0fc147e322688543bbb4d2a339c6 Mon Sep 17 00:00:00 2001 From: fiatrete Date: Fri, 16 Jun 2023 15:58:57 +0800 Subject: [PATCH] try to fix arguments return from GPT --- agent_jarvis/jarvis/ai_agent/gpt_agent.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/agent_jarvis/jarvis/ai_agent/gpt_agent.py b/agent_jarvis/jarvis/ai_agent/gpt_agent.py index dbd9b84..95245c6 100644 --- a/agent_jarvis/jarvis/ai_agent/gpt_agent.py +++ b/agent_jarvis/jarvis/ai_agent/gpt_agent.py @@ -55,10 +55,10 @@ class GptAgent(BaseAgent): "speak": assistant_reply, } elif reply_type == "function_call": - # TODO: Check arguments + arguments = fix_json_using_multiple_techniques(assistant_reply["arguments"]) return { "function": assistant_reply["name"], - "arguments": json.loads(assistant_reply["arguments"]) + "arguments": arguments } async def feed_prompt(self, prompt): @@ -86,7 +86,6 @@ class GptAgent(BaseAgent): function_name: str = reply.get("function") if function_name is None: await self._caller_context.reply_text(reply["speak"]) - pass else: arguments: Dict = reply["arguments"]