davemasino commited on
Commit
631cdcb
·
verified ·
1 Parent(s): 57586d5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -9
app.py CHANGED
@@ -16,8 +16,8 @@ from tools.final_answer import FinalAnswerTool as FinalAnswer
16
 
17
  model = LiteLLMModel(
18
  model_id='gemini/gemini-2.0-flash',
19
- api_key=os.getenv("GEMINI_API_KEY")
20
- #api_base=None,
21
  )
22
 
23
  web_search = WebSearch()
@@ -34,16 +34,16 @@ with open(os.path.join(CURRENT_DIR, "prompts.yaml"), 'r') as stream:
34
  agent = CodeAgent(
35
  model=model,
36
  tools=[web_search, visit_webpage, suggest_menu, catering_service_tool, superhero_party_theme_generator],
37
- # managed_agents=[],
38
  max_steps=10,
39
  verbosity_level=2,
40
- # grammar=None,
41
- # planning_interval=None,
42
  name="AlfredAgent",
43
- # description=None,
44
- # executor_type='local',
45
- # executor_kwargs={},
46
- # max_print_outputs_length=None,
47
  prompt_templates=prompt_templates
48
  )
49
  if __name__ == "__main__":
 
16
 
17
  model = LiteLLMModel(
18
  model_id='gemini/gemini-2.0-flash',
19
+ api_key=os.getenv("GEMINI_API_KEY"),
20
+ api_base=None,
21
  )
22
 
23
  web_search = WebSearch()
 
34
  agent = CodeAgent(
35
  model=model,
36
  tools=[web_search, visit_webpage, suggest_menu, catering_service_tool, superhero_party_theme_generator],
37
+ managed_agents=[],
38
  max_steps=10,
39
  verbosity_level=2,
40
+ grammar=None,
41
+ planning_interval=None,
42
  name="AlfredAgent",
43
+ description="Code agents example using smolagents",
44
+ #executor_type='local',
45
+ #executor_kwargs={},
46
+ max_print_outputs_length=None,
47
  prompt_templates=prompt_templates
48
  )
49
  if __name__ == "__main__":