Chat_engine_configuration

class BaseChatEngineConfiguration(BaseConfiguration):
    """Base configuration class for chat engines.

    This class defines the standard configuration structure for all chat engines,
    including retriever, llm, and postprocessor components.
    """

    retriever: Any = Field(
        ...,
        description="The retriever configuration for the augmentation pipeline.",
    )
    llm: Any = Field(
        ..., description="The llm configuration for the chat engine."
    )
    postprocessors: List[Any] = Field(
        ..., description="The list of postprocessors for the chat engine."
    )
    prompt_templates: ChatEnginePromptTemplates = Field(
        ...,
        description="The prompt templates configuration for the chat engine.",
        default_factory=ChatEnginePromptTemplates,
    )

    @field_validator("retriever")
    @classmethod
    def _validate_retriever(cls, value: Any, info: ValidationInfo) -> Any:
        """Validate retriever configuration against registered retriever types.

        Args:
            value: The retriever configuration to validate
            info: Validation context information

        Returns:
            Validated retriever configuration
        """
        return super()._validate(
            value,
            info=info,
            registry=RetrieverConfigurationRegistry,
        )

    @field_validator("llm")
    @classmethod
    def _validate_llm(cls, value: Any, info: ValidationInfo) -> Any:
        """Validate llm configuration against registered llm types.

        Args:
            value: The llm configuration to validate
            info: Validation context information

        Returns:
            Validated llm configuration
        """
        return super()._validate(
            value,
            info=info,
            registry=LLMConfigurationRegistry,
        )

    @field_validator("postprocessors")
    @classmethod
    def _validate_postprocessors(
        cls, value: Any, info: ValidationInfo
    ) -> List[Any]:
        """Validate postprocessors configuration against registered postprocessor types.

        Args:
            value: The postprocessor configurations to validate
            info: Validation context information

        Returns:
            List of validated postprocessor configurations
        """
        return super()._validate(
            value,
            info=info,
            registry=PostProcessorConfigurationRegistry,
        )

class ChatEngineConfigurationRegistry(ConfigurationRegistry):
    """Registry for chat engine configurations.

    Maps chat engine names to their corresponding configuration classes
    to facilitate configuration validation and factory creation.

    Attributes:
        _key_class: The enumeration class for chat engine names.
    """

    _key_class: Type = ChatEngineName

class ChatEngineName(str, Enum):
    """Enum defining available chat engine types."""

    LANGFUSE = "langfuse"