File size: 387 Bytes
4731a7f 7479a8f 4731a7f 7479a8f 4731a7f 7479a8f 4731a7f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
# 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
)
|