Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -38,7 +38,31 @@ def invoke(openai_api_key, prompt, rag_option):
|
|
38 |
#}
|
39 |
|
40 |
#additional_stages = [match_stage]
|
41 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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)
|