CyberAssassin commited on
Commit
cbf8aae
·
verified ·
1 Parent(s): 9d155b6

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +5 -4
agent.py CHANGED
@@ -134,11 +134,12 @@ vector_store = SupabaseVectorStore(
134
  query_name="match_documents"
135
  )
136
 
137
- # Create retriever with search parameters
138
  retriever = vector_store.as_retriever(
139
  search_kwargs={
140
- "k": 3, # Number of results to return
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 in knowledge base",
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