debug

app.py

@@ -10,7 +10,7 @@ from pydantic import BaseModel
 from agent.agent_graph.StateTasks import ProblemState
 import subprocess
 from Queue_Producer import send_message
-
+import redis
 ##################################################
 # START CONSUMERS in a separate process
 ##################################################
@@ -21,12 +21,22 @@ for i in range(1,4):  # Start 3 consumers
 ##################################################
 # START API
 ##################################################
-
+redis_host = os.environ["REDIS_HOST"]
+redis_port = os.environ["REDIS_PORT"]   
+redis_password = os.environ["REDIS_PASSWORD"]
+
+# Create Redis connection (global to make the get very light)
+redis_conn = redis.Redis(
+        host=redis_host,
+        port=redis_port,
+        decode_responses=True,
+        username="default",
+        password=redis_password,
+    )
 # Create app instance
 app = FastAPI()
 print("Starting API Server...")
 
-
 class RequestModel(BaseModel):
     prompt: str
     ht_token : str
@@ -35,6 +45,10 @@ class RequestModel(BaseModel):
     memory: list[str]
     last_state : str
 
+class RequestAnswer(BaseModel):
+    user_id: str
+    msg_id:str
+
 
 ##################################################
 # ROUTES
@@ -65,12 +79,16 @@ def old_call(request: RequestModel):
     return {"Data": answer}
 """
     
-@app.post("/call/")
-def call(request: RequestModel):
-    
-    
-
+@app.post("/Message/Send/")
+def call(request: RequestModel): 
     return send_message(json.dumps(request.model_dump()))
 
+
+@app.get("/Message/Answer/")
+def call(request: RequestAnswer): 
+    return redis_conn.get(f'ANSWER_FOR_USER_ID{request.user_id}_OF_{request.msg_id}',default="No ANSWER YET")
+
+
+
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)