Update app.py

app.py

@@ -18,6 +18,12 @@ import faiss
 from fastapi import FastAPI
 from ctransformers import AutoModelForCausalLM
 from src.core.cognitive_engine import CognitiveEngine
+from pydantic import BaseModel
+from memory import add_to_memory, get_memory
+
+class GenerateRequest(BaseModel):
+    prompt: str
+    session_id: str
 
 # Load LLM model
 llm_model = AutoModelForCausalLM.from_pretrained(
@@ -44,15 +50,20 @@ def home():
 def status():
     return {"status": "active", "agents": ["planner", "executor", "critic"]}
 
-@app.get("/generate")
-def generate(prompt: str):
-    response = llm_model(prompt)
-    add_to_memory(prompt, response)
-    return {"response": response}
+@app.post("/generate")
+def generate_post(data: GenerateRequest):
+    response = llm_model(data.prompt)
+    add_to_memory(data.session_id, data.prompt, response)
+    return {"session_id": data.session_id, "response": response}
 
 @app.get("/memory")
 def view_memory():
     return {"history": get_memory()}
+    
+@app.get("/memory/{session_id}")
+def session_memory(session_id: str):
+    return {"session_id": session_id, "history": get_memory(session_id)}
+
 
 # Test the model at startup (optional)
 if __name__ == "__main__":