Update app.py

app.py

@@ -54,31 +54,31 @@ tools = [web_search]
 
 #LLMS
 # Create LLM class 
-#gemini_llm = ChatGoogleGenerativeAI(
-#    model= "gemini-2.0-flash", # replace with "gemini-2.0-flash"
-#    temperature=1.0,
-#    max_tokens=None,
-#    timeout=None,
-#    max_retries=2,
-#    google_api_key=GEMINI_API_KEY,
-#)
+gemini_llm = ChatGoogleGenerativeAI(
+    model= "gemini-2.0-flash-exp", # replace with "gemini-2.0-flash"
+    temperature=1.0,
+    max_tokens=None,
+    timeout=None,
+    max_retries=2,
+    google_api_key=GEMINI_API_KEY,
+)
 # Bind tools to the model
 #general_model = gemini_llm.bind_tools([get_weather_forecast])
 
 #AGENTS
 web_research_agent = create_react_agent(
-    model="gemini-2.0-flash",  # gpt-4o-mini-2024-07-18
+    model=gemini_llm,
     tools=[web_search],
     prompt=prompt_search,
-    name="web_research_agent",
+    name="web_research_agent"
 )
 
 supervisor = create_supervisor(
-    model=init_chat_model("gemini-2.0-flash"),
+    model=gemini_llm,
     agents=[web_research_agent],
     prompt=prompt_recomendado,
     add_handoff_back_messages=True,
-    output_mode="full_history",
+    output_mode="full_history"
 ).compile()