Fix/hit test tsne issue (#2581)

Co-authored-by: jyong <jyong@dify.ai>
2024-02-27 17:30:52 +08:00 · 2024-02-27 17:30:52 +08:00 · 920b2c2b40
commit 920b2c2b40
parent ac96d192a6
4 changed files with 6 additions and 5 deletions
--- a/api/core/rag/datasource/retrieval_service.py
+++ b/api/core/rag/datasource/retrieval_service.py
@ -101,7 +101,7 @@ class RetrievalService:

            documents = keyword.search(
                query,
-                k=top_k
+                top_k=top_k
            )
            all_documents.extend(documents)

@ -121,7 +121,7 @@ class RetrievalService:
            documents = vector.search_by_vector(
                query,
                search_type='similarity_score_threshold',
-                k=top_k,
+                top_k=top_k,
                score_threshold=score_threshold,
                filter={
                    'group_id': [dataset.id]