Update app.py

app.py

@@ -11,7 +11,9 @@ from langchain_mcp_adapters.client import MultiServerMCPClient
 from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain.agents import create_agent
 from langchain_core.messages import HumanMessage
+from langgraph.store.memory import InMemoryStore
 
+my_store = InMemoryStore()
 
 
 # --- GLOBAL STATE ---
@@ -82,7 +84,8 @@ async def initialize_agent(api_key: str = Form(...), model_name: str = Form(...)
         agent_executor = create_agent(
             llm,                        
             mcp_tools,                  
-            system_prompt = system_prompt
+            system_prompt = system_prompt,
+            store=my_store
             )
         
         return {"status": "Success", "message": f"Agent initialized with {model_name}"}