jonathanjordan21 commited on
Commit
2864bac
·
verified ·
1 Parent(s): 06131ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -299,7 +299,7 @@ app = FastAPI()
299
  def get_data(query: str):
300
  # query = "How much money did the robbers steal?"
301
  # found_docs = [x.model_dump() for x in qdrant.similarity_search(query)]
302
- found_docs = [x[0].model_dump() for x in qdrant.similarity_search_with_score(query, k=10, score_threshold=0.)]
303
 
304
  for doc in found_docs:
305
  doc.pop("id", None)
 
299
  def get_data(query: str):
300
  # query = "How much money did the robbers steal?"
301
  # found_docs = [x.model_dump() for x in qdrant.similarity_search(query)]
302
+ found_docs = [x[0].model_dump() for x in qdrant.similarity_search_with_score(query, k=10, score_threshold=0.4)]
303
 
304
  for doc in found_docs:
305
  doc.pop("id", None)