fix: Fix profile generator in-prompt example for functions compatibil…

…ity mode - Updated the in-prompt example in the profile generator to be compatible with functions compatibility mode. - Modified the example call section to correctly reflect the structure of function calls.
Significant-Gravitas · Nov 7, 2023 · cd55da5 · cd55da5
1 parent 1b4e921
commit cd55da5
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 8 deletions.
diff --git a/autogpts/autogpt/autogpt/agent_factory/profile_generator.py b/autogpts/autogpt/autogpt/agent_factory/profile_generator.py
@@ -36,9 +36,10 @@ class AgentProfileGeneratorConfiguration(SystemConfiguration):
             "\n"
             "Example Input:\n"
             '"""Help me with marketing my business"""\n\n'
-            "Example Function Call:\n"
+            "Example Call:\n"
             "```\n"
-            "{"
+            "["  # tool_calls
+            '{"type": "function", "function": {'
             '"name": "create_agent",'
             ' "arguments": {'
             '"name": "CMOGPT",'
@@ -65,7 +66,9 @@ class AgentProfileGeneratorConfiguration(SystemConfiguration):
             "]"  # constraints
             "}"  # directives
             "}"  # arguments
-            "}\n"
+            "}"  # function
+            "}"  # tool call
+            "]\n"  # tool_calls
             "```"
         )
     )

diff --git a/autogpts/autogpt/autogpt/agents/prompt_strategies/one_shot.py b/autogpts/autogpt/autogpt/agents/prompt_strategies/one_shot.py
@@ -427,7 +427,9 @@ def extract_command(
             raise InvalidAgentResponseError("No 'tool_calls' in assistant reply")
         assistant_reply_json["command"] = {
             "name": assistant_reply["tool_calls"][0]["function"]["name"],
-            "args": json.loads(assistant_reply["tool_calls"][0]["function"]["arguments"]),
+            "args": json.loads(
+                assistant_reply["tool_calls"][0]["function"]["arguments"]
+            ),
         }
     try:
         if not isinstance(assistant_reply_json, dict):

diff --git a/autogpts/autogpt/autogpt/core/resource/model_providers/openai.py b/autogpts/autogpt/autogpt/core/resource/model_providers/openai.py
@@ -368,8 +368,7 @@ def _get_completion_kwargs(
         if functions:
             if OPEN_AI_CHAT_MODELS[model_name].has_function_call_api:
                 completion_kwargs["tools"] = [
-                    {"type": "function", "function": f.schema}
-                    for f in functions
+                    {"type": "function", "function": f.schema} for f in functions
                 ]
                 if len(functions) == 1:
                     # force the model to call the only specified function

diff --git a/autogpts/autogpt/autogpt/core/resource/model_providers/schema.py b/autogpts/autogpt/autogpt/core/resource/model_providers/schema.py
@@ -78,13 +78,13 @@ class AssistantFunctionCallDict(TypedDict):
 
 
 class AssistantToolCall(BaseModel):
-    id: str
+    # id: str
     type: Literal["function"]
     function: AssistantFunctionCall
 
 
 class AssistantToolCallDict(TypedDict):
-    id: str
+    # id: str
     type: Literal["function"]
     function: AssistantFunctionCallDict