Regenerate types for session.usage_info and apply PR feedback

Author: SteveSandersonMS

dotnet/src/Generated/SessionEvents.cs

@@ -6,7 +6,7 @@
 //
 // Generated from: @github/copilot/session-events.schema.json
 // Generated by: scripts/generate-session-types.ts
-// Generated at: 2026-01-15T19:22:26.479Z
+// Generated at: 2026-01-16T00:52:52.781Z
 //
 // To update these types:
 // 1. Update the schema in copilot-agent-runtime
@@ -39,6 +39,7 @@ internal class SessionEventConverter : JsonConverter<SessionEvent>
             ["session.model_change"] = typeof(SessionModelChangeEvent),
             ["session.handoff"] = typeof(SessionHandoffEvent),
             ["session.truncation"] = typeof(SessionTruncationEvent),
+            ["session.usage_info"] = typeof(SessionUsageInfoEvent),
             ["session.compaction_start"] = typeof(SessionCompactionStartEvent),
             ["session.compaction_complete"] = typeof(SessionCompactionCompleteEvent),
             ["user.message"] = typeof(UserMessageEvent),
@@ -209,6 +210,17 @@ public partial class SessionTruncationEvent : SessionEvent
         public SessionTruncationData Data { get; set; }
     }
 
+    /// <summary>
+    /// Event: session.usage_info
+    /// </summary>
+    public partial class SessionUsageInfoEvent : SessionEvent
+    {
+        public override string Type => "session.usage_info";
+
+        [JsonPropertyName("data")]
+        public SessionUsageInfoData Data { get; set; }
+    }
+
     /// <summary>
     /// Event: session.compaction_start
     /// </summary>
@@ -592,6 +604,18 @@ public partial class SessionTruncationData
         public string PerformedBy { get; set; }
     }
 
+    public partial class SessionUsageInfoData
+    {
+        [JsonPropertyName("tokenLimit")]
+        public double TokenLimit { get; set; }
+
+        [JsonPropertyName("currentTokens")]
+        public double CurrentTokens { get; set; }
+
+        [JsonPropertyName("messagesLength")]
+        public double MessagesLength { get; set; }
+    }
+
     public partial class SessionCompactionStartData
     {
     }

go/generated/session_events.go

@@ -3,7 +3,7 @@
  *
  * Generated from: @github/copilot/session-events.schema.json
  * Generated by: scripts/generate-session-types.ts
- * Generated at: 2026-01-15T19:22:25.859Z
+ * Generated at: 2026-01-16T00:52:51.450Z
  *
  * To update these types:
  * 1. Update the schema in copilot-agent-runtime
@@ -115,6 +115,18 @@ export type SessionEvent =
         performedBy: string;
       };
     }
+  | {
+      id: string;
+      timestamp: string;
+      parentId: string | null;
+      ephemeral: true;
+      type: "session.usage_info";
+      data: {
+        tokenLimit: number;
+        currentTokens: number;
+        messagesLength: number;
+      };
+    }
   | {
       id: string;
       timestamp: string;

nodejs/src/generated/session-events.ts

@@ -51,6 +51,7 @@ export async function createSdkTestContext() {
 
     // Wire up to Vitest lifecycle
     beforeEach(async (testContext) => {
+        // Must be inside beforeEach - vitest requires test context
         onTestFailed(() => {
             anyTestFailed = true;
         });

nodejs/test/e2e/harness/sdkTestContext.ts

@@ -3,7 +3,7 @@
 
 Generated from: @github/copilot/session-events.schema.json
 Generated by: scripts/generate-session-types.ts
-Generated at: 2026-01-15T19:22:26.227Z
+Generated at: 2026-01-16T00:52:51.909Z
 
 To update these types:
 1. Update the schema in copilot-agent-runtime
@@ -314,6 +314,8 @@ class Data:
     pre_truncation_tokens_in_messages: Optional[float] = None
     token_limit: Optional[float] = None
     tokens_removed_during_truncation: Optional[float] = None
+    current_tokens: Optional[float] = None
+    messages_length: Optional[float] = None
     compaction_tokens_used: Optional[CompactionTokensUsed] = None
     error: Optional[Union[ErrorClass, str]] = None
     messages_removed: Optional[float] = None
@@ -397,6 +399,8 @@ def from_dict(obj: Any) -> 'Data':
         pre_truncation_tokens_in_messages = from_union([from_float, from_none], obj.get("preTruncationTokensInMessages"))
         token_limit = from_union([from_float, from_none], obj.get("tokenLimit"))
         tokens_removed_during_truncation = from_union([from_float, from_none], obj.get("tokensRemovedDuringTruncation"))
+        current_tokens = from_union([from_float, from_none], obj.get("currentTokens"))
+        messages_length = from_union([from_float, from_none], obj.get("messagesLength"))
         compaction_tokens_used = from_union([CompactionTokensUsed.from_dict, from_none], obj.get("compactionTokensUsed"))
         error = from_union([ErrorClass.from_dict, from_str, from_none], obj.get("error"))
         messages_removed = from_union([from_float, from_none], obj.get("messagesRemoved"))
@@ -448,7 +452,7 @@ def from_dict(obj: Any) -> 'Data':
         metadata = from_union([Metadata.from_dict, from_none], obj.get("metadata"))
         name = from_union([from_str, from_none], obj.get("name"))
         role = from_union([Role, from_none], obj.get("role"))
-        return Data(copilot_version, producer, selected_model, session_id, start_time, version, event_count, resume_time, error_type, message, stack, info_type, new_model, previous_model, context, handoff_time, remote_session_id, repository, source_type, summary, messages_removed_during_truncation, performed_by, post_truncation_messages_length, post_truncation_tokens_in_messages, pre_truncation_messages_length, pre_truncation_tokens_in_messages, token_limit, tokens_removed_during_truncation, compaction_tokens_used, error, messages_removed, post_compaction_tokens, pre_compaction_messages_length, pre_compaction_tokens, success, summary_content, tokens_removed, attachments, content, source, transformed_content, turn_id, intent, reasoning_id, delta_content, message_id, parent_tool_call_id, tool_requests, total_response_size_bytes, api_call_id, cache_read_tokens, cache_write_tokens, cost, duration, initiator, input_tokens, model, output_tokens, provider_call_id, quota_snapshots, reason, arguments, tool_call_id, tool_name, partial_output, is_user_requested, result, tool_telemetry, agent_description, agent_display_name, agent_name, tools, hook_invocation_id, hook_type, input, output, metadata, name, role)
+        return Data(copilot_version, producer, selected_model, session_id, start_time, version, event_count, resume_time, error_type, message, stack, info_type, new_model, previous_model, context, handoff_time, remote_session_id, repository, source_type, summary, messages_removed_during_truncation, performed_by, post_truncation_messages_length, post_truncation_tokens_in_messages, pre_truncation_messages_length, pre_truncation_tokens_in_messages, token_limit, tokens_removed_during_truncation, current_tokens, messages_length, compaction_tokens_used, error, messages_removed, post_compaction_tokens, pre_compaction_messages_length, pre_compaction_tokens, success, summary_content, tokens_removed, attachments, content, source, transformed_content, turn_id, intent, reasoning_id, delta_content, message_id, parent_tool_call_id, tool_requests, total_response_size_bytes, api_call_id, cache_read_tokens, cache_write_tokens, cost, duration, initiator, input_tokens, model, output_tokens, provider_call_id, quota_snapshots, reason, arguments, tool_call_id, tool_name, partial_output, is_user_requested, result, tool_telemetry, agent_description, agent_display_name, agent_name, tools, hook_invocation_id, hook_type, input, output, metadata, name, role)
 
     def to_dict(self) -> dict:
         result: dict = {}
@@ -508,6 +512,10 @@ def to_dict(self) -> dict:
             result["tokenLimit"] = from_union([to_float, from_none], self.token_limit)
         if self.tokens_removed_during_truncation is not None:
             result["tokensRemovedDuringTruncation"] = from_union([to_float, from_none], self.tokens_removed_during_truncation)
+        if self.current_tokens is not None:
+            result["currentTokens"] = from_union([to_float, from_none], self.current_tokens)
+        if self.messages_length is not None:
+            result["messagesLength"] = from_union([to_float, from_none], self.messages_length)
         if self.compaction_tokens_used is not None:
             result["compactionTokensUsed"] = from_union([lambda x: to_class(CompactionTokensUsed, x), from_none], self.compaction_tokens_used)
         if self.error is not None:
@@ -636,6 +644,7 @@ class SessionEventType(Enum):
     SESSION_RESUME = "session.resume"
     SESSION_START = "session.start"
     SESSION_TRUNCATION = "session.truncation"
+    SESSION_USAGE_INFO = "session.usage_info"
     SUBAGENT_COMPLETED = "subagent.completed"
     SUBAGENT_FAILED = "subagent.failed"
     SUBAGENT_SELECTED = "subagent.selected"

python/copilot/generated/session_events.py

@@ -5,29 +5,26 @@
 
 from .testharness import E2ETestContext
 
-# Track if any test failed to avoid writing corrupted snapshots
-_any_test_failed = False
-
 
 @pytest.hookimpl(tryfirst=True, hookwrapper=True)
 def pytest_runtest_makereport(item, call):
     """Track test failures to avoid writing corrupted snapshots."""
-    global _any_test_failed
     outcome = yield
     rep = outcome.get_result()
     if rep.when == "call" and rep.failed:
-        _any_test_failed = True
+        # Store on the item's stash so the fixture can access it
+        item.session.stash.setdefault("any_test_failed", False)
+        item.session.stash["any_test_failed"] = True
 
 
 @pytest_asyncio.fixture(scope="module", loop_scope="module")
-async def ctx():
+async def ctx(request):
     """Create and teardown a test context shared across all tests in this module."""
-    global _any_test_failed
-    _any_test_failed = False  # Reset for each module
     context = E2ETestContext()
     await context.setup()
     yield context
-    await context.teardown(test_failed=_any_test_failed)
+    any_failed = request.session.stash.get("any_test_failed", False)
+    await context.teardown(test_failed=any_failed)
 
 
 @pytest_asyncio.fixture(autouse=True, loop_scope="module")