Fix/hit test tsne issue (#2581)

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

            documents = vector.search_by_vector(
                query=query,
-                k=1,
+                top_k=1,
                score_threshold=score_threshold,
                filter={
                    'group_id': [dataset.id]