Unverified Commit 33278073 authored by ゆり's avatar ゆり Committed by GitHub
Browse files

typing: Add type hints to TurnMetrics class in context.py (#30552)


Co-authored-by: default avatarzkexorability <zkexorability@gmail.com>
Co-authored-by: default avatarClaude Opus 4.5 <noreply@anthropic.com>
parent 1adeb3b8
...@@ -74,24 +74,24 @@ class TurnMetrics: ...@@ -74,24 +74,24 @@ class TurnMetrics:
def __init__( def __init__(
self, self,
input_tokens=0, input_tokens: int = 0,
output_tokens=0, output_tokens: int = 0,
cached_input_tokens=0, cached_input_tokens: int = 0,
tool_output_tokens=0, tool_output_tokens: int = 0,
): ) -> None:
self.input_tokens = input_tokens self.input_tokens = input_tokens
self.output_tokens = output_tokens self.output_tokens = output_tokens
self.cached_input_tokens = cached_input_tokens self.cached_input_tokens = cached_input_tokens
self.tool_output_tokens = tool_output_tokens self.tool_output_tokens = tool_output_tokens
def reset(self): def reset(self) -> None:
"""Reset counters for a new turn.""" """Reset counters for a new turn."""
self.input_tokens = 0 self.input_tokens = 0
self.output_tokens = 0 self.output_tokens = 0
self.cached_input_tokens = 0 self.cached_input_tokens = 0
self.tool_output_tokens = 0 self.tool_output_tokens = 0
def copy(self): def copy(self) -> "TurnMetrics":
"""Create a copy of this turn's token counts.""" """Create a copy of this turn's token counts."""
return TurnMetrics( return TurnMetrics(
self.input_tokens, self.input_tokens,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment