AdityaPandey commited on
Commit
1ce3fe9
·
verified ·
1 Parent(s): 1e44ec3

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +4 -1
agent.py CHANGED
@@ -127,9 +127,12 @@ supabase: Client = create_client(
127
  os.environ.get("SUPABASE_SERVICE_KEY"))
128
  vector_store = SupabaseVectorStore(
129
  client=supabase,
130
- embedding= embeddings,
131
  table_name="documents",
132
  query_name="match_documents",
 
 
 
133
  )
134
  create_retriever_tool = create_retriever_tool(
135
  retriever=vector_store.as_retriever(),
 
127
  os.environ.get("SUPABASE_SERVICE_KEY"))
128
  vector_store = SupabaseVectorStore(
129
  client=supabase,
130
+ embedding=embeddings,
131
  table_name="documents",
132
  query_name="match_documents",
133
+ # Add these required parameters
134
+ match_count=5, # Number of documents to return
135
+ match_threshold=0.7, # Similarity threshold (between 0 and 1)
136
  )
137
  create_retriever_tool = create_retriever_tool(
138
  retriever=vector_store.as_retriever(),