From 32dbbbaac376757ddc535d40eef64d3d64259c3f Mon Sep 17 00:00:00 2001 From: Dmitry Paramonov Date: Fri, 20 Dec 2024 15:21:21 +0300 Subject: [PATCH] fix: Fix return types --- agents-api/agents_api/queries/tools/delete_tool.py | 2 +- agents-api/agents_api/queries/tools/get_tool.py | 2 +- agents-api/agents_api/queries/tools/list_tools.py | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/agents-api/agents_api/queries/tools/delete_tool.py b/agents-api/agents_api/queries/tools/delete_tool.py index 59f561cf1..17535e1e4 100644 --- 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 index 3662725b8..af63be0c9 100644 --- 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 index 59fb1eff5..3dac84875 100644 --- 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)