chore: Disable all the routes except of tasks and agents

julep-ai · Aug 1, 2024 · a1da3e6 · a1da3e6
1 parent 9683d78
commit a1da3e6
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 12 deletions.
diff --git 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
@@ -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)