SheikhIrtiza commited on
Commit
477164e
·
verified ·
1 Parent(s): 9419a63

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +44 -5
app.py CHANGED
@@ -1,3 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import yaml
2
  import os
3
  from smolagents import GradioUI, CodeAgent, HfApiModel
@@ -12,6 +54,7 @@ from tools.final_answer import FinalAnswerTool as FinalAnswer
12
 
13
  model = HfApiModel(
14
  model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
 
15
  )
16
 
17
  get_weather_data = GetWeatherData()
@@ -25,16 +68,12 @@ agent = CodeAgent(
25
  model=model,
26
  tools=[get_weather_data],
27
  managed_agents=[],
28
- class='CodeAgent',
29
- max_steps=20,
30
  verbosity_level=2,
31
  grammar=None,
32
  planning_interval=None,
33
  name=None,
34
  description=None,
35
- executor_type='local',
36
- executor_kwargs={},
37
- max_print_outputs_length=None,
38
  prompt_templates=prompt_templates
39
  )
40
  if __name__ == "__main__":
 
1
+ # import yaml
2
+ # import os
3
+ # from smolagents import GradioUI, CodeAgent, HfApiModel
4
+
5
+ # # Get current directory path
6
+ # CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
7
+
8
+ # from tools.get_weather_data import SimpleTool as GetWeatherData
9
+ # from tools.final_answer import FinalAnswerTool as FinalAnswer
10
+
11
+
12
+
13
+ # model = HfApiModel(
14
+ # model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
15
+ # )
16
+
17
+ # get_weather_data = GetWeatherData()
18
+ # final_answer = FinalAnswer()
19
+
20
+
21
+ # with open(os.path.join(CURRENT_DIR, "prompts.yaml"), 'r') as stream:
22
+ # prompt_templates = yaml.safe_load(stream)
23
+
24
+ # agent = CodeAgent(
25
+ # model=model,
26
+ # tools=[get_weather_data],
27
+ # managed_agents=[],
28
+ # class='CodeAgent',
29
+ # max_steps=20,
30
+ # verbosity_level=2,
31
+ # grammar=None,
32
+ # planning_interval=None,
33
+ # name=None,
34
+ # description=None,
35
+ # executor_type='local',
36
+ # executor_kwargs={},
37
+ # max_print_outputs_length=None,
38
+ # prompt_templates=prompt_templates
39
+ # )
40
+ # if __name__ == "__main__":
41
+ # GradioUI(agent).launch()
42
+
43
  import yaml
44
  import os
45
  from smolagents import GradioUI, CodeAgent, HfApiModel
 
54
 
55
  model = HfApiModel(
56
  model_id='Qwen/Qwen2.5-Coder-32B-Instruct',
57
+ provider=None,
58
  )
59
 
60
  get_weather_data = GetWeatherData()
 
68
  model=model,
69
  tools=[get_weather_data],
70
  managed_agents=[],
71
+ max_steps=6,
 
72
  verbosity_level=2,
73
  grammar=None,
74
  planning_interval=None,
75
  name=None,
76
  description=None,
 
 
 
77
  prompt_templates=prompt_templates
78
  )
79
  if __name__ == "__main__":