Browse Source

fix: ruff formatter

Yeuoly 5 months ago
parent
commit
b980c07af8

+ 1 - 0
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:

+ 3 - 3
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 = {