dify/api/configs/remote_settings_sources/base.py

from collections.abc import Mapping
from typing import Any

from pydantic import AliasChoices
from pydantic.fields import FieldInfo


class RemoteSettingsSource:
    def __init__(self, configs: Mapping[str, Any]):
        pass

    def get_field_value(self, field: FieldInfo, field_name: str) -> tuple[Any, str, bool]:
        raise NotImplementedError

    @staticmethod
    def resolve_field_value(
        remote_configs: Mapping[str, Any], field: FieldInfo, field_name: str
    ) -> tuple[Any, str, bool]:
        field_value = remote_configs.get(field_name)
        if field_value is not None:
            return field_value, field_name, False

        validation_alias = field.validation_alias
        if isinstance(validation_alias, str):
            aliases = [validation_alias]
        elif isinstance(validation_alias, AliasChoices):
            aliases = [alias for alias in validation_alias.choices if isinstance(alias, str)]
        else:
            aliases = []

        for alias in aliases:
            field_value = remote_configs.get(alias)
            if field_value is not None:
                return field_value, alias, False

        return None, field_name, False

    def prepare_field_value(self, field_name: str, field: FieldInfo, value: Any, value_is_complex: bool):
        return value