KrishnaHuYaar commited on
Commit
c3dcd36
·
verified ·
1 Parent(s): 1645e94

Update llm.py

Browse files
Files changed (1) hide show
  1. llm.py +5 -4
llm.py CHANGED
@@ -3,12 +3,13 @@ from langchain_openai import ChatOpenAI
3
 
4
  def get_llm():
5
  return ChatOpenAI(
6
- model=os.getenv(
7
- "OPENROUTER_MODEL",
8
- "meta-llama/llama-3-8b-instruct"
9
- ),
10
  api_key=os.getenv("OPENROUTER_API_KEY"),
11
  base_url="https://openrouter.ai/api/v1",
 
 
 
 
12
  temperature=0.4,
13
  max_tokens=700
14
  )
 
3
 
4
  def get_llm():
5
  return ChatOpenAI(
6
+ model="meta-llama/llama-3-8b-instruct",
 
 
 
7
  api_key=os.getenv("OPENROUTER_API_KEY"),
8
  base_url="https://openrouter.ai/api/v1",
9
+ default_headers={
10
+ "HTTP-Referer": "https://huggingface.co",
11
+ "X-Title": "Taskify AI"
12
+ },
13
  temperature=0.4,
14
  max_tokens=700
15
  )