cyberosa commited on
Commit
f136769
·
1 Parent(s): 17e1495

replacing correct model for openrouter models

Browse files
Files changed (1) hide show
  1. agents.py +4 -8
agents.py CHANGED
@@ -1,5 +1,6 @@
1
  from smolagents import (
2
  LiteLLMModel,
 
3
  ToolCallingAgent,
4
  CodeAgent,
5
  VisitWebpageTool,
@@ -15,15 +16,10 @@ gemini_model = LiteLLMModel(
15
  model_id="gemini/gemini-2.0-flash-lite",
16
  api_key=gemini_api_key,
17
  )
18
- model = LiteLLMModel(
19
- #model_id="deepseek/deepseek-r1-0528-qwen3-8b:free",
20
- model_id="openrouter/google/gemini-2.5-flash-preview-05-20",
21
- api_base="https://openrouter.ai/api/v1",
22
  api_key=open_router_api_key,
23
- headers={
24
- "HTTP-Referer": "https://huggingface.co/spaces/cyberosa/SkynetAgent", # Replace with your site if deployed
25
- "X-Title": "SkynetMultiAgent"
26
- }
27
  )
28
 
29
 
 
1
  from smolagents import (
2
  LiteLLMModel,
3
+ OpenAIServerModel,
4
  ToolCallingAgent,
5
  CodeAgent,
6
  VisitWebpageTool,
 
16
  model_id="gemini/gemini-2.0-flash-lite",
17
  api_key=gemini_api_key,
18
  )
19
+ model = OpenAIServerModel(
20
+ model_id='deepseek/deepseek-r1-0528-qwen3-8b:free',
21
+ api_base='https://openrouter.ai/api/v1',
 
22
  api_key=open_router_api_key,
 
 
 
 
23
  )
24
 
25