Commit ·
cc41b07
1
Parent(s): 8c96734
updated agent again
Browse files
agent.py
CHANGED
|
@@ -191,6 +191,7 @@ def build_graph(provider: str = "groq"):
|
|
| 191 |
"""Retriever node"""
|
| 192 |
print(f"[Retriever] Input: {state['messages'][0].content}")
|
| 193 |
similar_question = vector_store.similarity_search(state["messages"][0].content)
|
|
|
|
| 194 |
example_msg = HumanMessage(
|
| 195 |
content=f"Here I provide a similar question and answer for reference: \n\n{similar_question[0].page_content}",
|
| 196 |
)
|
|
|
|
| 191 |
"""Retriever node"""
|
| 192 |
print(f"[Retriever] Input: {state['messages'][0].content}")
|
| 193 |
similar_question = vector_store.similarity_search(state["messages"][0].content)
|
| 194 |
+
print(f"[Retriever] similar_question: {similar_question}")
|
| 195 |
example_msg = HumanMessage(
|
| 196 |
content=f"Here I provide a similar question and answer for reference: \n\n{similar_question[0].page_content}",
|
| 197 |
)
|