Spaces:
Runtime error
Runtime error
Update agent.py
Browse files
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 |
-
|
| 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()
|