bstraehle commited on
Commit
3cc300e
·
verified ·
1 Parent(s): 868ab08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -7
app.py CHANGED
@@ -30,14 +30,15 @@ def invoke(openai_api_key, prompt, rag_option):
30
  else:
31
  ### Pre-retrieval processing: index filter
32
  ### Post-retrieval processing: result filter
33
- match_stage = {
34
- "$match": {
35
- "accommodates": { "$eq": 2},
36
- "bedrooms": { "$eq": 1}
37
- }
38
- }
39
 
40
- additional_stages = [match_stage]
 
41
  ###
42
 
43
  search_results = rag_retrieval(openai_api_key, prompt, db, collection, additional_stages)
 
30
  else:
31
  ### Pre-retrieval processing: index filter
32
  ### Post-retrieval processing: result filter
33
+ #match_stage = {
34
+ # "$match": {
35
+ # "accommodates": { "$eq": 2},
36
+ # "bedrooms": { "$eq": 1}
37
+ # }
38
+ #}
39
 
40
+ #additional_stages = [match_stage]
41
+ additional_stages = []
42
  ###
43
 
44
  search_results = rag_retrieval(openai_api_key, prompt, db, collection, additional_stages)