Fix/agent external knowledge retrieval (#9241)

2024-10-11 19:21:03 +08:00
parent 44f6a536d2
commit 42b02b3a5f
7 changed files with 225 additions and 92 deletions
--- a/api/core/rag/retrieval/dataset_retrieval.py
+++ b/api/core/rag/retrieval/dataset_retrieval.py
@@ -539,7 +539,7 @@ class DatasetRetrieval:
                continue

            # pass if dataset is not available
-            if dataset and dataset.available_document_count == 0:
+            if dataset and dataset.provider != "external" and dataset.available_document_count == 0:
                continue

            available_datasets.append(dataset)