jonathanjordan21 commited on
Commit
53da814
·
verified ·
1 Parent(s): ea8be0e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -107,10 +107,12 @@ app = FastAPI()
107
  def get_data(query: str):
108
  # query = "How much money did the robbers steal?"
109
  found_docs = [x.model_dump() for x in qdrant.similarity_search(query)]
110
- found_docs.pop("id", None)
111
- for k,v in found_docs["metadata"].keys():
112
- if k[0] == "_":
113
- found_docs["metadata"].pop(k)
 
 
114
 
115
  return {
116
  "data": found_docs
 
107
  def get_data(query: str):
108
  # query = "How much money did the robbers steal?"
109
  found_docs = [x.model_dump() for x in qdrant.similarity_search(query)]
110
+
111
+ for doc in found_docs:
112
+ doc.pop("id", None)
113
+ for k,v in doc["metadata"].keys():
114
+ if k[0] == "_":
115
+ doc["metadata"].pop(k)
116
 
117
  return {
118
  "data": found_docs