Usage Tracking with Tools

Description: Track API usage metrics including token counts and request statistics when using tools

Difficulty: intermediate

Category: tools

Python Code Example

import asyncio

from pydantic import BaseModel

from agents import Agent, Runner, Usage, function_tool


class Weather(BaseModel):
    city: str
    temperature_range: str
    conditions: str


@function_tool
def get_weather(city: str) -> Weather:
    """Get the current weather information for a specified city."""
    return Weather(city=city, temperature_range="14-20C", conditions="Sunny with wind.")


def print_usage(usage: Usage) -> None:
    print("\n=== Usage ===")
    print(f"Requests: {usage.requests}")
    print(f"Input tokens: {usage.input_tokens}")
    print(f"Output tokens: {usage.output_tokens}")
    print(f"Total tokens: {usage.total_tokens}")


async def main():
    agent = Agent(
        name="Usage Demo",
        instructions="You are a concise assistant. Use tools if needed.",
        tools=[get_weather],
    )

    result = await Runner.run(agent, "What's the weather in Tokyo?")

    print("\nFinal output:")
    print(result.final_output)

    # Access usage from the run context
    print_usage(result.context_wrapper.usage)


if __name__ == "__main__":
    asyncio.run(main())

Expected Output: Weather information with detailed usage metrics

About This Example

This is an interactive Python tutorial for the OpenAI Agents SDK. The code runs directly in your browser using PyOdide (Python in WebAssembly).

Key Features:

No installation required - runs entirely in your browser
Interactive code editor with syntax highlighting
Real-time execution using PyOdide
Complete OpenAI Agents SDK functionality
Connect with GitHub Models for AI capabilities

Related Examples:

Agents as Tools - Create a multi-agent system where agents use other agents as tools for translation
Forcing Tool Use - Learn how to force agents to use tools with different behaviors: default, first tool result, or custom logic

← Back to all examples

JavaScript is required to run the interactive code editor. Please enable JavaScript in your browser for the full experience.

Usage Tracking with Tools

Track API usage metrics including token counts and request statistics when using tools

python

Click "Run Code" to execute the Python script

Output will appear here...