hotfix(agents-api): Session APIs throw Internal Server Error

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

agents-api/agents_api/activities/utils.py

@@ -305,7 +305,6 @@ def get_handler(system: SystemDef) -> Callable:
     from ..models.docs.list_docs import list_docs as list_docs_query
     from ..models.entry.get_history import get_history as get_history_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/metrics/counters.py

@@ -34,17 +34,17 @@
     INF,
 )
 counter = Counter(
-    f"db_query_counter",
-    f"Number of db calls",
+    "db_query_counter",
+    "Number of db calls",
     labelnames=labelnames,
 )
 summary = Summary(
-    f"db_query_latency_summary",
+    "db_query_latency_summary",
     "Database query latency summary",
     labelnames=labelnames,
 )
 hist = Histogram(
-    f"db_query_latency_hist",
+    "db_query_latency_hist",
     "Database query latency histogram",
     labelnames=labelnames,
     buckets=buckets,

agents-api/agents_api/models/session/list_sessions.py

@@ -29,7 +29,13 @@
         TypeError: partialclass(HTTPException, status_code=400),
     }
 )
-@wrap_in_class(make_session)
+@wrap_in_class(
+    make_session,
+    transform=lambda d: {
+        "updated_at": d.pop("updated_at") / (1000000.0),
+        **d,
+    },
+)
 @cozo_query
 @beartype
 def list_sessions(

agents-api/agents_api/models/session/patch_session.py

@@ -46,7 +46,7 @@
     one=True,
     transform=lambda d: {
         "id": d["session_id"],
-        "updated_at": d.pop("updated_at")[0],
+        "updated_at": d.pop("updated_at")[0] / (1000000.0),
         "jobs": [],
         **d,
     },

agents-api/agents_api/models/utils.py

@@ -478,7 +478,12 @@ def _return_data(df: pd.DataFrame):
         transform = transform or (lambda x: x)
 
         if one:
-            assert len(data) >= 1, "Expected one result, got none"
+            resource_name = cls.__name__ if isinstance(cls, type) else "Resource"
+            if len(data) < 1:
+                raise HTTPException(
+                    status_code=404, detail=f"{resource_name} not found"
+                )
+
             obj: ModelT = cls(**transform(data[0]))
             return obj
 

agents-api/agents_api/routers/healthz/check_health.py

@@ -1,7 +1,6 @@
 import logging
 from uuid import UUID
 
-from ...autogen.openapi_model import Agent, ListResponse
 from ...models.agent.list_agents import list_agents as list_agents_query
 from .router import router
 
@@ -10,7 +9,7 @@
 async def check_health() -> dict:
     try:
         # Check if the database is reachable
-        agents = list_agents_query(
+        list_agents_query(
             developer_id=UUID("00000000-0000-0000-0000-000000000000"),
         )
     except Exception as e: