davemasino commited on
Commit
8e432bb
·
verified ·
1 Parent(s): 51f0764

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,8 +15,8 @@ from tools.final_answer import FinalAnswerTool as FinalAnswer
15
 
16
 
17
  model = LiteLLMModel(
18
- model_id='gemini/gemini-2.0-flash-001',
19
- api_base=None,
20
  )
21
 
22
  web_search = WebSearch()
 
15
 
16
 
17
  model = LiteLLMModel(
18
+ model_id='gemini/gemini-2.0-flash-001',
19
+ api_key=os.getenv("LITELLM_API_KEY")
20
  )
21
 
22
  web_search = WebSearch()