Function get_model_info

langroid/language_models/model_info.py:769–800

Get model information by name or enum value

(
    model: str | ModelName,
    fallback_models: List[str] = [],
)

def get_model_info(
    model: str | ModelName,
    fallback_models: List[str] = [],
) -> ModelInfo:
    """Get model information by name or enum value"""
    # Sequence of models to try, starting with the primary model
    models_to_try = [model] + fallback_models

    # Find the first model in the sequence that has info defined using next()
    # on a generator expression that filters out None results from _get_model_info
    found_info = next(
        (info for m in models_to_try if (info := _get_model_info(m)) is not None),
        None,  # Default value if the iterator is exhausted (no valid info found)
    )

    if found_info is not None:
        return found_info

    normalized_models = _normalize_model_names(models_to_try)
    found_info = next(
        (
            info
            for normalized_model in normalized_models
            if (info := _get_model_info(normalized_model)) is not None
        ),
        None,
    )
    if found_info is not None:
        return found_info

    _warn_unknown_model(models_to_try)
    return ModelInfo()


def _get_model_info(model: str | ModelName) -> ModelInfo | None: