fix(ui-trajectory): consume prompt changes once

2026-08-10 19:26:54 +08:00
parent a342b329e3
commit 3469de58eb
2 changed files with 71 additions and 2 deletions
@@ -46,6 +46,7 @@ function headerFor(
 function applyHeader(
  request: Extract<RequestView, { purpose: 'assistant' }>,
  header: TrajectoryRequestHeaderState | undefined,
+  includeChange: boolean,
 ): Extract<RequestView, { purpose: 'assistant' }> {
  return header === undefined
    ? request
@@ -53,7 +54,7 @@ function applyHeader(
      ...request,
      prompt: header.prompt,
      requestConfig: header.prompt.config,
-      ...(header.change === undefined ? {} : { promptChange: header.change }),
+      ...(includeChange && header.change !== undefined ? { promptChange: header.change } : {}),
    }
 }

@@ -150,6 +151,7 @@ export class TrajectorySnapshotBuilder implements ConversationViewBuilder<
    const boundaries: { seq: number; time: number }[] = []
    const turnEndings: { turn: number; time: number; error?: string }[] = []
    const callSchemas = new Map<string, ConversationPromptSnapshot['tools'][number]>()
+    const consumedPromptChanges = new Set<number>()
    let partial: TrajectorySnapshot['partial'] = null
    const runningCalls: TrajectorySnapshot['runningCalls'][number][] = []

@@ -163,7 +165,12 @@ export class TrajectorySnapshotBuilder implements ConversationViewBuilder<
        const header = data.request === undefined ? undefined : headerFor(data.request, headers)
        if (data.node !== undefined) finalized.push(withRequestConfig(data.node, header?.prompt))
        if (data.partial !== null) partial = data.partial
-        if (data.request !== undefined) requests.push(applyHeader(data.request, header))
+        if (data.request !== undefined) {
+          const includeChange = header?.change !== undefined
+            && !consumedPromptChanges.has(header.seq)
+          requests.push(applyHeader(data.request, header, includeChange))
+          if (includeChange) consumedPromptChanges.add(header.seq)
+        }
        continue
      }
      if (data.kind === 'tool') {