Spaces:
Sleeping
Sleeping
File size: 404 Bytes
0fe6232 |
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
)
|