From fffd283b0ccee95db391f8ceec342fe67d8f1fbf Mon Sep 17 00:00:00 2001
From: tabacoWang <7a6ac0@users.noreply.github.com>
Date: Thu, 2 May 2024 13:45:19 +0800
Subject: [PATCH] fix: fix: Change the type from int to float

---
 backend/apps/rag/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/backend/apps/rag/utils.py b/backend/apps/rag/utils.py
index 9941eb46..b1142e85 100644
--- 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)