def _get_model()

in 0_basic-agent/LangGraph/my_agent/utils/nodes.py [0:0]

• 11 lines of code
• 4 McCabe index (conditional complexity)

def _get_model(model_name: str):
    if model_name == "azureopenai":
        model = AzureChatOpenAI(temperature=0, model_name="gpt-4o-mini")
    elif model_name == "openai":
        model = ChatOpenAI(temperature=0, model_name="gpt-4o-mini")
    elif model_name == "anthropic":
        model = ChatAnthropic(temperature=0, model_name="claude-3-sonnet-20240229")
    else:
        raise ValueError(f"Unsupported model type: {model_name}")

    model = model.bind_tools(tools)
    return model