Skip to content

datarobot_genai.core.model_info

model_info

litellm.get_model_info for datarobot/-prefixed gateway models.

get_model_info 

get_model_info(model: str) -> ModelInfo

litellm.get_model_info, resolving the datarobot/ prefix and azure names.

Source code in datarobot_genai/core/model_info.py 
26
27
28
29
30
31
32
33
34
def get_model_info(model: str) -> ModelInfo:
    """``litellm.get_model_info``, resolving the ``datarobot/`` prefix and azure names."""
    import litellm  # noqa: PLC0415 — lazy so importing this module stays cheap

    if model.startswith("datarobot/"):
        model = model.removeprefix("datarobot/")
        if model.startswith("azure/"):
            model = re.sub(r"(gpt-\d)-(\d)(?=-|$)", r"\1.\2", model)  # gpt-5-1 -> gpt-5.1
    return litellm.get_model_info(model)