fix KeyError cost_usd

This commit is contained in:
taku_sid 2025-06-13 06:18:51 +00:00
parent 54bff2e85d
commit 1894744b43
2 changed files with 4 additions and 4 deletions

View file

@ -86,13 +86,13 @@ class InternalClient:
# Map total_cost to total_cost_usd for consistency
return ResultMessage(
subtype=data["subtype"],
cost_usd=data["cost_usd"],
# cost_usd=data["cost_usd"],
duration_ms=data["duration_ms"],
duration_api_ms=data["duration_api_ms"],
is_error=data["is_error"],
num_turns=data["num_turns"],
session_id=data["session_id"],
total_cost_usd=data["total_cost"],
# total_cost_usd=data["total_cost"],
usage=data.get("usage"),
result=data.get("result"),
)

View file

@ -75,13 +75,13 @@ class ResultMessage:
"""Result message with cost and usage information."""
subtype: str
cost_usd: float
duration_ms: int
duration_api_ms: int
is_error: bool
num_turns: int
session_id: str
total_cost_usd: float
cost_usd: float | None = None
total_cost_usd: float | None = None
usage: dict[str, Any] | None = None
result: str | None = None