ernani commited on
Commit
5957e2a
·
1 Parent(s): b953b80

added memory_store and search_tool [the implementation was missing]

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -8,6 +8,8 @@ from langgraph.prebuilt import ToolNode
8
  from langgraph.graph import START, StateGraph
9
  from langgraph.prebuilt import tools_condition
10
  from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
 
 
11
 
12
  from tools import weather_info_tool, hub_stats_tool, memory_management_tool, duckduckgo_search_tool
13
  from retriever import load_guest_dataset
@@ -22,6 +24,9 @@ llm = HuggingFaceEndpoint(
22
 
23
  chat = ChatHuggingFace(llm=llm, verbose=True)
24
 
 
 
 
25
  # adding tools
26
 
27
  guest_info_tool = load_guest_dataset()
@@ -41,7 +46,7 @@ def assistant(state: AgentState):
41
 
42
  # Check if the query is about an unfamiliar guest
43
  if "guest" in state["messages"][-1].content:
44
- search_results = search_tool.forward(state["messages"][-1].content)
45
  return {
46
  "messages": [search_results]
47
  }
 
8
  from langgraph.graph import START, StateGraph
9
  from langgraph.prebuilt import tools_condition
10
  from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
11
+ from langchain.memory import ChatMessageHistory
12
+ from langchain.memory import ConversationBufferMemory
13
 
14
  from tools import weather_info_tool, hub_stats_tool, memory_management_tool, duckduckgo_search_tool
15
  from retriever import load_guest_dataset
 
24
 
25
  chat = ChatHuggingFace(llm=llm, verbose=True)
26
 
27
+ # Initialize memory store
28
+ memory_store = ConversationBufferMemory(memory_key="chat_history", return_messages=True)
29
+
30
  # adding tools
31
 
32
  guest_info_tool = load_guest_dataset()
 
46
 
47
  # Check if the query is about an unfamiliar guest
48
  if "guest" in state["messages"][-1].content:
49
+ search_results = duckduckgo_search_tool.forward(state["messages"][-1].content)
50
  return {
51
  "messages": [search_results]
52
  }