ai-agent / llm.py
NithikaShree's picture
Update llm.py
ae7b495 verified
raw
history blame contribute delete
283 Bytes
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
)