lint fix

api/core/agent/entities.py

@@ -1,5 +1,5 @@
 from enum import Enum
-from typing import Literal, Any, Union, Optional
+from typing import Any, Literal, Optional, Union
 
 from pydantic import BaseModel
 

api/core/app/app_config/base_app_config_manager.py

@@ -1,4 +1,4 @@
-from typing import Union, Optional
+from typing import Optional, Union
 
 from core.app.app_config.entities import AppAdditionalFeatures, EasyUIBasedAppModelConfigFrom
 from core.app.app_config.features.file_upload.manager import FileUploadConfigManager
@@ -6,8 +6,9 @@ from core.app.app_config.features.more_like_this.manager import MoreLikeThisConf
 from core.app.app_config.features.opening_statement.manager import OpeningStatementConfigManager
 from core.app.app_config.features.retrieval_resource.manager import RetrievalResourceConfigManager
 from core.app.app_config.features.speech_to_text.manager import SpeechToTextConfigManager
-from core.app.app_config.features.suggested_questions_after_answer.manager import \
-    SuggestedQuestionsAfterAnswerConfigManager
+from core.app.app_config.features.suggested_questions_after_answer.manager import (
+    SuggestedQuestionsAfterAnswerConfigManager,
+)
 from core.app.app_config.features.text_to_speech.manager import TextToSpeechConfigManager
 from models.model import AppModelConfig
 

api/core/app/app_config/easy_ui_based_app/model_config/converter.py

@@ -2,7 +2,6 @@ from typing import cast
 
 from core.app.app_config.entities import EasyUIBasedAppConfig
 from core.app.entities.app_invoke_entities import EasyUIBasedModelConfigEntity
-
 from core.entities.model_entities import ModelStatus
 from core.errors.error import ModelCurrentlyNotSupportError, ProviderTokenNotInitError, QuotaExceededError
 from core.model_runtime.entities.model_entities import ModelType

api/core/app/app_config/easy_ui_based_app/model_config/manager.py

@@ -1,5 +1,5 @@
 from core.app.app_config.entities import ModelConfigEntity
-from core.model_runtime.entities.model_entities import ModelType, ModelPropertyKey
+from core.model_runtime.entities.model_entities import ModelPropertyKey, ModelType
 from core.model_runtime.model_providers import model_provider_factory
 from core.provider_manager import ProviderManager
 

api/core/app/app_config/easy_ui_based_app/prompt_template/manager.py

@@ -1,5 +1,8 @@
-from core.app.app_config.entities import PromptTemplateEntity, \
-    AdvancedChatPromptTemplateEntity, AdvancedCompletionPromptTemplateEntity
+from core.app.app_config.entities import (
+    AdvancedChatPromptTemplateEntity,
+    AdvancedCompletionPromptTemplateEntity,
+    PromptTemplateEntity,
+)
 from core.model_runtime.entities.message_entities import PromptMessageRole
 from core.prompt.simple_prompt_transform import ModelMode
 from models.model import AppMode

api/core/app/app_config/easy_ui_based_app/variables/manager.py

@@ -1,13 +1,12 @@
 import re
-from typing import Tuple
 
-from core.app.app_config.entities import VariableEntity, ExternalDataVariableEntity
+from core.app.app_config.entities import ExternalDataVariableEntity, VariableEntity
 from core.external_data_tool.factory import ExternalDataToolFactory
 
 
 class BasicVariablesConfigManager:
     @classmethod
-    def convert(cls, config: dict) -> Tuple[list[VariableEntity], list[ExternalDataVariableEntity]]:
+    def convert(cls, config: dict) -> tuple[list[VariableEntity], list[ExternalDataVariableEntity]]:
         """
         Convert model config to model config
 

api/core/app/app_config/features/opening_statement/manager.py

@@ -1,9 +1,8 @@
-from typing import Tuple
 
 
 class OpeningStatementConfigManager:
     @classmethod
-    def convert(cls, config: dict) -> Tuple[str, list]:
+    def convert(cls, config: dict) -> tuple[str, list]:
         """
         Convert model config to model config
 

api/core/app/apps/advanced_chat/app_config_manager.py

@@ -5,11 +5,12 @@ from core.app.app_config.features.file_upload.manager import FileUploadConfigMan
 from core.app.app_config.features.opening_statement.manager import OpeningStatementConfigManager
 from core.app.app_config.features.retrieval_resource.manager import RetrievalResourceConfigManager
 from core.app.app_config.features.speech_to_text.manager import SpeechToTextConfigManager
-from core.app.app_config.features.suggested_questions_after_answer.manager import \
-    SuggestedQuestionsAfterAnswerConfigManager
+from core.app.app_config.features.suggested_questions_after_answer.manager import (
+    SuggestedQuestionsAfterAnswerConfigManager,
+)
 from core.app.app_config.features.text_to_speech.manager import TextToSpeechConfigManager
 from core.app.app_config.workflow_ui_based_app.variables.manager import WorkflowVariablesConfigManager
-from models.model import AppMode, App
+from models.model import App, AppMode
 from models.workflow import Workflow
 
 

api/core/app/apps/agent_chat/app_config_manager.py

@@ -3,22 +3,23 @@ from typing import Optional
 
 from core.agent.entities import AgentEntity
 from core.app.app_config.base_app_config_manager import BaseAppConfigManager
+from core.app.app_config.common.sensitive_word_avoidance.manager import SensitiveWordAvoidanceConfigManager
 from core.app.app_config.easy_ui_based_app.agent.manager import AgentConfigManager
 from core.app.app_config.easy_ui_based_app.dataset.manager import DatasetConfigManager
 from core.app.app_config.easy_ui_based_app.model_config.manager import ModelConfigManager
 from core.app.app_config.easy_ui_based_app.prompt_template.manager import PromptTemplateConfigManager
 from core.app.app_config.easy_ui_based_app.variables.manager import BasicVariablesConfigManager
-from core.app.app_config.common.sensitive_word_avoidance.manager import SensitiveWordAvoidanceConfigManager
-from core.app.app_config.entities import EasyUIBasedAppConfig, EasyUIBasedAppModelConfigFrom, DatasetEntity
+from core.app.app_config.entities import EasyUIBasedAppConfig, EasyUIBasedAppModelConfigFrom
 from core.app.app_config.features.file_upload.manager import FileUploadConfigManager
 from core.app.app_config.features.opening_statement.manager import OpeningStatementConfigManager
 from core.app.app_config.features.retrieval_resource.manager import RetrievalResourceConfigManager
 from core.app.app_config.features.speech_to_text.manager import SpeechToTextConfigManager
-from core.app.app_config.features.suggested_questions_after_answer.manager import \
-    SuggestedQuestionsAfterAnswerConfigManager
+from core.app.app_config.features.suggested_questions_after_answer.manager import (
+    SuggestedQuestionsAfterAnswerConfigManager,
+)
 from core.app.app_config.features.text_to_speech.manager import TextToSpeechConfigManager
 from core.entities.agent_entities import PlanningStrategy
-from models.model import AppMode, App, AppModelConfig
+from models.model import App, AppMode, AppModelConfig
 
 OLD_TOOLS = ["dataset", "google_search", "web_reader", "wikipedia", "current_datetime"]
 

api/core/app/apps/base_app_runner.py

@@ -2,14 +2,15 @@ import time
 from collections.abc import Generator
 from typing import Optional, Union, cast
 
-from core.app.app_config.entities import PromptTemplateEntity, ExternalDataVariableEntity
+from core.app.app_config.entities import ExternalDataVariableEntity, PromptTemplateEntity
 from core.app.app_queue_manager import AppQueueManager, PublishFrom
-from core.app.features.annotation_reply.annotation_reply import AnnotationReplyFeature
-from core.app.features.hosting_moderation.hosting_moderation import HostingModerationFeature
 from core.app.entities.app_invoke_entities import (
     EasyUIBasedAppGenerateEntity,
-    InvokeFrom, EasyUIBasedModelConfigEntity,
+    EasyUIBasedModelConfigEntity,
+    InvokeFrom,
 )
+from core.app.features.annotation_reply.annotation_reply import AnnotationReplyFeature
+from core.app.features.hosting_moderation.hosting_moderation import HostingModerationFeature
 from core.external_data_tool.external_data_fetch import ExternalDataFetch
 from core.file.file_obj import FileObj
 from core.memory.token_buffer_memory import TokenBufferMemory

api/core/app/apps/chat/app_config_manager.py

@@ -1,20 +1,21 @@
 from typing import Optional
 
 from core.app.app_config.base_app_config_manager import BaseAppConfigManager
+from core.app.app_config.common.sensitive_word_avoidance.manager import SensitiveWordAvoidanceConfigManager
 from core.app.app_config.easy_ui_based_app.dataset.manager import DatasetConfigManager
 from core.app.app_config.easy_ui_based_app.model_config.manager import ModelConfigManager
 from core.app.app_config.easy_ui_based_app.prompt_template.manager import PromptTemplateConfigManager
 from core.app.app_config.easy_ui_based_app.variables.manager import BasicVariablesConfigManager
-from core.app.app_config.common.sensitive_word_avoidance.manager import SensitiveWordAvoidanceConfigManager
 from core.app.app_config.entities import EasyUIBasedAppConfig, EasyUIBasedAppModelConfigFrom
 from core.app.app_config.features.file_upload.manager import FileUploadConfigManager
 from core.app.app_config.features.opening_statement.manager import OpeningStatementConfigManager
 from core.app.app_config.features.retrieval_resource.manager import RetrievalResourceConfigManager
 from core.app.app_config.features.speech_to_text.manager import SpeechToTextConfigManager
-from core.app.app_config.features.suggested_questions_after_answer.manager import \
-    SuggestedQuestionsAfterAnswerConfigManager
+from core.app.app_config.features.suggested_questions_after_answer.manager import (
+    SuggestedQuestionsAfterAnswerConfigManager,
+)
 from core.app.app_config.features.text_to_speech.manager import TextToSpeechConfigManager
-from models.model import AppMode, App, AppModelConfig
+from models.model import App, AppMode, AppModelConfig
 
 
 class ChatAppConfig(EasyUIBasedAppConfig):

api/core/app/apps/chat/app_runner.py

@@ -2,12 +2,12 @@ import logging
 from typing import cast
 
 from core.app.app_queue_manager import AppQueueManager, PublishFrom
-from core.app.apps.chat.app_config_manager import ChatAppConfig
 from core.app.apps.base_app_runner import AppRunner
-from core.callback_handler.index_tool_callback_handler import DatasetIndexToolCallbackHandler
+from core.app.apps.chat.app_config_manager import ChatAppConfig
 from core.app.entities.app_invoke_entities import (
     EasyUIBasedAppGenerateEntity,
 )
+from core.callback_handler.index_tool_callback_handler import DatasetIndexToolCallbackHandler
 from core.memory.token_buffer_memory import TokenBufferMemory
 from core.model_manager import ModelInstance
 from core.moderation.base import ModerationException

api/core/app/apps/completion/app_config_manager.py

@@ -1,16 +1,16 @@
 from typing import Optional
 
 from core.app.app_config.base_app_config_manager import BaseAppConfigManager
+from core.app.app_config.common.sensitive_word_avoidance.manager import SensitiveWordAvoidanceConfigManager
 from core.app.app_config.easy_ui_based_app.dataset.manager import DatasetConfigManager
 from core.app.app_config.easy_ui_based_app.model_config.manager import ModelConfigManager
 from core.app.app_config.easy_ui_based_app.prompt_template.manager import PromptTemplateConfigManager
 from core.app.app_config.easy_ui_based_app.variables.manager import BasicVariablesConfigManager
-from core.app.app_config.common.sensitive_word_avoidance.manager import SensitiveWordAvoidanceConfigManager
 from core.app.app_config.entities import EasyUIBasedAppConfig, EasyUIBasedAppModelConfigFrom
 from core.app.app_config.features.file_upload.manager import FileUploadConfigManager
 from core.app.app_config.features.more_like_this.manager import MoreLikeThisConfigManager
 from core.app.app_config.features.text_to_speech.manager import TextToSpeechConfigManager
-from models.model import AppMode, App, AppModelConfig
+from models.model import App, AppMode, AppModelConfig
 
 
 class CompletionAppConfig(EasyUIBasedAppConfig):

api/core/app/apps/completion/app_runner.py

@@ -2,12 +2,12 @@ import logging
 from typing import cast
 
 from core.app.app_queue_manager import AppQueueManager
-from core.app.apps.completion.app_config_manager import CompletionAppConfig
 from core.app.apps.base_app_runner import AppRunner
-from core.callback_handler.index_tool_callback_handler import DatasetIndexToolCallbackHandler
+from core.app.apps.completion.app_config_manager import CompletionAppConfig
 from core.app.entities.app_invoke_entities import (
     EasyUIBasedAppGenerateEntity,
 )
+from core.callback_handler.index_tool_callback_handler import DatasetIndexToolCallbackHandler
 from core.model_manager import ModelInstance
 from core.moderation.base import ModerationException
 from core.rag.retrieval.dataset_retrieval import DatasetRetrieval

api/core/app/apps/workflow/app_config_manager.py

@@ -4,7 +4,7 @@ from core.app.app_config.entities import WorkflowUIBasedAppConfig
 from core.app.app_config.features.file_upload.manager import FileUploadConfigManager
 from core.app.app_config.features.text_to_speech.manager import TextToSpeechConfigManager
 from core.app.app_config.workflow_ui_based_app.variables.manager import WorkflowVariablesConfigManager
-from models.model import AppMode, App
+from models.model import App, AppMode
 from models.workflow import Workflow