fix(agents-api): Fix default_settings being null

Signed-off-by: Diwank Singh Tomer <[email protected]>
julep-ai · Sep 17, 2024 · d200a35 · d200a35
1 parent b5f51a6
commit d200a35
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 8 deletions.
diff --git a/agents-api/agents_api/models/agent/list_agents.py b/agents-api/agents_api/models/agent/list_agents.py
@@ -73,6 +73,7 @@ def list_agents(
             created_at,
             updated_at,
             metadata,
+            default_settings,
             instructions,
         ] := input[developer_id],
             *agents {{
@@ -86,6 +87,27 @@ def list_agents(
                 metadata,
                 instructions,
             }},
+            *agent_default_settings {{
+                agent_id: id,
+                frequency_penalty,
+                presence_penalty,
+                length_penalty,
+                repetition_penalty,
+                top_p,
+                temperature,
+                min_p,
+                preset,
+            }},
+            default_settings = {{
+                "frequency_penalty": frequency_penalty,
+                "presence_penalty": presence_penalty,
+                "length_penalty": length_penalty,
+                "repetition_penalty": repetition_penalty,
+                "top_p": top_p,
+                "temperature": temperature,
+                "min_p": min_p,
+                "preset": preset,
+            }},
             {metadata_filter_str}
         
         :limit $limit

diff --git a/agents-api/tests/test_execution_workflow.py b/agents-api/tests/test_execution_workflow.py
@@ -497,14 +497,8 @@ async def _(
             activity for activity in activities_scheduled if activity
         ]
 
-        try:
-            future = handle.result()
-            breakpoint()
-            await future
-        except BaseException as exc:
-            print("exc", exc)
-            breakpoint()
-            raise
+        future = handle.result()
+        await future
 
         assert "wait_for_input_step" in activities_scheduled