Update app.py

app.py

@@ -7,6 +7,8 @@ from functools import partial
 from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain.agents import load_tools
 from langgraph.graph import StateGraph, START, END
+from langchain_community.tools.tavily_search import TavilySearchResults
+from langchain_core.tools import tool
 
 # Importa as funções e a classe de estado do outro arquivo
 from agente_functions import (
@@ -33,6 +35,15 @@ if not os.environ.get("GOOGLE_API_KEY") or not os.environ.get("TAVILY_API_KEY"):
     print("Por favor, defina as variáveis de ambiente GOOGLE_API_KEY e TAVILY_API_KEY.")
     exit()
 
+@tool
+def busca_web(query: str) -> list:
+  """
+  Busca na web por um termo específico
+  """
+  tavily_search = TavilySearchResults(max_results=2, search_depth='advanced', max_tokens=1000)
+  resultado_busca = tavily_search.invoke(query)
+  return resultado_busca
+
 # --- 2. Inicialização do Modelo e Ferramentas ---
 llm = ChatGoogleGenerativeAI(model="gemini-1.5-flash")
 web_tools = [busca_web]