From 56fc5e2cc5319a86b5aeb8d00263c183f6d5c36f Mon Sep 17 00:00:00 2001
From: crazywoola <427733928@qq.com>
Date: Wed, 30 Apr 2025 11:00:26 +0800
Subject: [PATCH] fix: lint

---
 api/core/agent/cot_agent_runner.py                          | 2 +-
 api/core/agent/fc_agent_runner.py                           | 2 +-
 .../tools/utils/dataset_retriever/dataset_retriever_tool.py | 6 +++---
 3 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/api/core/agent/cot_agent_runner.py b/api/core/agent/cot_agent_runner.py
index 507dedaedd..de3b7e1ad7 100644
--- a/api/core/agent/cot_agent_runner.py
+++ b/api/core/agent/cot_agent_runner.py
@@ -73,7 +73,7 @@ class CotAgentRunner(BaseAgentRunner, ABC):
         if app_config.dataset is not None:
             metadata_filtering_conditions = app_config.dataset.retrieve_config.metadata_filtering_conditions
             for key, dataset_retriever_tool in tool_instances.items():
-                if hasattr(dataset_retriever_tool, 'retrieval_tool'):
+                if hasattr(dataset_retriever_tool, "retrieval_tool"):
                     dataset_retriever_tool.retrieval_tool.metadata_filtering_conditions = metadata_filtering_conditions
 
         function_call_state = True
diff --git a/api/core/agent/fc_agent_runner.py b/api/core/agent/fc_agent_runner.py
index 419196b144..874bd6b93b 100644
--- a/api/core/agent/fc_agent_runner.py
+++ b/api/core/agent/fc_agent_runner.py
@@ -49,7 +49,7 @@ class FunctionCallAgentRunner(BaseAgentRunner):
         if app_config.dataset is not None:
             metadata_filtering_conditions = app_config.dataset.retrieve_config.metadata_filtering_conditions
             for key, dataset_retriever_tool in tool_instances.items():
-                if hasattr(dataset_retriever_tool, 'retrieval_tool'):
+                if hasattr(dataset_retriever_tool, "retrieval_tool"):
                     dataset_retriever_tool.retrieval_tool.metadata_filtering_conditions = metadata_filtering_conditions
 
         assert app_config.agent
diff --git a/api/core/tools/utils/dataset_retriever/dataset_retriever_tool.py b/api/core/tools/utils/dataset_retriever/dataset_retriever_tool.py
index 031ca517b2..dcd3d080f3 100644
--- a/api/core/tools/utils/dataset_retriever/dataset_retriever_tool.py
+++ b/api/core/tools/utils/dataset_retriever/dataset_retriever_tool.py
@@ -4,9 +4,9 @@ from pydantic import BaseModel, Field
 
 from core.rag.datasource.retrieval_service import RetrievalService
 from core.rag.entities.context_entities import DocumentContext
+from core.rag.entities.metadata_entities import MetadataCondition
 from core.rag.models.document import Document as RetrievalDocument
 from core.rag.retrieval.retrieval_methods import RetrievalMethod
-from core.rag.entities.metadata_entities import MetadataCondition
 from core.tools.utils.dataset_retriever.dataset_retriever_base_tool import DatasetRetrieverBaseTool
 from extensions.ext_database import db
 from models.dataset import Dataset
@@ -48,7 +48,7 @@ class DatasetRetrieverTool(DatasetRetrieverBaseTool):
             tenant_id=dataset.tenant_id,
             dataset_id=dataset.id,
             description=description,
-            metadata_filtering_conditions= MetadataCondition(),
+            metadata_filtering_conditions=MetadataCondition(),
             **kwargs,
         )
 
@@ -68,7 +68,7 @@ class DatasetRetrieverTool(DatasetRetrieverBaseTool):
                 dataset_id=dataset.id,
                 query=query,
                 external_retrieval_parameters=dataset.retrieval_model,
-                metadata_condition=self.metadata_filtering_conditions
+                metadata_condition=self.metadata_filtering_conditions,
             )
             for external_document in external_documents:
                 document = RetrievalDocument(