Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
@@ -126,17 +126,12 @@ embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-mpnet-b
|
|
126 |
supabase: Client = create_client(
|
127 |
os.environ.get("SUPABASE_URL"),
|
128 |
os.environ.get("SUPABASE_SERVICE_KEY"))
|
129 |
-
# Initialize vector store
|
130 |
vector_store = SupabaseVectorStore(
|
131 |
client=supabase,
|
132 |
embedding=embeddings,
|
133 |
table_name="documents",
|
134 |
-
query_name="match_documents"
|
135 |
-
columns={
|
136 |
-
"id": "document_id", # Maps to function output
|
137 |
-
"content": "document_content", # Maps to function output
|
138 |
-
"metadata": "document_metadata" # Maps to function output
|
139 |
-
}
|
140 |
)
|
141 |
|
142 |
# Create retriever with search parameters
|
|
|
126 |
supabase: Client = create_client(
|
127 |
os.environ.get("SUPABASE_URL"),
|
128 |
os.environ.get("SUPABASE_SERVICE_KEY"))
|
129 |
+
# Initialize vector store without the 'columns' parameter
|
130 |
vector_store = SupabaseVectorStore(
|
131 |
client=supabase,
|
132 |
embedding=embeddings,
|
133 |
table_name="documents",
|
134 |
+
query_name="match_documents"
|
|
|
|
|
|
|
|
|
|
|
135 |
)
|
136 |
|
137 |
# Create retriever with search parameters
|