Dinesh310's picture
Upload 16 files
0fe6232 verified
raw
history blame contribute delete
404 Bytes
# 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
)