Commit e80c67ad authored by takatost's avatar takatost

lint

parent 3114e0b9
...@@ -10,7 +10,7 @@ from controllers.console.wraps import account_initialization_required ...@@ -10,7 +10,7 @@ from controllers.console.wraps import account_initialization_required
from events.app_event import app_model_config_was_updated from events.app_event import app_model_config_was_updated
from extensions.ext_database import db from extensions.ext_database import db
from libs.login import login_required from libs.login import login_required
from models.model import AppModelConfig, AppMode from models.model import AppMode, AppModelConfig
from services.app_model_config_service import AppModelConfigService from services.app_model_config_service import AppModelConfigService
......
import uuid import uuid
from typing import Tuple
from core.agent.agent_executor import PlanningStrategy from core.agent.agent_executor import PlanningStrategy
from core.apps.config_validators.dataset import DatasetValidator from core.apps.config_validators.dataset import DatasetValidator
...@@ -9,7 +8,7 @@ OLD_TOOLS = ["dataset", "google_search", "web_reader", "wikipedia", "current_dat ...@@ -9,7 +8,7 @@ OLD_TOOLS = ["dataset", "google_search", "web_reader", "wikipedia", "current_dat
class AgentValidator: class AgentValidator:
@classmethod @classmethod
def validate_and_set_defaults(cls, tenant_id: str, config: dict) -> Tuple[dict, list[str]]: def validate_and_set_defaults(cls, tenant_id: str, config: dict) -> tuple[dict, list[str]]:
""" """
Validate and set defaults for agent feature Validate and set defaults for agent feature
......
import uuid import uuid
from typing import Tuple
from core.agent.agent_executor import PlanningStrategy from core.agent.agent_executor import PlanningStrategy
from models.model import AppMode from models.model import AppMode
...@@ -8,7 +7,7 @@ from services.dataset_service import DatasetService ...@@ -8,7 +7,7 @@ from services.dataset_service import DatasetService
class DatasetValidator: class DatasetValidator:
@classmethod @classmethod
def validate_and_set_defaults(cls, tenant_id: str, app_mode: AppMode, config: dict) -> Tuple[dict, list[str]]: def validate_and_set_defaults(cls, tenant_id: str, app_mode: AppMode, config: dict) -> tuple[dict, list[str]]:
""" """
Validate and set defaults for dataset feature Validate and set defaults for dataset feature
......
from typing import Tuple
from core.external_data_tool.factory import ExternalDataToolFactory from core.external_data_tool.factory import ExternalDataToolFactory
class ExternalDataToolsValidator: class ExternalDataToolsValidator:
@classmethod @classmethod
def validate_and_set_defaults(cls, tenant_id: str, config: dict) -> Tuple[dict, list[str]]: def validate_and_set_defaults(cls, tenant_id: str, config: dict) -> tuple[dict, list[str]]:
""" """
Validate and set defaults for external data fetch feature Validate and set defaults for external data fetch feature
......
from typing import Tuple
class FileUploadValidator: class FileUploadValidator:
@classmethod @classmethod
def validate_and_set_defaults(cls, config: dict) -> Tuple[dict, list[str]]: def validate_and_set_defaults(cls, config: dict) -> tuple[dict, list[str]]:
""" """
Validate and set defaults for file upload feature Validate and set defaults for file upload feature
......
from typing import Tuple
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.model_runtime.model_providers import model_provider_factory
from core.provider_manager import ProviderManager from core.provider_manager import ProviderManager
class ModelValidator: class ModelValidator:
@classmethod @classmethod
def validate_and_set_defaults(cls, tenant_id: str, config: dict) -> Tuple[dict, list[str]]: def validate_and_set_defaults(cls, tenant_id: str, config: dict) -> tuple[dict, list[str]]:
""" """
Validate and set defaults for model config Validate and set defaults for model config
......
import logging import logging
from typing import Tuple
from core.moderation.factory import ModerationFactory from core.moderation.factory import ModerationFactory
...@@ -8,7 +7,7 @@ logger = logging.getLogger(__name__) ...@@ -8,7 +7,7 @@ logger = logging.getLogger(__name__)
class ModerationValidator: class ModerationValidator:
@classmethod @classmethod
def validate_and_set_defaults(cls, tenant_id, config: dict) -> Tuple[dict, list[str]]: def validate_and_set_defaults(cls, tenant_id, config: dict) -> tuple[dict, list[str]]:
if not config.get("sensitive_word_avoidance"): if not config.get("sensitive_word_avoidance"):
config["sensitive_word_avoidance"] = { config["sensitive_word_avoidance"] = {
"enabled": False "enabled": False
......
from typing import Tuple
class MoreLikeThisValidator: class MoreLikeThisValidator:
@classmethod @classmethod
def validate_and_set_defaults(cls, config: dict) -> Tuple[dict, list[str]]: def validate_and_set_defaults(cls, config: dict) -> tuple[dict, list[str]]:
""" """
Validate and set defaults for more like this feature Validate and set defaults for more like this feature
......
from typing import Tuple
class OpeningStatementValidator: class OpeningStatementValidator:
@classmethod @classmethod
def validate_and_set_defaults(cls, config: dict) -> Tuple[dict, list[str]]: def validate_and_set_defaults(cls, config: dict) -> tuple[dict, list[str]]:
""" """
Validate and set defaults for opening statement feature Validate and set defaults for opening statement feature
......
from typing import Tuple
from core.entities.application_entities import PromptTemplateEntity from core.entities.application_entities import PromptTemplateEntity
from core.prompt.simple_prompt_transform import ModelMode from core.prompt.simple_prompt_transform import ModelMode
...@@ -7,7 +6,7 @@ from models.model import AppMode ...@@ -7,7 +6,7 @@ from models.model import AppMode
class PromptValidator: class PromptValidator:
@classmethod @classmethod
def validate_and_set_defaults(cls, app_mode: AppMode, config: dict) -> Tuple[dict, list[str]]: def validate_and_set_defaults(cls, app_mode: AppMode, config: dict) -> tuple[dict, list[str]]:
""" """
Validate pre_prompt and set defaults for prompt feature Validate pre_prompt and set defaults for prompt feature
depending on the config['model'] depending on the config['model']
......
from typing import Tuple
class RetrieverResourceValidator: class RetrieverResourceValidator:
@classmethod @classmethod
def validate_and_set_defaults(cls, config: dict) -> Tuple[dict, list[str]]: def validate_and_set_defaults(cls, config: dict) -> tuple[dict, list[str]]:
""" """
Validate and set defaults for retriever resource feature Validate and set defaults for retriever resource feature
......
from typing import Tuple
class SpeechToTextValidator: class SpeechToTextValidator:
@classmethod @classmethod
def validate_and_set_defaults(cls, config: dict) -> Tuple[dict, list[str]]: def validate_and_set_defaults(cls, config: dict) -> tuple[dict, list[str]]:
""" """
Validate and set defaults for speech to text feature Validate and set defaults for speech to text feature
......
from typing import Tuple
class SuggestedQuestionsValidator: class SuggestedQuestionsValidator:
@classmethod @classmethod
def validate_and_set_defaults(cls, config: dict) -> Tuple[dict, list[str]]: def validate_and_set_defaults(cls, config: dict) -> tuple[dict, list[str]]:
""" """
Validate and set defaults for suggested questions feature Validate and set defaults for suggested questions feature
......
from typing import Tuple
class TextToSpeechValidator: class TextToSpeechValidator:
@classmethod @classmethod
def validate_and_set_defaults(cls, config: dict) -> Tuple[dict, list[str]]: def validate_and_set_defaults(cls, config: dict) -> tuple[dict, list[str]]:
""" """
Validate and set defaults for text to speech feature Validate and set defaults for text to speech feature
......
import re import re
from typing import Tuple
class UserInputFormValidator: class UserInputFormValidator:
@classmethod @classmethod
def validate_and_set_defaults(cls, config: dict) -> Tuple[dict, list[str]]: def validate_and_set_defaults(cls, config: dict) -> tuple[dict, list[str]]:
""" """
Validate and set defaults for user input form Validate and set defaults for user input form
......
...@@ -9,7 +9,7 @@ from core.apps.config_validators.model import ModelValidator ...@@ -9,7 +9,7 @@ from core.apps.config_validators.model import ModelValidator
from core.entities.application_entities import InvokeFrom from core.entities.application_entities import InvokeFrom
from core.file.message_file_parser import MessageFileParser from core.file.message_file_parser import MessageFileParser
from extensions.ext_database import db from extensions.ext_database import db
from models.model import Account, App, AppModelConfig, Conversation, EndUser, Message, AppMode from models.model import Account, App, AppMode, AppModelConfig, Conversation, EndUser, Message
from services.app_model_config_service import AppModelConfigService from services.app_model_config_service import AppModelConfigService
from services.errors.app import MoreLikeThisDisabledError from services.errors.app import MoreLikeThisDisabledError
from services.errors.app_model_config import AppModelConfigBrokenError from services.errors.app_model_config import AppModelConfigBrokenError
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment