Merge pull request #1905 from 7a6ac0/r_score_type

fix: change r_score type to float
2024-05-02 02:22:59 -07:00 · 2024-05-02 02:22:59 -07:00 · e700552f93
parent d680d52b85 fffd283b0c
commit e700552f93
1 changed files with 1 additions and 1 deletions
--- a/backend/apps/rag/utils.py
+++ b/backend/apps/rag/utils.py
@ -53,7 +53,7 @@ def query_doc_with_hybrid_search(
    embedding_function,
    k: int,
    reranking_function,
-    r: int,
+    r: float,
 ):
    try:
        collection = CHROMA_CLIENT.get_collection(name=collection_name)