Source code for ai.analysis.model.assistant_chat_model

import enum



[docs]
class AssistantChatModel(enum.Enum):
    """Enum of supported assistant chat model identifiers.

    Attributes:
        GPT_4O (str): Identifier for the full GPT-4O model.
        GPT_4_1 (str): Identifier for the GPT-4.1 model.
        GPT_4_1_MINI (str): Identifier for the GPT-4.1 Mini model.
        GPT_4_1_NANO (str): Identifier for the GPT-4.1 Nano model.
        GPT_4O_MINI (str): Identifier for the GPT-4O Mini model.
        QWEN_2_5_7B (str): Identifier for the Qwen 2.5-7B Instruct model.
        QWEN_2_5_7B_TOGETHER (str): Identifier for the Qwen 2.5-7B Instruct Turbo model.
        LLAMA_70B (str): Identifier for the Meta LLaMA-3.3-70B Instruct model.
        LLAMA_3B (str): Identifier for the Meta LLaMA-3.2-3B Instruct model.
    """

    GPT_4O = "gpt-4o"
    GPT_4_1 = "gpt-4.1"
    GPT_4_1_MINI = "gpt-4.1-mini"
    GPT_4_1_NANO = "gpt-4.1-nano"
    GPT_4O_MINI = "gpt-4o-mini"
    QWEN_2_5_7B = "qwen/qwen-2.5-7b-instruct"
    QWEN_2_5_7B_TOGETHER = "Qwen/Qwen2.5-7B-Instruct-Turbo"
    DEEP_SEEK_R1 = "deepseek/deepseek-r1"
    LLAMA_70B = "meta-llama/llama-3.3-70b-instruct"
    LLAMA_3B = "meta-llama/llama-3.2-3b-instruct"