Spaces:
Build error
Build error
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=
|
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,
|