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 found")

llm = ChatCerebras(
    model="llama3.1-8b",
    temperature=0.7,
    api_key=CEREBRAS_API_KEY
)