Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
@@ -126,7 +126,7 @@ 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,
|
@@ -137,8 +137,8 @@ vector_store = SupabaseVectorStore(
|
|
137 |
# Create retriever with search parameters
|
138 |
retriever = vector_store.as_retriever(
|
139 |
search_kwargs={
|
140 |
-
"k": 3,
|
141 |
-
"filter": {}
|
142 |
}
|
143 |
)
|
144 |
|
|
|
126 |
supabase: Client = create_client(
|
127 |
os.environ.get("SUPABASE_URL"),
|
128 |
os.environ.get("SUPABASE_SERVICE_KEY"))
|
129 |
+
# Initialize vector store with correct configuration
|
130 |
vector_store = SupabaseVectorStore(
|
131 |
client=supabase,
|
132 |
embedding=embeddings,
|
|
|
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 |
|