def call_model()

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

• 7 lines of code
• 1 McCabe index (conditional complexity)

def call_model(state, config):
    messages = state["messages"]
    messages = [{"role": "system", "content": system_prompt}] + messages
    model_name = config.get("configurable", {}).get("model_name", "azureopenai")
    model = _get_model(model_name)
    response = model.invoke(messages)
    # We return a list, because this will get added to the existing list
    return {"messages": [response]}