TruongTrongTien commited on
Commit
aef9bf8
1 Parent(s): d6070c8

Phase2/TienTT: Add Query vector database and Delete tests

Browse files
app/modules/question_tests_retrieval/__init__.py CHANGED
@@ -25,6 +25,4 @@ async def send_jd(txt_file: Annotated[UploadFile, File(..., description="The JD
25
  return {"message": "Please upload only .txt file", "error": str(e)}
26
  except Exception as e:
27
  return {"message": "Please upload only .txt file", "error": str(e)}
28
-
29
- # @qtretrieval_router.post("/download_tests")
30
- # async def download_tests()
 
25
  return {"message": "Please upload only .txt file", "error": str(e)}
26
  except Exception as e:
27
  return {"message": "Please upload only .txt file", "error": str(e)}
28
+
 
 
app/modules/question_tests_retrieval/models/question_tests_logic.py CHANGED
@@ -36,7 +36,7 @@ gemini_evaluator = load_evaluator("embedding_distance", distance_metric=Embeddin
36
  # else:
37
  # return False
38
 
39
- def compare_vector(description_vector, max_number_of_points=10):
40
  similarity_list = qdrant_client.search(
41
  collection_name="question_tests",
42
  query_vector=description_vector,
 
36
  # else:
37
  # return False
38
 
39
+ def compare_vector(description_vector, max_number_of_points=3):
40
  similarity_list = qdrant_client.search(
41
  collection_name="question_tests",
42
  query_vector=description_vector,