From a1da3e6dce19dc50c5434f99e8c9c39aadfb5174 Mon Sep 17 00:00:00 2001 From: Dmitry Paramonov Date: Thu, 1 Aug 2024 17:06:59 +0300 Subject: [PATCH] chore: Disable all the routes except of tasks and agents --- .../agents_api/routers/agents/update_agent.py | 20 ++++++++++--------- agents-api/agents_api/web.py | 6 +++--- 2 files changed, 14 insertions(+), 12 deletions(-) diff --git a/agents-api/agents_api/routers/agents/update_agent.py b/agents-api/agents_api/routers/agents/update_agent.py index 8ac484fd2..e37296ef3 100644 --- a/agents-api/agents_api/routers/agents/update_agent.py +++ b/agents-api/agents_api/routers/agents/update_agent.py @@ -23,15 +23,17 @@ async def update_agent( x_developer_id: Annotated[UUID4, Depends(get_developer_id)], ) -> ResourceUpdatedResponse: try: - updated_agent = update_agent_query( - agent_id=agent_id, - developer_id=x_developer_id, - name=request.name, - about=request.about, - model=request.model, - default_settings=request.default_settings, - metadata=request.metadata, - instructions=request.instructions, + _, updated_agent = next( + update_agent_query( + agent_id=agent_id, + developer_id=x_developer_id, + name=request.name, + about=request.about, + model=request.model, + default_settings=request.default_settings, + metadata=request.metadata, + instructions=request.instructions, + ).iterrows() ) return ResourceUpdatedResponse(**updated_agent) except AgentNotFoundError as e: diff --git a/agents-api/agents_api/web.py b/agents-api/agents_api/web.py index 155ced099..c408c152b 100644 --- a/agents-api/agents_api/web.py +++ b/agents-api/agents_api/web.py @@ -90,9 +90,9 @@ def register_exceptions(app: FastAPI): register_exceptions(app) app.include_router(agents.router) -app.include_router(sessions.router) -app.include_router(users.router) -app.include_router(jobs.router) +# app.include_router(sessions.router) +# app.include_router(users.router) +# app.include_router(jobs.router) app.include_router(tasks.router)