Dinesh310 commited on
Commit
7479a8f
·
verified ·
1 Parent(s): 57ea49c

Create core/llm.py

Browse files
Files changed (1) hide show
  1. src/core/llm.py +15 -0
src/core/llm.py ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from langchain_openai import ChatOpenAI
2
+ from config.config import Config
3
+
4
+ def load_llm():
5
+ if not Config.OPENROUTER_API_KEY:
6
+ raise ValueError("OPENROUTER_API_KEY is not set")
7
+
8
+ try:
9
+ return ChatOpenAI(
10
+ model=Config.LLM_MODEL,
11
+ base_url=Config.OPENROUTER_BASE_URL,
12
+ api_key=Config.OPENROUTER_API_KEY
13
+ )
14
+ except Exception as e:
15
+ raise RuntimeError(f"Failed to initialize LLM: {e}")