Spaces:
Running
Running
Update custom_utils.py
Browse files- custom_utils.py +1 -2
custom_utils.py
CHANGED
@@ -130,13 +130,12 @@ def vector_search(openai_api_key, user_query, db, collection, additional_stages=
|
|
130 |
"limit": 20, # return top 20 matches
|
131 |
"filter": {
|
132 |
"$and": [
|
|
|
133 |
{"bedrooms": {"$eq": 1}}
|
134 |
]
|
135 |
},
|
136 |
}
|
137 |
}
|
138 |
-
|
139 |
-
# {"accommodates": {"$eq": 2}},
|
140 |
|
141 |
# Define the aggregate pipeline with the vector search stage and additional stages
|
142 |
pipeline = [vector_search_stage] + additional_stages
|
|
|
130 |
"limit": 20, # return top 20 matches
|
131 |
"filter": {
|
132 |
"$and": [
|
133 |
+
{"accommodates": {"$eq": 2}},
|
134 |
{"bedrooms": {"$eq": 1}}
|
135 |
]
|
136 |
},
|
137 |
}
|
138 |
}
|
|
|
|
|
139 |
|
140 |
# Define the aggregate pipeline with the vector search stage and additional stages
|
141 |
pipeline = [vector_search_stage] + additional_stages
|