Skip to content

definitions

mlflow_assistant.providers.definitions

Constants for the MLflow Assistant providers.

ParameterKeys

Keys and default parameter groupings for supported providers.

get_parameters(provider) classmethod

Return the list of parameters for the given provider name.

Source code in src/mlflow_assistant/providers/definitions.py 
31
32
33
34
35
36
37
38
39
@classmethod
def get_parameters(cls, provider: str) -> list[str]:
    """Return the list of parameters for the given provider name."""
    provider_map = {
        "openai": cls.PARAMETERS_OPENAI,
        "ollama": cls.PARAMETERS_OLLAMA,
        "databricks": cls.PARAMETERS_DATABRICKS,
    }
    return provider_map.get(provider.lower(), [])