ruaultadrienperso commited on
Commit
c1f2810
·
verified ·
1 Parent(s): 6ea6902

Upload agent

Browse files
Files changed (2) hide show
  1. agent.json +5 -2
  2. app.py +6 -3
agent.json CHANGED
@@ -6,9 +6,12 @@
6
  "final_answer"
7
  ],
8
  "model": {
9
- "class": "InferenceClientModel",
10
  "data": {
11
- "model_id": "Qwen/Qwen2.5-Coder-32B-Instruct"
 
 
 
12
  }
13
  },
14
  "managed_agents": {},
 
6
  "final_answer"
7
  ],
8
  "model": {
9
+ "class": "LiteLLMModel",
10
  "data": {
11
+ "temperature": 0.2,
12
+ "max_tokens": 10,
13
+ "model_id": "anthropic/claude-3-5-sonnet-latest",
14
+ "api_base": null
15
  }
16
  },
17
  "managed_agents": {},
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import yaml
2
  import os
3
- from smolagents import GradioUI, CodeAgent, InferenceClientModel
4
 
5
  # Get current directory path
6
  CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
@@ -11,8 +11,11 @@ from tools.final_answer import FinalAnswerTool as FinalAnswer
11
 
12
 
13
 
14
- model = InferenceClientModel(
15
- model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
 
 
 
16
  )
17
 
18
  web_search = WebSearch()
 
1
  import yaml
2
  import os
3
+ from smolagents import GradioUI, CodeAgent, LiteLLMModel
4
 
5
  # Get current directory path
6
  CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
 
11
 
12
 
13
 
14
+ model = LiteLLMModel(
15
+ temperature=0.2,
16
+ max_tokens=10,
17
+ model_id='anthropic/claude-3-5-sonnet-latest',
18
+ api_base=None,
19
  )
20
 
21
  web_search = WebSearch()