Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
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=
|
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(),
|