model_info

langroid/language_models/model_info.py

ModelProvider

Bases: str, Enum

Enum for model providers

ModelName

Bases: str, Enum

Parent class for all model name enums

OpenAIChatModel

Bases: ModelName

Enum for OpenAI Chat models

OpenAICompletionModel

Bases: str, Enum

Enum for OpenAI Completion models

AnthropicModel

Bases: ModelName

Enum for Anthropic models

DeepSeekModel

Bases: ModelName

Enum for DeepSeek models direct from DeepSeek API

GeminiModel

Bases: ModelName

Enum for Gemini models

OpenAI_API_ParamInfo

Bases: BaseModel

Parameters exclusive to some models, when using OpenAI API

ModelInfo

Bases: BaseModel

Consolidated information about LLM, related to capacity, cost and API idiosyncrasies. Reasonable defaults for all params in case there's no specific info available.

get_model_info(model, fallback_model='')

Get model information by name or enum value

Source code in langroid/language_models/model_info.py

def get_model_info(
    model: str | ModelName,
    fallback_model: str | ModelName = "",
) -> ModelInfo:
    """Get model information by name or enum value"""
    return _get_model_info(model) or _get_model_info(fallback_model) or ModelInfo()