ZeroTimo commited on
Commit
902a19b
·
verified ·
1 Parent(s): 5ebc793

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +1 -1
agent.py CHANGED
@@ -187,7 +187,7 @@ def build_graph(provider: str = "google"):
187
  # return {"messages": [sys_msg] + state["messages"] + [example_msg]}
188
 
189
  def retriever(state: MessagesState):
190
- """Retriever node"""
191
  current_question = state["messages"][-1].content
192
  similar_doc = vector_store.similarity_search(current_question, k=1)[0]
193
  page_content = similar_doc.page_content.strip()
 
187
  # return {"messages": [sys_msg] + state["messages"] + [example_msg]}
188
 
189
  def retriever(state: MessagesState):
190
+ """Retriever node"""
191
  current_question = state["messages"][-1].content
192
  similar_doc = vector_store.similarity_search(current_question, k=1)[0]
193
  page_content = similar_doc.page_content.strip()