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
EXA_API_KEY = "YOUR_EXA_API_KEY"
root_agent = Agent(
model="gemini-2.5-pro",
name="exa_agent",
instruction="Help users get information from Exa",
tools=[
MCPToolset(
connection_params=StdioConnectionParams(
server_params = StdioServerParameters(
command="npx",
args=[
"-y",
"exa-mcp-server",
# (Optional) Choose which tools to use
# If you don't pick any tools, all tools will be used by default
# "--tools=get_code_context_exa,web_search_exa",
],
env={
"EXA_API_KEY": EXA_API_KEY,
}
),
timeout=30,
),
)
],
)