# This file was auto-generated by Fern from our API Definition.

import enum
import typing

T_Result = typing.TypeVar("T_Result")


class SelfHostedModelModelServer(str, enum.Enum):
    """
    +sort=4
    +label=Model Server Type
    +usage=The type of model server being used
    """

    VLLM_OPENAI = "vllm-openai"
    TEI = "tei"
    INFINITY = "infinity"
    NEMO_RETRIEVER = "nemo-retriever"

    def visit(
        self,
        vllm_openai: typing.Callable[[], T_Result],
        tei: typing.Callable[[], T_Result],
        infinity: typing.Callable[[], T_Result],
        nemo_retriever: typing.Callable[[], T_Result],
    ) -> T_Result:
        if self is SelfHostedModelModelServer.VLLM_OPENAI:
            return vllm_openai()
        if self is SelfHostedModelModelServer.TEI:
            return tei()
        if self is SelfHostedModelModelServer.INFINITY:
            return infinity()
        if self is SelfHostedModelModelServer.NEMO_RETRIEVER:
            return nemo_retriever()
