diff --git a/scripts/trace_codegen/schemas/traces/Trace.json b/scripts/trace_codegen/schemas/traces/Trace.json
index 52a963d0f80..b0a81fafb49 100644
--- a/scripts/trace_codegen/schemas/traces/Trace.json
+++ b/scripts/trace_codegen/schemas/traces/Trace.json
@@ -92,6 +92,12 @@
             "null"
           ]
         },
+        "priced_calls": {
+          "format": "uint64",
+          "maximum": 18446744073709551615,
+          "minimum": 0,
+          "type": "integer"
+        },
         "spend": {
           "format": "double",
           "type": [
@@ -115,7 +121,8 @@
         "llm_calls",
         "tool_calls",
         "duration_ms",
-        "spend"
+        "spend",
+        "priced_calls"
       ],
       "type": "object"
     },
@@ -205,6 +212,22 @@
             "null"
           ]
         },
+        "spend_log_request_id": {
+          "type": [
+            "string",
+            "null"
+          ]
+        },
+        "spend_match": {
+          "anyOf": [
+            {
+              "$ref": "#/$defs/SpendMatch"
+            },
+            {
+              "type": "null"
+            }
+          ]
+        },
         "start_offset_ms": {
           "format": "double",
           "type": "number"
@@ -236,7 +259,9 @@
         "input_tokens",
         "output_tokens",
         "litellm_request_id",
-        "spend"
+        "spend",
+        "spend_log_request_id",
+        "spend_match"
       ],
       "type": "object"
     },
@@ -265,6 +290,16 @@
       ],
       "type": "string"
     },
+    "SpendMatch": {
+      "enum": [
+        "matched",
+        "no_call_id",
+        "no_spend_log",
+        "ambiguous",
+        "incomplete_evidence"
+      ],
+      "type": "string"
+    },
     "TraceCapture": {
       "properties": {
         "actors": {
@@ -368,6 +403,12 @@
           "minimum": 0,
           "type": "integer"
         },
+        "priced_calls": {
+          "format": "uint64",
+          "maximum": 18446744073709551615,
+          "minimum": 0,
+          "type": "integer"
+        },
         "resolution_limited": {
           "type": "boolean",
           "x-python-optional": true
@@ -429,7 +470,8 @@
         "input_tokens",
         "output_tokens",
         "models",
-        "spend"
+        "spend",
+        "priced_calls"
       ],
       "type": "object"
     }