fix: Fix return type

julep-ai · Sep 13, 2024 · f8ce313 · f8ce313
1 parent 14a1a01
commit f8ce313
Showing 1 changed file with 3 additions and 8 deletions.
diff --git a/agents-api/agents_api/routers/sessions/create_or_update_session.py b/agents-api/agents_api/routers/sessions/create_or_update_session.py
@@ -7,7 +7,7 @@
 
 from ...autogen.openapi_model import (
     CreateOrUpdateSessionRequest,
-    ResourceCreatedResponse,
+    ResourceUpdatedResponse,
 )
 from ...dependencies.developer_id import get_developer_id
 from ...models.session.create_or_update_session import (
@@ -21,14 +21,9 @@ async def create_or_update_session(
     x_developer_id: Annotated[UUID4, Depends(get_developer_id)],
     session_id: UUID,
     data: CreateOrUpdateSessionRequest,
-) -> ResourceCreatedResponse:
-    session = create_session_query(
+) -> ResourceUpdatedResponse:
+    return create_session_query(
         developer_id=x_developer_id,
         session_id=session_id,
         data=data,
     )
-
-    return ResourceCreatedResponse(
-        id=session.id,
-        created_at=session.created_at,
-    )