lint fix

api/core/agent/base_agent_runner.py

@@ -9,12 +9,13 @@ from core.agent.entities import AgentEntity, AgentToolEntity
 from core.app.app_queue_manager import AppQueueManager
 from core.app.apps.agent_chat.app_config_manager import AgentChatAppConfig
 from core.app.apps.base_app_runner import AppRunner
-from core.callback_handler.agent_tool_callback_handler import DifyAgentCallbackHandler
-from core.callback_handler.index_tool_callback_handler import DatasetIndexToolCallbackHandler
 from core.app.entities.app_invoke_entities import (
     EasyUIBasedAppGenerateEntity,
-    InvokeFrom, EasyUIBasedModelConfigEntity,
+    EasyUIBasedModelConfigEntity,
+    InvokeFrom,
 )
+from core.callback_handler.agent_tool_callback_handler import DifyAgentCallbackHandler
+from core.callback_handler.index_tool_callback_handler import DatasetIndexToolCallbackHandler
 from core.file.message_file_parser import FileTransferMethod
 from core.memory.token_buffer_memory import TokenBufferMemory
 from core.model_manager import ModelInstance

api/core/agent/cot_agent_runner.py

@@ -4,8 +4,8 @@ from collections.abc import Generator
 from typing import Literal, Union
 
 from core.agent.base_agent_runner import BaseAgentRunner
-from core.app.app_queue_manager import PublishFrom
 from core.agent.entities import AgentPromptEntity, AgentScratchpadUnit
+from core.app.app_queue_manager import PublishFrom
 from core.model_runtime.entities.llm_entities import LLMResult, LLMResultChunk, LLMResultChunkDelta, LLMUsage
 from core.model_runtime.entities.message_entities import (
     AssistantPromptMessage,

api/core/app/app_manager.py

@@ -9,26 +9,26 @@ from flask import Flask, current_app
 from pydantic import ValidationError
 
 from core.app.app_config.easy_ui_based_app.model_config.converter import EasyUIBasedModelConfigEntityConverter
-from core.app.app_config.entities import EasyUIBasedAppModelConfigFrom, EasyUIBasedAppConfig, VariableEntity
+from core.app.app_config.entities import EasyUIBasedAppConfig, EasyUIBasedAppModelConfigFrom, VariableEntity
+from core.app.app_queue_manager import AppQueueManager, ConversationTaskStoppedException, PublishFrom
 from core.app.apps.agent_chat.app_config_manager import AgentChatAppConfigManager
 from core.app.apps.agent_chat.app_runner import AgentChatAppRunner
-from core.app.app_queue_manager import AppQueueManager, ConversationTaskStoppedException, PublishFrom
 from core.app.apps.chat.app_config_manager import ChatAppConfigManager
 from core.app.apps.chat.app_runner import ChatAppRunner
 from core.app.apps.completion.app_config_manager import CompletionAppConfigManager
 from core.app.apps.completion.app_runner import CompletionAppRunner
-from core.app.generate_task_pipeline import GenerateTaskPipeline
 from core.app.entities.app_invoke_entities import (
     EasyUIBasedAppGenerateEntity,
     InvokeFrom,
 )
+from core.app.generate_task_pipeline import GenerateTaskPipeline
 from core.file.file_obj import FileObj
 from core.model_runtime.errors.invoke import InvokeAuthorizationError, InvokeError
 from core.model_runtime.model_providers.__base.large_language_model import LargeLanguageModel
 from core.prompt.utils.prompt_template_parser import PromptTemplateParser
 from extensions.ext_database import db
 from models.account import Account
-from models.model import App, Conversation, EndUser, Message, MessageFile, AppMode, AppModelConfig
+from models.model import App, AppMode, AppModelConfig, Conversation, EndUser, Message, MessageFile
 
 logger = logging.getLogger(__name__)
 

api/core/memory/token_buffer_memory.py

@@ -1,4 +1,3 @@
-from core.app.app_config.entities import FileUploadEntity
 from core.app.app_config.features.file_upload.manager import FileUploadConfigManager
 from core.file.message_file_parser import MessageFileParser
 from core.model_manager import ModelInstance

api/core/prompt/advanced_prompt_transform.py

@@ -1,6 +1,6 @@
 from typing import Optional
 
-from core.app.app_config.entities import PromptTemplateEntity, AdvancedCompletionPromptTemplateEntity
+from core.app.app_config.entities import AdvancedCompletionPromptTemplateEntity, PromptTemplateEntity
 from core.app.entities.app_invoke_entities import EasyUIBasedModelConfigEntity
 from core.file.file_obj import FileObj
 from core.memory.token_buffer_memory import TokenBufferMemory

api/core/rag/retrieval/dataset_retrieval.py

@@ -3,9 +3,9 @@ from typing import Optional, cast
 from langchain.tools import BaseTool
 
 from core.app.app_config.entities import DatasetEntity, DatasetRetrieveConfigEntity
+from core.app.entities.app_invoke_entities import EasyUIBasedModelConfigEntity, InvokeFrom
 from core.callback_handler.index_tool_callback_handler import DatasetIndexToolCallbackHandler
 from core.entities.agent_entities import PlanningStrategy
-from core.app.entities.app_invoke_entities import InvokeFrom, EasyUIBasedModelConfigEntity
 from core.memory.token_buffer_memory import TokenBufferMemory
 from core.model_runtime.entities.model_entities import ModelFeature
 from core.model_runtime.model_providers.__base.large_language_model import LargeLanguageModel

api/core/tools/tool/dataset_retriever_tool.py

@@ -3,8 +3,8 @@ from typing import Any
 from langchain.tools import BaseTool
 
 from core.app.app_config.entities import DatasetRetrieveConfigEntity
-from core.callback_handler.index_tool_callback_handler import DatasetIndexToolCallbackHandler
 from core.app.entities.app_invoke_entities import InvokeFrom
+from core.callback_handler.index_tool_callback_handler import DatasetIndexToolCallbackHandler
 from core.rag.retrieval.dataset_retrieval import DatasetRetrieval
 from core.tools.entities.common_entities import I18nObject
 from core.tools.entities.tool_entities import ToolDescription, ToolIdentity, ToolInvokeMessage, ToolParameter

api/services/workflow/workflow_converter.py

@@ -1,8 +1,15 @@
 import json
 from typing import Optional
 
-from core.app.app_config.entities import VariableEntity, ExternalDataVariableEntity, DatasetEntity, \
-    DatasetRetrieveConfigEntity, ModelConfigEntity, PromptTemplateEntity, FileUploadEntity
+from core.app.app_config.entities import (
+    DatasetEntity,
+    DatasetRetrieveConfigEntity,
+    ExternalDataVariableEntity,
+    FileUploadEntity,
+    ModelConfigEntity,
+    PromptTemplateEntity,
+    VariableEntity,
+)
 from core.app.app_manager import EasyUIBasedAppManager
 from core.helper import encrypter
 from core.model_runtime.entities.llm_entities import LLMMode