Skip to content

constants

mlflow_assistant.utils.constants

Constants and enumerations for MLflow Assistant.

This module defines configuration keys, default values, API endpoints, model definitions, and other constants used throughout MLflow Assistant. It includes enums for AI providers (OpenAI, Ollama) and their supported models.

Command

Bases: Enum

Special commands for interactive chat sessions.

description property

Get the description for a command.

DatabricksModel

Bases: Enum

Databricks models supported by MLflow Assistant.

choices() classmethod

Get all available Databricks model choices.

Source code in src/mlflow_assistant/utils/constants.py 
92
93
94
95
@classmethod
def choices(cls):
    """Get all available Databricks model choices."""
    return [model.value for model in cls]

OllamaModel

Bases: Enum

Default Ollama models supported by MLflow Assistant.

choices() classmethod

Get all available Ollama model choices.

Source code in src/mlflow_assistant/utils/constants.py 
80
81
82
83
@classmethod
def choices(cls):
    """Get all available Ollama model choices."""
    return [model.value for model in cls]

OpenAIModel

Bases: Enum

OpenAI models supported by MLflow Assistant.

choices() classmethod

Get all available OpenAI model choices.

Source code in src/mlflow_assistant/utils/constants.py 
66
67
68
69
@classmethod
def choices(cls):
    """Get all available OpenAI model choices."""
    return [model.value for model in cls]

Provider

Bases: Enum

AI provider types supported by MLflow Assistant.

get_default_model(provider) classmethod

Get the default model for a provider.

Source code in src/mlflow_assistant/utils/constants.py 
106
107
108
109
110
111
112
113
114
@classmethod
def get_default_model(cls, provider):
    """Get the default model for a provider."""
    defaults = {
        cls.OPENAI: OpenAIModel.GPT35.value,
        cls.OLLAMA: OllamaModel.LLAMA32.value,
        cls.DATABRICKS: DatabricksModel.DATABRICKS_META_LLAMA3.value,
    }
    return defaults.get(provider)

get_default_temperature(provider) classmethod

Get the default temperature for a provider.

Source code in src/mlflow_assistant/utils/constants.py 
116
117
118
119
120
121
122
123
124
@classmethod
def get_default_temperature(cls, provider):
    """Get the default temperature for a provider."""
    defaults = {
        cls.OPENAI: 0.7,
        cls.DATABRICKS: 0.7,
        cls.OLLAMA: 0.7,
    }
    return defaults.get(provider)