# src/llm.py
from langchain_openai import ChatOpenAI
from src.config.config import (
    LLM_MODEL,
    OPENROUTER_BASE_URL,
    OPENROUTER_API_KEY
)

def load_llm():
    if not OPENROUTER_API_KEY:
        raise EnvironmentError("OPENROUTER_API_KEY not set")

    return ChatOpenAI(
        model=LLM_MODEL,
        base_url=OPENROUTER_BASE_URL,
        api_key=OPENROUTER_API_KEY
    )