Update src/core/llm.py

src/core/llm.py

@@ -1,15 +1,17 @@
+# src/llm.py
 from langchain_openai import ChatOpenAI
-from src.config.config import Config
+from src.config.config import (
+    LLM_MODEL,
+    OPENROUTER_BASE_URL,
+    OPENROUTER_API_KEY
+)
 
 def load_llm():
-    if not Config.OPENROUTER_API_KEY:
-        raise ValueError("OPENROUTER_API_KEY is not set")
+    if not OPENROUTER_API_KEY:
+        raise EnvironmentError("OPENROUTER_API_KEY not set")
 
-    try:
-        return ChatOpenAI(
-            model=Config.LLM_MODEL,
-            base_url=Config.OPENROUTER_BASE_URL,
-            api_key=Config.OPENROUTER_API_KEY
-        )
-    except Exception as e:
-        raise RuntimeError(f"Failed to initialize LLM: {e}")
+    return ChatOpenAI(
+        model=LLM_MODEL,
+        base_url=OPENROUTER_BASE_URL,
+        api_key=OPENROUTER_API_KEY
+    )