AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-18 23:55:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: ruff formatter

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
api/core/provider_manager.py
View File

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

6
api/core/tools/utils/dataset_retriever/dataset_multi_retriever_tool.py
View File

@ -88,7 +88,7 @@ class DatasetMultiRetrieverTool(DatasetRetrieverBaseTool):
DocumentSegment.dataset_id.in_(self.dataset_ids), DocumentSegment.dataset_id.in_(self.dataset_ids),
DocumentSegment.completed_at.isnot(None), DocumentSegment.completed_at.isnot(None),
DocumentSegment.status == "completed", DocumentSegment.status == "completed",
DocumentSegment.enabled == True, # noqa DocumentSegment.enabled == True,
DocumentSegment.index_node_id.in_(index_node_ids), DocumentSegment.index_node_id.in_(index_node_ids),
).all() ).all()
@ -109,8 +109,8 @@ class DatasetMultiRetrieverTool(DatasetRetrieverBaseTool):
dataset = Dataset.query.filter_by(id=segment.dataset_id).first() dataset = Dataset.query.filter_by(id=segment.dataset_id).first()
document = Document.query.filter( document = Document.query.filter(
Document.id == segment.document_id, Document.id == segment.document_id,
Document.enabled == True, # noqa Document.enabled == True,
Document.archived == False, # noqa Document.archived == False,
).first() ).first()
if dataset and document: if dataset and document:
source = { source = {