from google.adk.agents import Agent
from google.adk.tools.mcp_tool.mcp_session_manager import StdioConnectionParams
from google.adk.tools.mcp_tool.mcp_toolset import MCPToolset
from mcp import StdioServerParameters
import os
# Get API key from environment
TAVILY_API_KEY = os.getenv("TAVILY_API_KEY")
root_agent = Agent(
model="gemini-2.5-pro",
name="tavily_agent",
instruction="You are a helpful assistant that uses Tavily to search the web, extract content, and explore websites.",
tools=[
MCPToolset(
connection_params=StdioConnectionParams(
server_params=StdioServerParameters(
command="npx",
args=[
"-y",
"tavily-mcp@latest",
],
env={
"TAVILY_API_KEY": TAVILY_API_KEY,
}
),
timeout=30,
),
)
],
)