Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
@@ -134,11 +134,12 @@ vector_store = SupabaseVectorStore(
|
|
134 |
query_name="match_documents"
|
135 |
)
|
136 |
|
137 |
-
# Create retriever with search
|
138 |
retriever = vector_store.as_retriever(
|
139 |
search_kwargs={
|
140 |
-
"k": 3,
|
141 |
-
"filter": {} # Default filter
|
|
|
142 |
}
|
143 |
)
|
144 |
|
@@ -146,7 +147,7 @@ retriever = vector_store.as_retriever(
|
|
146 |
retriever_tool = create_retriever_tool(
|
147 |
retriever=retriever,
|
148 |
name="document_retriever",
|
149 |
-
description="Searches for similar documents
|
150 |
)
|
151 |
|
152 |
|
|
|
134 |
query_name="match_documents"
|
135 |
)
|
136 |
|
137 |
+
# Create retriever with proper search configuration
|
138 |
retriever = vector_store.as_retriever(
|
139 |
search_kwargs={
|
140 |
+
"k": 3, # Number of results
|
141 |
+
"filter": {}, # Default filter
|
142 |
+
"score_threshold": 0.7 # Optional similarity threshold
|
143 |
}
|
144 |
)
|
145 |
|
|
|
147 |
retriever_tool = create_retriever_tool(
|
148 |
retriever=retriever,
|
149 |
name="document_retriever",
|
150 |
+
description="Searches for similar documents using vector similarity",
|
151 |
)
|
152 |
|
153 |
|