Update app.py

app.py

@@ -13,10 +13,10 @@ from tools.superhero_party_theme_generator import SuperheroPartyThemeTool as Sup
 from tools.final_answer import FinalAnswerTool as FinalAnswer
 
 
-
 model = LiteLLMModel(
-model_id='gemini/gemini-2.5-flash-preview-04-17',
-api_base=None,
+    model_id='gemini/gemini-2.5-flash-preview-04-17',
+    api_key=os.env['GEMINI_API_KEY'],
+    api_base=None,
 )
 
 web_search = WebSearch()
@@ -34,7 +34,6 @@ agent = CodeAgent(
     model=model,
     tools=[web_search, visit_webpage, suggest_menu, catering_service_tool, superhero_party_theme_generator],
     managed_agents=[],
-    class='CodeAgent',
     max_steps=10,
     verbosity_level=2,
     grammar=None,