config/models.py

from langchain.chat_models import init_chat_model

def get_default_model():
    """Get the default chat model"""
    return init_chat_model(
        "gemini-2.5-flash",
        model_provider="google_genai",
    )

def get_model(model_name: str, provider: str):
    """Get a specific model"""
    return init_chat_model(model_name, model_provider=provider)