fix: Fix imports

agents-api/agents_api/routers/agents/create_agent.py

@@ -7,7 +7,7 @@
 
 from ...autogen.openapi_model import CreateAgentRequest, ResourceCreatedResponse
 from ...dependencies.developer_id import get_developer_id
-from ...models.agent.create_agent import create_agent_query
+from ...models.agent.create_agent import create_agent as create_agent_query
 from .router import router
 
 

agents-api/agents_api/routers/agents/delete_agent.py

@@ -8,7 +8,7 @@
 from ...common.exceptions.agents import AgentNotFoundError
 from ...common.utils.datetime import utcnow
 from ...dependencies.developer_id import get_developer_id
-from ...models.agent.delete_agent import delete_agent_query
+from ...models.agent.delete_agent import delete_agent as delete_agent_query
 from .router import router
 
 

agents-api/agents_api/routers/agents/get_agent_details.py

@@ -7,7 +7,7 @@
 from ...autogen.openapi_model import Agent
 from ...common.exceptions.agents import AgentNotFoundError
 from ...dependencies.developer_id import get_developer_id
-from ...models.agent.get_agent import get_agent_query
+from ...models.agent.get_agent import get_agent as get_agent_query
 from .router import router
 
 

agents-api/agents_api/routers/agents/patch_agent.py

@@ -7,7 +7,7 @@
 from ...autogen.openapi_model import PatchAgentRequest, ResourceUpdatedResponse
 from ...common.exceptions.agents import AgentNotFoundError
 from ...dependencies.developer_id import get_developer_id
-from ...models.agent.patch_agent import patch_agent_query
+from ...models.agent.patch_agent import patch_agent as patch_agent_query
 from .router import router
 
 

agents-api/agents_api/routers/agents/update_agent.py

@@ -7,7 +7,7 @@
 from ...autogen.openapi_model import ResourceUpdatedResponse, UpdateAgentRequest
 from ...common.exceptions.agents import AgentNotFoundError
 from ...dependencies.developer_id import get_developer_id
-from ...models.agent.update_agent import update_agent_query
+from ...models.agent.update_agent import update_agent as update_agent_query
 from .router import router
 
 

agents-api/agents_api/routers/sessions/create_session.py

@@ -11,7 +11,7 @@
     ResourceCreatedResponse,
 )
 from ...dependencies.developer_id import get_developer_id
-from ...models.session.create_session import create_session_query
+from ...models.session.create_session import create_session as create_session_query
 from .router import router
 
 

agents-api/agents_api/routers/sessions/delete_session.py

@@ -8,7 +8,7 @@
 from ...common.exceptions.sessions import SessionNotFoundError
 from ...common.utils.datetime import utcnow
 from ...dependencies.developer_id import get_developer_id
-from ...models.session.delete_session import delete_session_query
+from ...models.session.delete_session import delete_session as delete_session_query
 from .router import router
 
 

agents-api/agents_api/routers/sessions/get_session.py

@@ -5,7 +5,7 @@
 
 from ...autogen.openapi_model import Session
 from ...dependencies.developer_id import get_developer_id
-from ...models.session.get_session import get_session_query
+from ...models.session.get_session import get_session as get_session_query
 from .router import router
 
 

agents-api/agents_api/routers/sessions/patch_session.py

@@ -10,7 +10,7 @@
 )
 from ...common.exceptions.sessions import SessionNotFoundError
 from ...dependencies.developer_id import get_developer_id
-from ...models.session.patch_session import patch_session_query
+from ...models.session.patch_session import patch_session as patch_session_query
 from .router import router
 
 

agents-api/agents_api/routers/sessions/update_session.py

@@ -10,7 +10,7 @@
 )
 from ...common.exceptions.sessions import SessionNotFoundError
 from ...dependencies.developer_id import get_developer_id
-from ...models.session.update_session import update_session_query
+from ...models.session.update_session import update_session as update_session_query
 from .router import router
 
 

agents-api/agents_api/routers/tasks/routers.py

@@ -24,8 +24,8 @@
 from agents_api.clients.temporal import run_task_execution_workflow
 from agents_api.common.protocol.tasks import ExecutionInput
 from agents_api.dependencies.developer_id import get_developer_id
-from agents_api.models.execution.create_execution import create_execution_query
-from agents_api.models.execution.get_execution import get_execution_query
+from agents_api.models.execution.create_execution import create_execution as create_execution_query
+from agents_api.models.execution.get_execution import get_execution as get_execution_query
 from agents_api.models.execution.get_execution_transition import (
     get_execution_transition_query,
 )
@@ -41,9 +41,9 @@
 from agents_api.models.execution.update_execution_transition import (
     update_execution_transition_query,
 )
-from agents_api.models.task.create_task import create_task_query
-from agents_api.models.task.get_task import get_task_query
-from agents_api.models.task.list_tasks import list_tasks_query
+from agents_api.models.task.create_task import create_task as create_task_query
+from agents_api.models.task.get_task import get_task as get_task_query
+from agents_api.models.task.list_tasks import list_tasks as list_tasks_query
 
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.DEBUG)

agents-api/agents_api/routers/users/create_user.py

@@ -7,7 +7,7 @@
 
 from ...autogen.openapi_model import CreateUserRequest, ResourceCreatedResponse
 from ...dependencies.developer_id import get_developer_id
-from ...models.user.create_user import create_user_query
+from ...models.user.create_user import create_user as create_user_query
 from .router import router
 
 

agents-api/agents_api/routers/users/get_user_details.py

@@ -7,7 +7,7 @@
 from ...autogen.openapi_model import User
 from ...common.exceptions.users import UserNotFoundError
 from ...dependencies.developer_id import get_developer_id
-from ...models.user.get_user import get_user_query
+from ...models.user.get_user import get_user as get_user_query
 from .router import router
 
 

agents-api/agents_api/routers/users/list_users.py

@@ -7,7 +7,7 @@
 
 from ...autogen.openapi_model import User
 from ...dependencies.developer_id import get_developer_id
-from ...models.user.list_users import list_users_query
+from ...models.user.list_users import list_users as list_users_query
 from .router import router
 
 

agents-api/agents_api/routers/users/patch_user.py

@@ -7,7 +7,7 @@
 from ...autogen.openapi_model import PatchUserRequest, ResourceUpdatedResponse
 from ...common.exceptions.users import UserNotFoundError
 from ...dependencies.developer_id import get_developer_id
-from ...models.user.patch_user import patch_user_query
+from ...models.user.patch_user import patch_user as patch_user_query
 from .router import router
 
 

agents-api/agents_api/routers/users/update_user.py

@@ -7,7 +7,7 @@
 from ...autogen.openapi_model import ResourceUpdatedResponse, UpdateUserRequest
 from ...common.exceptions.users import UserNotFoundError
 from ...dependencies.developer_id import get_developer_id
-from ...models.user.update_user import update_user_query
+from ...models.user.update_user import update_user as update_user_query
 from .router import router