bstraehle commited on
Commit
0ca68e8
·
verified ·
1 Parent(s): 2c368a3

Update custom_utils.py

Browse files
Files changed (1) hide show
  1. custom_utils.py +0 -7
custom_utils.py CHANGED
@@ -184,13 +184,6 @@ def vector_search(user_query, db, collection, additional_stages=[], vector_index
184
 
185
  return list(results)
186
 
187
- class SearchResultItem(BaseModel):
188
- name: str
189
- accommodates: Optional[int] = None
190
- bedrooms: Optional[int] = None
191
- address: Address
192
- space: str = None
193
-
194
  def handle_user_prompt(openai_api_key, prompt, db, collection, stages=[], vector_index="vector_index"):
195
  openai.api_key = openai_api_key
196
 
 
184
 
185
  return list(results)
186
 
 
 
 
 
 
 
 
187
  def handle_user_prompt(openai_api_key, prompt, db, collection, stages=[], vector_index="vector_index"):
188
  openai.api_key = openai_api_key
189