Dinesh310 commited on
Commit
2fd7630
·
verified ·
1 Parent(s): af61802

Update src/rag_engine.py

Browse files
Files changed (1) hide show
  1. src/rag_engine.py +16 -9
src/rag_engine.py CHANGED
@@ -20,15 +20,22 @@ class ProjectRAGEngine:
20
 
21
  # ✅ OpenRouter LLM (Chat only)
22
  self.llm = ChatOpenAI(
23
- model="openai/gpt-oss-120b:free",
24
- temperature=0,
25
- openai_api_key=os.getenv("OPENROUTER_API_KEY"),
26
- openai_api_base="https://openrouter.ai/api/v1",
27
- default_headers={
28
- "HTTP-Referer": "http://localhost:8501",
29
- "X-Title": "Project-RAG-App"
30
- }
31
- )
 
 
 
 
 
 
 
32
 
33
  self.vector_store = None
34
 
 
20
 
21
  # ✅ OpenRouter LLM (Chat only)
22
  self.llm = ChatOpenAI(
23
+ model="openai/gpt-oss-120b:free",
24
+ temperature=0,
25
+
26
+ # ✅ CORRECT PARAM
27
+ api_key=os.getenv("OPENROUTER_API_KEY"),
28
+
29
+ # ✅ OpenRouter endpoint
30
+ base_url="https://openrouter.ai/api/v1",
31
+
32
+ # ✅ REQUIRED HEADERS FOR OPENROUTER
33
+ default_headers={
34
+ "Authorization": f"Bearer {os.getenv('OPENROUTER_API_KEY')}",
35
+ "HTTP-Referer": "http://localhost:8501",
36
+ "X-Title": "Project-RAG-App"
37
+ })
38
+
39
 
40
  self.vector_store = None
41