updating tools - memory persistency

app.py

@@ -10,9 +10,8 @@ from langgraph.prebuilt import tools_condition
 from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
 from langgraph.prebuilt import create_react_agent
 from langgraph.store.memory import InMemoryStore
-from langmem import create_manage_memory_tool, create_search_memory_tool
 
-from tools import weather_info_tool, hub_stats_tool, memory_management_tool, duckduckgo_search_tool
+from tools import weather_info_tool, hub_stats_tool, duckduckgo_search_tool, manage_memory_tool, search_memory_tool
 from retriever import load_guest_dataset
 
 # Initialize the Hugging Face model
@@ -34,23 +33,19 @@ store = InMemoryStore(
 ) 
 
 
-agent = create_react_agent(
-    llm,
-    tools=[
-        # Memory tools use LangGraph's BaseStore for persistence (4)
-        create_manage_memory_tool(namespace=("memories",)),
-        create_search_memory_tool(namespace=("memories",)),
-    ],
-    store=store,
-)
-
 # adding tools
 
 guest_info_tool = load_guest_dataset()
 
-tools = [guest_info_tool, weather_info_tool, hub_stats_tool, duckduckgo_search_tool, memory_management_tool]
+tools = [guest_info_tool, weather_info_tool, hub_stats_tool, duckduckgo_search_tool, manage_memory_tool, search_memory_tool]
 chat_with_tools = chat.bind_tools(tools)
 
+agent = create_react_agent(
+    "openai:gpt-4o",
+    tools=tools,
+    store=store,
+)
+
 # Generate the AgentState and Agent graph
 class AgentState(TypedDict):
     messages: Annotated[list[AnyMessage], add_messages]

tools.py

@@ -3,6 +3,7 @@ from langchain_community.tools import DuckDuckGoSearchRun
 from langgraph.store.memory import InMemoryStore
 import random
 from huggingface_hub import list_models
+from langmem import create_manage_memory_tool, create_search_memory_tool
 
 # Initialize the DuckDuckGo search tool
 search_tool = DuckDuckGoSearchRun()
@@ -72,5 +73,6 @@ duckduckgo_search_tool = Tool(
     description="Searches the web for information."
 )
 
-
+manage_memory_tool = create_manage_memory_tool(namespace=("memories",))
+search_memory_tool = create_search_memory_tool(namespace=("memories",))