bstraehle commited on
Commit
075f373
·
verified ·
1 Parent(s): 4b5f3af

Update custom_utils.py

Browse files
Files changed (1) hide show
  1. 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