Fix some RAG bugs (#2570)

Co-authored-by: jyong <jyong@dify.ai>
2024-02-27 11:39:05 +08:00 · 2024-02-27 11:39:05 +08:00 · 5b953c1ef2
commit 5b953c1ef2
parent 562ca45e07
10 changed files with 33 additions and 59 deletions
--- a/api/core/rag/datasource/retrieval_service.py
+++ b/api/core/rag/datasource/retrieval_service.py
@ -39,7 +39,8 @@ class RetrievalService:
                'flask_app': current_app._get_current_object(),
                'dataset_id': dataset_id,
                'query': query,
-                'top_k': top_k
+                'top_k': top_k,
+                'all_documents': all_documents
            })
            threads.append(keyword_thread)
            keyword_thread.start()