Update llm.py

llm.py

@@ -1,9 +1,13 @@
+import os
 from langchain_cerebras import ChatCerebras
-from dotenv import load_dotenv
 
-load_dotenv()
+CEREBRAS_API_KEY = os.environ.get("CEREBRAS_API_KEY")
+
+if not CEREBRAS_API_KEY:
+    raise ValueError("CEREBRAS_API_KEY not found")
 
 llm = ChatCerebras(
-    model="gpt-oss-120b",
-    temperature=0.7
+    model="llama3.1-8b",
+    temperature=0.7,
+    api_key=CEREBRAS_API_KEY
 )