Ahmed Yousry commited on
Commit
dd3f56b
·
verified ·
1 Parent(s): 45a17f0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,12 +12,12 @@ from tools.catering_service_tool import SimpleTool as CateringServiceTool
12
  from tools.superhero_party_theme_generator import SuperheroPartyThemeTool as SuperheroPartyThemeGenerator
13
  from tools.final_answer import FinalAnswerTool as FinalAnswer
14
 
15
-
16
 
17
  model = LiteLLMModel(
18
  max_tokens=8192,
19
  model_id='gemini/gemini-2.0-flash',
20
- api_base=None,
21
  )
22
 
23
  web_search = WebSearch()
 
12
  from tools.superhero_party_theme_generator import SuperheroPartyThemeTool as SuperheroPartyThemeGenerator
13
  from tools.final_answer import FinalAnswerTool as FinalAnswer
14
 
15
+ GEMINI_API_KEY = os.environ.get('GEMINI_API_KEY')
16
 
17
  model = LiteLLMModel(
18
  max_tokens=8192,
19
  model_id='gemini/gemini-2.0-flash',
20
+ api_key=GEMINI_API_KEY,
21
  )
22
 
23
  web_search = WebSearch()