llm.py

import os
from langchain_cerebras import ChatCerebras

CEREBRAS_API_KEY = os.environ.get("CEREBRAS_API_KEY")

if not CEREBRAS_API_KEY:
    raise ValueError("CEREBRAS_API_KEY not set")

llm = ChatCerebras(
    model="gpt-oss-120b",
    temperature=0.6,
    api_key=CEREBRAS_API_KEY
)