Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -130,7 +130,7 @@ vector_store = SupabaseVectorStore(
|
|
| 130 |
table_name="documents",
|
| 131 |
query_name="match_documents_langchain"
|
| 132 |
)
|
| 133 |
-
|
| 134 |
|
| 135 |
retriever_tool = create_retriever_tool(
|
| 136 |
retriever=vector_store.as_retriever(),
|
|
@@ -160,6 +160,7 @@ def build_graph(provider: str = "groq"):
|
|
| 160 |
|
| 161 |
def retriever(state: MessagesState):
|
| 162 |
similar = vector_store.similarity_search(state["messages"][0].content)
|
|
|
|
| 163 |
example_msg = HumanMessage(
|
| 164 |
content=f"Here’s a similar QA pair for grounding:\n\n{similar[0].page_content}"
|
| 165 |
)
|
|
|
|
| 130 |
table_name="documents",
|
| 131 |
query_name="match_documents_langchain"
|
| 132 |
)
|
| 133 |
+
|
| 134 |
|
| 135 |
retriever_tool = create_retriever_tool(
|
| 136 |
retriever=vector_store.as_retriever(),
|
|
|
|
| 160 |
|
| 161 |
def retriever(state: MessagesState):
|
| 162 |
similar = vector_store.similarity_search(state["messages"][0].content)
|
| 163 |
+
print("🔍 Retrieved vector documents:", similar)
|
| 164 |
example_msg = HumanMessage(
|
| 165 |
content=f"Here’s a similar QA pair for grounding:\n\n{similar[0].page_content}"
|
| 166 |
)
|