Spaces:
Running
Running
Update agent.py
Browse files
agent.py
CHANGED
@@ -149,19 +149,19 @@ supabase: Client = create_client(
|
|
149 |
|
150 |
# Initialize Supabase vector store
|
151 |
try:
|
152 |
-
vector_store = SupabaseVectorStore(
|
153 |
-
|
154 |
-
|
155 |
-
|
156 |
-
|
157 |
-
)
|
158 |
-
|
159 |
-
# Create retriever tool
|
160 |
-
create_retriever_tool = create_retriever_tool(
|
161 |
-
|
162 |
-
|
163 |
-
|
164 |
-
)
|
165 |
except Exception as e:
|
166 |
print(f"Error: {e}")
|
167 |
|
|
|
149 |
|
150 |
# Initialize Supabase vector store
|
151 |
try:
|
152 |
+
vector_store = SupabaseVectorStore(
|
153 |
+
client=supabase,
|
154 |
+
embedding=embeddings,
|
155 |
+
table_name="documents",
|
156 |
+
query_name="match_documents_langchain"
|
157 |
+
)
|
158 |
+
|
159 |
+
# Create retriever tool
|
160 |
+
create_retriever_tool = create_retriever_tool(
|
161 |
+
retriever=vector_store.as_retriever(),
|
162 |
+
name="Question Search",
|
163 |
+
description="A tool to retrieve similar questions from a vector store."
|
164 |
+
)
|
165 |
except Exception as e:
|
166 |
print(f"Error: {e}")
|
167 |
|