From b97e4b4d065b4e90d4bb719063415dbf2b626055 Mon Sep 17 00:00:00 2001 From: jsonbailey Date: Mon, 18 May 2026 13:55:01 -0700 Subject: [PATCH] fix: Use summary.tokens instead of summary.usage in examples Co-Authored-By: Claude Opus 4.7 (1M context) --- features/create_agent/create_agent_example.py | 8 ++++---- .../create_agent_graph_example.py | 16 ++++++++-------- .../bedrock/converse/bedrock_example.py | 8 ++++---- .../gemini/generate_content/gemini_example.py | 8 ++++---- .../langchain/invoke/langchain_example.py | 8 ++++---- .../react_agent/langgraph_agent_example.py | 8 ++++---- .../state_graph/langgraph_multi_agent_example.py | 8 ++++---- .../openai/chat_completions/openai_example.py | 8 ++++---- 8 files changed, 36 insertions(+), 36 deletions(-) diff --git a/features/create_agent/create_agent_example.py b/features/create_agent/create_agent_example.py index 43d6c99..a1a4d76 100644 --- a/features/create_agent/create_agent_example.py +++ b/features/create_agent/create_agent_example.py @@ -85,10 +85,10 @@ async def async_main(): print("\nMetrics tracked:") print(f" Duration: {summary.duration_ms}ms") print(f" Success: {summary.success}") - if summary.usage: - print(f" Input tokens: {summary.usage.input}") - print(f" Output tokens: {summary.usage.output}") - print(f" Total tokens: {summary.usage.total}") + if summary.tokens: + print(f" Input tokens: {summary.tokens.input}") + print(f" Output tokens: {summary.tokens.output}") + print(f" Total tokens: {summary.tokens.total}") if summary.tool_calls: print(f" Tool calls: {', '.join(summary.tool_calls)}") diff --git a/features/create_agent_graph/create_agent_graph_example.py b/features/create_agent_graph/create_agent_graph_example.py index 5ada8a0..ba646ac 100644 --- a/features/create_agent_graph/create_agent_graph_example.py +++ b/features/create_agent_graph/create_agent_graph_example.py @@ -92,10 +92,10 @@ async def async_main(): print(f" Success: {summary.success}") if summary.path: print(f" Path: {' -> '.join(summary.path)}") - if summary.usage: - print(f" Input tokens: {summary.usage.input}") - print(f" Output tokens: {summary.usage.output}") - print(f" Total tokens: {summary.usage.total}") + if summary.tokens: + print(f" Input tokens: {summary.tokens.input}") + print(f" Output tokens: {summary.tokens.output}") + print(f" Total tokens: {summary.tokens.total}") if summary.node_metrics: print("\nPer-node metrics:") @@ -105,10 +105,10 @@ async def async_main(): print(f" Duration: {node_summary.duration_ms}ms") if node_summary.success is not None: print(f" Success: {node_summary.success}") - if node_summary.usage: - print(f" Input tokens: {node_summary.usage.input}") - print(f" Output tokens: {node_summary.usage.output}") - print(f" Total tokens: {node_summary.usage.total}") + if node_summary.tokens: + print(f" Input tokens: {node_summary.tokens.input}") + print(f" Output tokens: {node_summary.tokens.output}") + print(f" Total tokens: {node_summary.tokens.total}") if node_summary.tool_calls: print(f" Tool calls: {', '.join(node_summary.tool_calls)}") diff --git a/getting_started/bedrock/converse/bedrock_example.py b/getting_started/bedrock/converse/bedrock_example.py index 85d8ba9..6105133 100755 --- a/getting_started/bedrock/converse/bedrock_example.py +++ b/getting_started/bedrock/converse/bedrock_example.py @@ -122,10 +122,10 @@ def main(): print("\nDone! The AI config was evaluated and the following metrics were tracked:") print(f" Duration: {summary.duration_ms}ms") print(f" Success: {summary.success}") - if summary.usage: - print(f" Input tokens: {summary.usage.input}") - print(f" Output tokens: {summary.usage.output}") - print(f" Total tokens: {summary.usage.total}") + if summary.tokens: + print(f" Input tokens: {summary.tokens.input}") + print(f" Output tokens: {summary.tokens.output}") + print(f" Total tokens: {summary.tokens.total}") if summary.tool_calls: print(f" Tool calls: {', '.join(summary.tool_calls)}") diff --git a/getting_started/gemini/generate_content/gemini_example.py b/getting_started/gemini/generate_content/gemini_example.py index 086204c..3e4621f 100644 --- a/getting_started/gemini/generate_content/gemini_example.py +++ b/getting_started/gemini/generate_content/gemini_example.py @@ -177,10 +177,10 @@ def main(): print("\nDone! The AI config was evaluated and the following metrics were tracked:") print(f" Duration: {summary.duration_ms}ms") print(f" Success: {summary.success}") - if summary.usage: - print(f" Input tokens: {summary.usage.input}") - print(f" Output tokens: {summary.usage.output}") - print(f" Total tokens: {summary.usage.total}") + if summary.tokens: + print(f" Input tokens: {summary.tokens.input}") + print(f" Output tokens: {summary.tokens.output}") + print(f" Total tokens: {summary.tokens.total}") if summary.tool_calls: print(f" Tool calls: {', '.join(summary.tool_calls)}") diff --git a/getting_started/langchain/invoke/langchain_example.py b/getting_started/langchain/invoke/langchain_example.py index 3b763d2..1c4b7d3 100644 --- a/getting_started/langchain/invoke/langchain_example.py +++ b/getting_started/langchain/invoke/langchain_example.py @@ -111,10 +111,10 @@ async def async_main(): print("\nDone! The AI config was evaluated and the following metrics were tracked:") print(f" Duration: {summary.duration_ms}ms") print(f" Success: {summary.success}") - if summary.usage: - print(f" Input tokens: {summary.usage.input}") - print(f" Output tokens: {summary.usage.output}") - print(f" Total tokens: {summary.usage.total}") + if summary.tokens: + print(f" Input tokens: {summary.tokens.input}") + print(f" Output tokens: {summary.tokens.output}") + print(f" Total tokens: {summary.tokens.total}") if summary.tool_calls: print(f" Tool calls: {', '.join(summary.tool_calls)}") diff --git a/getting_started/langgraph/react_agent/langgraph_agent_example.py b/getting_started/langgraph/react_agent/langgraph_agent_example.py index ab299e9..f804f61 100644 --- a/getting_started/langgraph/react_agent/langgraph_agent_example.py +++ b/getting_started/langgraph/react_agent/langgraph_agent_example.py @@ -114,10 +114,10 @@ def main(): print("\nDone! The agent config was evaluated and the following metrics were tracked:") print(f" Duration: {summary.duration_ms}ms") print(f" Success: {summary.success}") - if summary.usage: - print(f" Input tokens: {summary.usage.input}") - print(f" Output tokens: {summary.usage.output}") - print(f" Total tokens: {summary.usage.total}") + if summary.tokens: + print(f" Input tokens: {summary.tokens.input}") + print(f" Output tokens: {summary.tokens.output}") + print(f" Total tokens: {summary.tokens.total}") if summary.tool_calls: print(f" Tool calls: {', '.join(summary.tool_calls)}") diff --git a/getting_started/langgraph/state_graph/langgraph_multi_agent_example.py b/getting_started/langgraph/state_graph/langgraph_multi_agent_example.py index b2b1691..4243f95 100644 --- a/getting_started/langgraph/state_graph/langgraph_multi_agent_example.py +++ b/getting_started/langgraph/state_graph/langgraph_multi_agent_example.py @@ -56,10 +56,10 @@ def track_langgraph_metrics(tracker, func, prev_message_count=0): new_messages = result["messages"][prev_message_count:] for message in new_messages: metrics = get_ai_metrics_from_response(message) - if metrics.usage: - total_input_tokens += metrics.usage.input - total_output_tokens += metrics.usage.output - total_tokens += metrics.usage.total + if metrics.tokens: + total_input_tokens += metrics.tokens.input + total_output_tokens += metrics.tokens.output + total_tokens += metrics.tokens.total if total_tokens > 0: tracker.track_tokens( TokenUsage( diff --git a/getting_started/openai/chat_completions/openai_example.py b/getting_started/openai/chat_completions/openai_example.py index 19fd0cf..fb4b3bf 100755 --- a/getting_started/openai/chat_completions/openai_example.py +++ b/getting_started/openai/chat_completions/openai_example.py @@ -101,10 +101,10 @@ def main(): print("\nDone! The AI config was evaluated and the following metrics were tracked:") print(f" Duration: {summary.duration_ms}ms") print(f" Success: {summary.success}") - if summary.usage: - print(f" Input tokens: {summary.usage.input}") - print(f" Output tokens: {summary.usage.output}") - print(f" Total tokens: {summary.usage.total}") + if summary.tokens: + print(f" Input tokens: {summary.tokens.input}") + print(f" Output tokens: {summary.tokens.output}") + print(f" Total tokens: {summary.tokens.total}") if summary.tool_calls: print(f" Tool calls: {', '.join(summary.tool_calls)}")