更新 RedisRag 搜索方法，使用相似性搜索并返回相关性评分

terateams · Nov 19, 2024 · 92f7626 · 92f7626
1 parent 57dddbd
commit 92f7626
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -177,7 +177,7 @@ async def redis_rag_search(
     try:
         cachekey = f"rag_{md5hash(query.query)}"
         vdb = RedisRag.get_vectordb(query.index)
-        result = await vdb.amax_marginal_relevance_search(
+        result = await vdb.asimilarity_search_with_relevance_scores(
             query.query, k=query.topk, return_metadata=True
         )
         if not result: