dify/api/core/evaluation/entities/config_entity.py

from enum import StrEnum

from pydantic import BaseModel


class EvaluationFrameworkEnum(StrEnum):
    RAGAS = "ragas"
    DEEPEVAL = "deepeval"
    NONE = "none"


class BaseEvaluationConfig(BaseModel):
    """Base configuration for evaluation frameworks."""

    pass


class RagasConfig(BaseEvaluationConfig):
    """RAGAS-specific configuration."""

    pass


class DeepEvalConfig(BaseEvaluationConfig):
    """DeepEval-specific configuration."""

    pass