Google Gemini Function Calling Intermediate

import google.generativeai as genai

# Define your actual Python function
def get_weather(location: str, unit: str = "celsius") -> dict:
    """Get current weather for a location.

    Args:
        location: City name, e.g. 'Tokyo'
        unit: Temperature unit (celsius or fahrenheit)
    """
    # Your implementation here
    return {"temp": 22, "unit": unit, "condition": "Sunny"}

# Gemini can infer the schema from Python type hints!
model = genai.GenerativeModel(
    model_name="gemini-2.0-flash",
    tools=[get_weather]
)

chat = model.start_chat()
response = chat.send_message("What's the weather in Tokyo?")

# Check for function calls
for part in response.parts:
    if part.function_call:
        name = part.function_call.name
        args = dict(part.function_call.args)
        print(f"Function: {name}, Args: {args}")

        # Execute and send result back
        result = get_weather(**args)
        response = chat.send_message(
            genai.protos.Content(parts=[
                genai.protos.Part(function_response=genai.protos.FunctionResponse(
                    name=name,
                    response={"result": result}
                ))
            ])
        )
        print(response.text)

# Enable automatic function calling
model = genai.GenerativeModel(
    model_name="gemini-2.0-flash",
    tools=[get_weather]
)

# The SDK handles the entire tool use loop automatically!
response = model.generate_content(
    "What's the weather in Tokyo?",
    tool_config={"function_calling_config": "AUTO"}
)

# response.text already contains the final answer
print(response.text)
# "The weather in Tokyo is currently 22°C and sunny!"