fix: Fix return types

julep-ai · Dec 20, 2024 · 32dbbba · 32dbbba
1 parent 59b24ac
commit 32dbbba
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/agents-api/agents_api/queries/tools/delete_tool.py b/agents-api/agents_api/queries/tools/delete_tool.py
@@ -50,7 +50,7 @@ def delete_tool(
     developer_id: UUID,
     agent_id: UUID,
     tool_id: UUID,
-) -> tuple[list[str], dict]:
+) -> tuple[list[str], list]:
     developer_id = str(developer_id)
     agent_id = str(agent_id)
     tool_id = str(tool_id)

diff --git a/agents-api/agents_api/queries/tools/get_tool.py b/agents-api/agents_api/queries/tools/get_tool.py
@@ -50,7 +50,7 @@ def get_tool(
     developer_id: UUID,
     agent_id: UUID,
     tool_id: UUID,
-) -> tuple[list[str], dict]:
+) -> tuple[list[str], list]:
     developer_id = str(developer_id)
     agent_id = str(agent_id)
     tool_id = str(tool_id)

diff --git a/agents-api/agents_api/queries/tools/list_tools.py b/agents-api/agents_api/queries/tools/list_tools.py
@@ -28,7 +28,7 @@
 """)
 
 if not sql_query.is_valid():
-    raise InvalidSQLQuery("get_tool")
+    raise InvalidSQLQuery("list_tools")
 
 
 # @rewrap_exceptions(
@@ -59,7 +59,7 @@ def list_tools(
     offset: int = 0,
     sort_by: Literal["created_at", "updated_at"] = "created_at",
     direction: Literal["asc", "desc"] = "desc",
-) -> tuple[list[str], dict]:
+) -> tuple[list[str], list]:
     developer_id = str(developer_id)
     agent_id = str(agent_id)