fix StepModel.reasoning_token_count and cached_token_count (#1987)

2025-03-21 00:02:22 -07:00
parent f924185caf
commit fe713d8194
1 changed files with 2 additions and 2 deletions
--- 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(