Update api.py

api.py

@@ -1,3 +1,5 @@
+# api.py
+import time
 from fastapi import FastAPI, Query, HTTPException
 from pydantic import BaseModel
 from typing import List, Optional, Any
@@ -60,6 +62,7 @@ class AskResponse(BaseModel):
     rewrite: str
     retrieved: List[RetrievedChunk]
     trace_id: str
+    latency_sec: float   # ⬅️ latency included in response
 
 
 class SwitchThreadRequest(BaseModel):
@@ -103,6 +106,9 @@ def api_ask(
     # combine body + query flag (OR)
     search_flag = bool(payload.search_outside_thread or search_outside_thread)
 
+    # ---- measure latency for core RAG pipeline ----
+    t0 = time.perf_counter()
+
     # rewrite using thread + entity memory
     rewrite = rewrite_query(payload.text, session)
 
@@ -117,6 +123,8 @@ def api_ask(
     # build answer
     answer, citations = build_answer(payload.text, rewrite, retrieved)
 
+    elapsed = time.perf_counter() - t0  # seconds
+
     # log and get trace_id
     trace_id = log_trace(payload.session_id, payload.text, rewrite, retrieved, answer, citations)
 
@@ -150,6 +158,7 @@ def api_ask(
         rewrite=rewrite,
         retrieved=retrieved_out,
         trace_id=trace_id,
+        latency_sec=elapsed,
     )