from typing import Dict

# --- Core ADK Imports ---
from google.adk.agents import Agent
from google.adk.tools import agent_tool, ToolContext
from vertexai.preview.reasoning_engines import AdkApp

# --- 1. Define Function

def get_weather(
    tool_context: ToolContext,
    skip_summary: bool = False
) -> str:
    """Gets the weather.
    
    Args:
        skip_summary (bool): Set to True to return the raw, final string "sunny".
    """
    
    print(f"Tool called. LLM set skip_summary={skip_summary}")
    to_return = "Super Duper Very Sunny!!!"

    if skip_summary:
        print("Tool: Setting skip_summarization=True")
        tool_context.actions.skip_summarization = True
        #print(to_return) uncomment to print the output, since otherwise nothing is returned.

    return to_return


# --- 2. The Agent Setup

root_agent = Agent(
    name="weather_bot",
    model="gemini-2.5-pro",
    instruction="You are a weather bot.",
    tools=[
        get_weather,
    ]
)