From fe713d819428d3268f298997ea9d2755bdd81721 Mon Sep 17 00:00:00 2001 From: Shuchang Zheng Date: Fri, 21 Mar 2025 00:02:22 -0700 Subject: [PATCH] fix StepModel.reasoning_token_count and cached_token_count (#1987) --- skyvern/forge/sdk/models.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/skyvern/forge/sdk/models.py b/skyvern/forge/sdk/models.py index 7b6854f9..1091dac7 100644 --- a/skyvern/forge/sdk/models.py +++ b/skyvern/forge/sdk/models.py @@ -52,8 +52,8 @@ class Step(BaseModel): organization_id: str | None = None input_token_count: int = 0 output_token_count: int = 0 - reasoning_token_count: int = 0 - cached_token_count: int = 0 + reasoning_token_count: int | None = None + cached_token_count: int | None = None step_cost: float = 0 def validate_update(