Update api.py

api.py

@@ -1,6 +1,6 @@
 from fastapi import FastAPI
 from pydantic import BaseModel
-from typing import Any, Dict
+from typing import Any, Dict, List
 import os
 
 from process_aware_rag import ProcessAwareRAG
@@ -8,8 +8,14 @@ from process_aware_rag import ProcessAwareRAG
 app = FastAPI(title="Legal Process Guide API", version="1.0.0")
 
 
+class Message(BaseModel):
+    role: str   # "user" or "assistant"
+    content: str
+
+
 class QueryRequest(BaseModel):
     query: str
+    history: List[Message] = []   # user sends past chat messages
 
 
 class QueryResponse(BaseModel):
@@ -31,7 +37,8 @@ def health() -> Dict[str, str]:
 
 @app.post("/query", response_model=QueryResponse)
 def query_endpoint(req: QueryRequest) -> QueryResponse:
-    result = app.state.rag.process_query(req.query)
+    # Pass query + history into RAG
+    result = app.state.rag.process_query(req.query, [m.dict() for m in req.history])
     return QueryResponse(
         response=result["response"],
         classification=result["classification"],