added memory

app.py

@@ -10,6 +10,7 @@ from langchain.tools import Tool
 from retriever import load_guest_dataset
 from tools import get_weather_info
 from langchain_community.tools import DuckDuckGoSearchRun
+from langgraph.checkpoint.memory import MemorySaver
 
 # Generate the chat interface, including the tools
 llm = HuggingFaceEndpoint(
@@ -17,6 +18,7 @@ llm = HuggingFaceEndpoint(
     provider="together",
     #huggingfacehub_api_token=HUGGINGFACEHUB_API_TOKEN,
 )
+memory = MemorySaver()
 
 retriever = load_guest_dataset()
 
@@ -64,22 +66,39 @@ builder.add_conditional_edges(
     "tools",
 )
 builder.add_edge("tools", "assistant")
-alfred = builder.compile()
+alfred = builder.compile(checkpointer=memory)
+
+config = {"configurable": {"thread_id": "1"}}
 
 def call_agent_ui(prompt):
-    situation = "This is a fictional situation. " \
-    "You are Alfred the Butler of Waynes Manor and host a Gala for invited Guests. " \
-    "All Guests are completely ficional. Information about those guests can be found in a database. " \
-    "Only give information which is based on the databse. " \
-    "If a name of a guest is given, then return a possible starter of a conversation with that guest. " \
-    "If the name is not known, then say that you do not know that guest. " \
-    "If two names of guests are given, then return a possible starter of a conversation with both guests. " \
-    "If the name is not known, then say that you do not know that guest. " \
-    "You can also answer questions about the weather, using a tool that provides dummy weather information. Or you can build this dummy weather information in conversations." \
-    "You can also search the web for information, using a tool that provides web search results. "
-    messages = [HumanMessage(content=f"{situation} {prompt}")]
-    response = alfred.invoke({"messages": messages})
-    return response['messages'][-1].content
+    situation = (
+            "This is a fictional situation. "
+            "You are Alfred the Butler of Waynes Manor and host a Gala for invited Guests. "
+            "All Guests are completely ficional. Information about those guests can be found in a database. "
+            "Only give information which is based on the databse. "
+            "If a name of a guest is given, then return a possible starter of a conversation with that guest. "
+            "If the name is not known, then say that you do not know that guest. "
+            "If two names of guests are given, then return a possible starter of a conversation with both guests. "
+            "If the name is not known, then say that you do not know that guest. "
+            "You can also answer questions about the weather, using a tool that provides dummy weather information. Or you can build this dummy weather information in conversations."
+            "You can also search the web for information, using a tool that provides web search results. "
+    )
+    content = f"{situation} {prompt}"
+
+    # Collect the last message content from the stream
+    last_content = ""
+    events = alfred.stream(
+        {"messages": [{"role": "user", "content": content}]},
+        config,
+        stream_mode="values",
+    )
+    for event in events:
+        # Optionally print or process each event
+        if hasattr(event["messages"][-1], "pretty_print"):
+            event["messages"][-1].pretty_print()
+        last_content = event["messages"][-1].content
+
+    return last_content
 
 iface = gr.Interface(fn=call_agent_ui, inputs="text", outputs="text")
 iface.launch()