bstraehle commited on
Commit
7ad4a46
·
verified ·
1 Parent(s): d824f11

Update custom_utils.py

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