bstraehle commited on
Commit
6eee7c9
·
verified ·
1 Parent(s): 075f373

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +25 -1
app.py CHANGED
@@ -38,7 +38,31 @@ def invoke(openai_api_key, prompt, rag_option):
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)
 
38
  #}
39
 
40
  #additional_stages = [match_stage]
41
+ ###
42
+ projection_stage = {
43
+ "$project": {
44
+ "_id": 0,
45
+ "name": 1,
46
+ "accommodates": 1,
47
+ "address.street": 1,
48
+ "address.government_area": 1,
49
+ "address.market": 1,
50
+ "address.country": 1,
51
+ "address.country_code": 1,
52
+ "address.location.type": 1,
53
+ "address.location.coordinates": 1,
54
+ "address.location.is_location_exact": 1,
55
+ "summary": 1,
56
+ "space": 1,
57
+ "neighborhood_overview": 1,
58
+ "notes": 1,
59
+ "score": {"$meta": "vectorSearchScore"}
60
+ }
61
+ }
62
+
63
+ additional_stages = [projection_stage]
64
+ ###
65
+ #additional_stages = []
66
  ###
67
 
68
  search_results = rag_retrieval(openai_api_key, prompt, db, collection, additional_stages)