Browse Source

fix:
fix: Change the type from int to float

tabacoWang 1 year ago
parent
commit
fffd283b0c
1 changed files with 1 additions and 1 deletions
  1. 1 1
      backend/apps/rag/utils.py

+ 1 - 1
backend/apps/rag/utils.py

@@ -53,7 +53,7 @@ def query_doc_with_hybrid_search(
     embedding_function,
     embedding_function,
     k: int,
     k: int,
     reranking_function,
     reranking_function,
-    r: int,
+    r: float,
 ):
 ):
     try:
     try:
         collection = CHROMA_CLIENT.get_collection(name=collection_name)
         collection = CHROMA_CLIENT.get_collection(name=collection_name)