fixed state bug

app.py

@@ -2,6 +2,7 @@ from fastapi import FastAPI
 import uvicorn
 import sys
 import os
+import json
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..')))
 from Server import get_response
 from pydantic import BaseModel
@@ -16,6 +17,7 @@ class RequestModel(BaseModel):
     user_email : str
     user_name : str
     memory: list[str]
+    last_state : str
 
 
 @app.get("/")
@@ -25,17 +27,23 @@ def read_root():
 
 @app.post("/call/")
 def call(request: RequestModel):
-    state: ProblemState = {
-        "question": request.prompt,
-        "memory": request.memory
-    }
-    
+    # fill with last state 
+    try:
+        state = json.loads(request.last_state) 
+    except Exception:
+        state: ProblemState = {
+            "question": request.prompt,
+            "memory": request.memory
+        }
+
     answer = get_response(request.prompt, request.memory,request.ht_token,state,request.user_email,request.user_name)
     
     # drop unserlizable keys
     for k in ["llm","rag_model"]:
         answer[k] = ""
-    return {"answer": answer}
+
+
+    return {"Data": answer}