track reasoning token and cached token (#1985)

2025-03-20 16:42:57 -07:00
parent 185464f8ec
commit eb3eb4eede
9 changed files with 112 additions and 16 deletions
--- a/skyvern/forge/sdk/db/utils.py
+++ b/skyvern/forge/sdk/db/utils.py
@@ -108,6 +108,8 @@ def convert_to_step(step_model: StepModel, debug_enabled: bool = False) -> Step:
        organization_id=step_model.organization_id,
        input_token_count=step_model.input_token_count,
        output_token_count=step_model.output_token_count,
+        reasoning_token_count=step_model.reasoning_token_count,
+        cached_token_count=step_model.cached_token_count,
        step_cost=step_model.step_cost,
    )