fix: ruff formatter

This commit is contained in:
Yeuoly 2024-12-26 13:22:18 +08:00
parent e231cf2c48
commit b980c07af8
2 changed files with 4 additions and 3 deletions

View File

@ -735,6 +735,7 @@ class ProviderManager:
)
# Get cached provider credentials
# error occurs
cached_provider_credentials = provider_credentials_cache.get()
if not cached_provider_credentials:

View File

@ -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 = {