fix(agents-api): Disable tool retrieval (for now)

Signed-off-by: Diwank Singh Tomer <[email protected]>
julep-ai · Apr 15, 2024 · be5373c · be5373c
1 parent f2d865a
commit be5373c
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 8 deletions.
diff --git a/agents-api/agents_api/models/agent/create_tools.py b/agents-api/agents_api/models/agent/create_tools.py
@@ -31,7 +31,7 @@ def create_tools_query(
     functions_input: list[list] = []
 
     for function, embedding in zip(functions, embeddings):
-        parameters = function.parameters.model_dump_json()
+        parameters = function.parameters.model_dump()
         functions_input.append(
             [
                 str(agent_id),

diff --git a/agents-api/agents_api/models/entry/proc_mem_context.py b/agents-api/agents_api/models/entry/proc_mem_context.py
@@ -133,17 +133,24 @@ def proc_mem_context_query(
         # Search for tools
         ?[role, name, content, token_count, created_at, index] :=
             *_input{{agent_id, tool_query}},
-            ~agent_functions:embedding_space {{
+            # ~agent_functions:embedding_space {{
+            #     agent_id,
+            #     name: fn_name,
+            #     description,
+            #     parameters,
+            #     updated_at: created_at |
+            #     query: tool_query,
+            #     k: $k_tools,
+            #     ef: 128,
+            #     radius: $tools_radius,
+            #     bind_distance: distance,
+            # }},
+            *agent_functions {{
                 agent_id,
                 name: fn_name,
                 description,
                 parameters,
-                updated_at: created_at |
-                query: tool_query,
-                k: $k_tools,
-                ef: 128,
-                radius: $tools_radius,
-                bind_distance: distance,
+                updated_at: created_at,
             }},
 
             role = "system",