fix(agents-api): Minor fixes

Signed-off-by: Diwank Singh Tomer <[email protected]>

agents-api/agents_api/activities/utils.py

@@ -304,6 +304,7 @@ def get_handler(system: SystemDef) -> Callable:
     from ..models.docs.delete_doc import delete_doc as delete_doc_query
     from ..models.docs.list_docs import list_docs as list_docs_query
     from ..models.session.create_session import create_session as create_session_query
+    from ..models.session.delete_session import delete_session as delete_session_query
     from ..models.session.get_session import get_session as get_session_query
     from ..models.session.list_sessions import list_sessions as list_sessions_query
     from ..models.session.update_session import update_session as update_session_query

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

@@ -8,10 +8,8 @@
 
 from ...autogen.openapi_model import (
     CreateSessionRequest,
-    ResourceCreatedResponse,
     Session,
 )
-from ...common.utils.datetime import utcnow
 from ...metrics.counters import increase_counter
 from ..utils import partialclass, pg_query, rewrap_exceptions, wrap_in_class
 

agents-api/tests/fixtures.py

@@ -1,6 +1,5 @@
 import random
 import string
-import time
 from uuid import UUID
 
 from fastapi.testclient import TestClient

agents-api/tests/test_session_queries.py

@@ -10,7 +10,6 @@
     CreateOrUpdateSessionRequest,
     CreateSessionRequest,
     PatchSessionRequest,
-    ResourceCreatedResponse,
     ResourceDeletedResponse,
     ResourceUpdatedResponse,
     Session,