diff --git a/api/core/provider_manager.py b/api/core/provider_manager.py index f9f5672b15..7132843e79 100644 --- a/api/core/provider_manager.py +++ b/api/core/provider_manager.py @@ -735,6 +735,7 @@ class ProviderManager: ) # Get cached provider credentials + # error occurs cached_provider_credentials = provider_credentials_cache.get() if not cached_provider_credentials: diff --git a/api/core/tools/utils/dataset_retriever/dataset_multi_retriever_tool.py b/api/core/tools/utils/dataset_retriever/dataset_multi_retriever_tool.py index 484f6ea069..802bee4217 100644 --- a/api/core/tools/utils/dataset_retriever/dataset_multi_retriever_tool.py +++ b/api/core/tools/utils/dataset_retriever/dataset_multi_retriever_tool.py @@ -88,7 +88,7 @@ class DatasetMultiRetrieverTool(DatasetRetrieverBaseTool): DocumentSegment.dataset_id.in_(self.dataset_ids), DocumentSegment.completed_at.isnot(None), DocumentSegment.status == "completed", - DocumentSegment.enabled == True, # noqa + DocumentSegment.enabled == True, DocumentSegment.index_node_id.in_(index_node_ids), ).all() @@ -109,8 +109,8 @@ class DatasetMultiRetrieverTool(DatasetRetrieverBaseTool): dataset = Dataset.query.filter_by(id=segment.dataset_id).first() document = Document.query.filter( Document.id == segment.document_id, - Document.enabled == True, # noqa - Document.archived == False, # noqa + Document.enabled == True, + Document.archived == False, ).first() if dataset and document: source = {