feat: more message related logs

Author: PoolPirate

src/proxy/bridge-non-streaming.ts

@@ -155,6 +155,7 @@ async function collectFullResponse(
               scheduleToolCallBridgeEnd();
             },
             (_info: McpToolCallUpdateInfo) => {},
+            undefined,
             (checkpointBytes) => {
               updateConversationCheckpoint(convKey, checkpointBytes);
               bridgeCloseController.noteCheckpoint();

src/proxy/bridge-streaming.ts

@@ -40,6 +40,7 @@ import {
   createConnectFrameParser,
   createThinkingTagFilter,
   type McpToolCallUpdateInfo,
+  type StepUpdateInfo,
   parseConnectEndStream,
   processServerMessage,
   scheduleBridgeEnd,
@@ -471,6 +472,47 @@ function createBridgeStreamResponse(
                 });
 
               },
+              (info: StepUpdateInfo) => {
+                const existingActiveBridge = activeBridges.get(bridgeKey);
+                if (existingActiveBridge) {
+                  existingActiveBridge.diagnostics = {
+                    announcedToolCallIds:
+                      existingActiveBridge.diagnostics?.announcedToolCallIds ?? [],
+                    publishedToolCallIds:
+                      existingActiveBridge.diagnostics?.publishedToolCallIds ?? [],
+                    lastMcpUpdate:
+                      info.updateCase === "stepCompleted"
+                        ? `${info.updateCase}:${info.stepId}:${info.stepDurationMs ?? "unknown"}`
+                        : `${info.updateCase}:${info.stepId}`,
+                    publishedAtMs: existingActiveBridge.diagnostics?.publishedAtMs,
+                    lastResumeAttemptAtMs:
+                      existingActiveBridge.diagnostics?.lastResumeAttemptAtMs,
+                  };
+                }
+                logPluginInfo("Tracking Cursor step boundary in streaming bridge", {
+                  modelId,
+                  bridgeKey,
+                  convKey,
+                  updateCase: info.updateCase,
+                  stepId: info.stepId,
+                  stepDurationMs: info.stepDurationMs,
+                  toolCallsFlushed,
+                  mcpExecReceived,
+                  pendingExecToolCallIds: sortedIds(
+                    state.pendingExecs.map((candidate) => candidate.toolCallId),
+                  ),
+                  streamedToolCallIds: sortedIds(streamedToolCalls.keys()),
+                  hasStoredActiveBridge: Boolean(existingActiveBridge),
+                  storedActiveBridgePendingExecToolCallIds: existingActiveBridge
+                    ? sortedIds(
+                        existingActiveBridge.pendingExecs.map(
+                          (candidate) => candidate.toolCallId,
+                        ),
+                      )
+                    : [],
+                  storedActiveBridgeDiagnostics: existingActiveBridge?.diagnostics,
+                });
+              },
               (checkpointBytes) => {
                 updateConversationCheckpoint(convKey, checkpointBytes);
                 bridgeCloseController.noteCheckpoint();

src/proxy/stream-dispatch.ts

@@ -83,6 +83,12 @@ export interface McpToolCallUpdateInfo {
   toolName?: string;
 }
 
+export interface StepUpdateInfo {
+  updateCase: "stepStarted" | "stepCompleted";
+  stepId: string;
+  stepDurationMs?: string;
+}
+
 export function parseConnectEndStream(data: Uint8Array): Error | null {
   try {
     const payload = JSON.parse(new TextDecoder().decode(data));
@@ -228,6 +234,7 @@ export function processServerMessage(
   onText: (text: string, isThinking?: boolean) => void,
   onMcpExec: (exec: PendingExec) => void,
   onMcpToolCallUpdate?: (info: McpToolCallUpdateInfo) => void,
+  onStepUpdate?: (info: StepUpdateInfo) => void,
   onCheckpoint?: (checkpointBytes: Uint8Array) => void,
   onTurnEnded?: () => void,
   onUnsupportedMessage?: (info: UnsupportedServerMessageInfo) => void,
@@ -241,6 +248,7 @@ export function processServerMessage(
       state,
       onText,
       onMcpToolCallUpdate,
+      onStepUpdate,
       onTurnEnded,
       onUnsupportedMessage,
     );
@@ -288,6 +296,7 @@ function handleInteractionUpdate(
   state: StreamState,
   onText: (text: string, isThinking?: boolean) => void,
   onMcpToolCallUpdate?: (info: McpToolCallUpdateInfo) => void,
+  onStepUpdate?: (info: StepUpdateInfo) => void,
   onTurnEnded?: () => void,
   onUnsupportedMessage?: (info: UnsupportedServerMessageInfo) => void,
 ): void {
@@ -325,6 +334,21 @@ function handleInteractionUpdate(
     }
   }
 
+  if (updateCase === "stepStarted" || updateCase === "stepCompleted") {
+    const stepValue = update.message?.value;
+    const stepId = stepValue?.stepId;
+    if (stepId !== undefined && stepId !== null) {
+      onStepUpdate?.({
+        updateCase,
+        stepId: String(stepId),
+        stepDurationMs:
+          updateCase === "stepCompleted" && stepValue?.stepDurationMs !== undefined
+            ? String(stepValue.stepDurationMs)
+            : undefined,
+      });
+    }
+  }
+
   if (updateCase === "textDelta") {
     const delta = update.message.value.text || "";
     if (delta) onText(delta, false);