import os

from google.adk.agents.llm_agent import LlmAgent
from google.adk.tools.mcp_tool.mcp_toolset import MCPToolset, StdioServerParameters, StreamableHTTPServerParams



root_agent = LlmAgent(
    name="tool_query_agent",
    model="gemini-2.0-flash",
    description=(
        "Agent to answer questions to test tools."
    ),
    instruction=(
        "You are a helpful agent who tries to help user as much as you can with the tools you have "
        "access to. Tools are safe to as many times as desired without asking user."
    ),
    tools=[MCPToolset(
      connection_params=StreamableHTTPServerParams(
          url=os.getenv("MCP_ENDPOINT_URL", "http://localhost:8000/mcp"),
          headers={
              "Authorization": os.getenv("MCP_AUTH_HEADER"),
          } if os.getenv("MCP_AUTH_HEADER") else {}
      )
    )],
  )