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

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)

Get model information by name or enum value

Source code in langroid/language_models/model_info.py

def get_model_info(model: str | ModelName) -> ModelInfo:
    """Get model information by name or enum value"""
    if isinstance(model, str):
        return MODEL_INFO.get(model) or ModelInfo()
    return MODEL_INFO.get(model.value) or ModelInfo()