Enforced chat_name along with chat_id (#1)

- Enforced chat_name along with chat_id (c2201953887a18526296e481ed7e1016349d1f23)


Co-authored-by: Swastik <stellar413@users.noreply.huggingface.co>

api_routes_v2.py

@@ -58,6 +58,7 @@ class ChatResponse(BaseModel):
     api_response: Dict[str, Any]
     intent: Dict[str, Any]
     chat_id: str
+    chat_name: Optional[str] = None
     state: str
     # REMOVED: history field (use separate endpoint to get messages)
     file: Optional[bool] = None 
@@ -649,6 +650,7 @@ def _assistant_response_payload(
     
     # Get file metadata from session
     session = session_manager.get_session(chat_id) or {}
+    chat_name = session.get("chat_name")
     file_metadata = session.get("file_metadata", {})
     
     return ChatResponse(
@@ -661,6 +663,7 @@ def _assistant_response_payload(
         api_response=api_data,
         intent=intent,
         chat_id=chat_id,
+        chat_name=chat_name,
         state=state,
         file=file_metadata.get("has_file", False),
         fileName=file_metadata.get("file_name"),
@@ -1689,7 +1692,11 @@ async def chat_unified_stream(
 
     def emit(obj: Dict[str, Any]) -> bytes:
         obj.setdefault("chat_id", chat_id)
-        obj.setdefault("state", session_manager.get_session(chat_id).get("state", "initial"))
+        # CHANGE: Fetch latest chat_name from session_manager to ensure it's up-to-date
+        # (e.g. if generated during the stream)
+        current_session = session_manager.get_session(chat_id) or {}
+        obj.setdefault("chat_name", current_session.get("chat_name"))
+        obj.setdefault("state", current_session.get("state", "initial"))
         return (json.dumps(obj, ensure_ascii=False) + "\n").encode("utf-8")
 
     def stream_gen() -> Generator[bytes, None, None]:
@@ -2035,7 +2042,10 @@ async def chat_unified_stream(
 @router.post("/chats")
 def create_chat():
     chat_id = session_manager.create_session()
-    return {"chat_id": chat_id}
+    session = session_manager.get_session(chat_id)
+    return {"chat_id": chat_id,
+            "chat_name": session.get("chat_name")
+            }
 
 # ========================
 # FILE UPLOAD (to S3, no presigned URLs) — still available
@@ -2061,6 +2071,7 @@ async def upload_file_to_chat(chat_id: str, file: UploadFile = File(...)):
             "presigned_expires_at": meta.get("presigned_expires_at")
         },
         "chat_id": chat_id,
+        "chat_name": (session_manager.get_session(chat_id) or {}).get("chat_name"),
         "next_action": "💬 Now tell me what you'd like to do with this document"
     }