refactor: identify and freeze messages at creation

packages/session-query/tool-session-query/tests/sqlite-integration.spec.ts

@@ -4,7 +4,7 @@ import { mkdtemp, rm } from 'node:fs/promises'
 import { tmpdir } from 'node:os'
 import { join } from 'node:path'
 import type { Agent } from '@deepseek-ai/dsh-agent'
-import { CallId } from '@deepseek-ai/dsh-llm'
+import { createUserMessage, CallId  } from '@deepseek-ai/dsh-llm'
 import SessionStore, {
   SESSION_FORMAT_VERSION,
   SessionId,
@@ -54,10 +54,10 @@ describe('tool-session-query with the real SQLite provider', () => {
       type: 'user/message',
       seq: 0,
       time: 2,
-      data: {
+      data: createUserMessage({
         content: [{ type: 'text', text: 'persisted integration needle' }],
         source: { kind: 'user' },
-      },
+      }),
       surfaceOp: 'append',
     }])
 
@@ -67,7 +67,9 @@ describe('tool-session-query with the real SQLite provider', () => {
     caller.append('turn/start', { turn: 1, trigger: { kind: 'message', source: { kind: 'user' } } })
     caller.append(
       'user/message',
-      { content: [{ type: 'text', text: 'live integration needle' }], source: { kind: 'user' } },
+      createUserMessage({
+        content: [{ type: 'text', text: 'live integration needle' }], source: { kind: 'user' },
+      }),
       { surfaceOp: 'append' },
     )
     caller.append('step/start', { turn: 1, step: 1 })
@@ -123,40 +125,40 @@ describe('tool-session-query with the real SQLite provider', () => {
         type: 'user/message',
         seq: 0,
         time: base + 123,
-        data: {
+        data: createUserMessage({
           content: [{ type: 'text', text: 'fractional integration needle' }],
           source: { kind: 'user' },
-        },
+        }),
         surfaceOp: 'append',
       },
       {
         type: 'user/message',
         seq: 1,
         time: base + 124,
-        data: {
+        data: createUserMessage({
           content: [{ type: 'text', text: 'fractional integration needle' }],
           source: { kind: 'user' },
-        },
+        }),
         surfaceOp: 'append',
       },
       {
         type: 'user/message',
         seq: 2,
         time: -124,
-        data: {
+        data: createUserMessage({
           content: [{ type: 'text', text: 'pre-epoch fractional needle' }],
           source: { kind: 'user' },
-        },
+        }),
         surfaceOp: 'append',
       },
       {
         type: 'user/message',
         seq: 3,
         time: -123,
-        data: {
+        data: createUserMessage({
           content: [{ type: 'text', text: 'pre-epoch fractional needle' }],
           source: { kind: 'user' },
-        },
+        }),
         surfaceOp: 'append',
       },
     ])

packages/session-query/tool-session-query/tests/tool-session-query.spec.ts

@@ -1,7 +1,7 @@
 import { afterEach, describe, expect, it, vi } from 'vitest'
 import { Context, type Fiber } from 'cordis'
 import type { Agent } from '@deepseek-ai/dsh-agent'
-import { CallId, HarnessError } from '@deepseek-ai/dsh-llm'
+import { createUserMessage, CallId, HarnessError , createMessage } from '@deepseek-ai/dsh-llm'
 import { MAX_TIMER_DELAY_MS, TimeoutReason } from '@deepseek-ai/dsh-timeout'
 import * as TimeoutPolicy from '@deepseek-ai/dsh-timeout-policy'
 import SessionStore, {
@@ -67,7 +67,9 @@ function openStep(session: Session, text = 'prior needle'): void {
   session.append('turn/start', { turn: 1, trigger: { kind: 'message', source: { kind: 'user' } } })
   session.append(
     'user/message',
-    { content: [{ type: 'text', text }], source: { kind: 'user' } },
+    createUserMessage({
+      content: [{ type: 'text', text }], source: { kind: 'user' },
+    }),
     { surfaceOp: 'append' },
   )
   session.append('step/start', { turn: 1, step: 1 })
@@ -871,7 +873,9 @@ describe('workspace authority and lineage redaction', () => {
     const target = createSession(mounted.ctx, `${toolName}-failure-target`, '/work')
     target.append(
       'user/message',
-      { content: [{ type: 'text', text: 'event' }], source: { kind: 'user' } },
+      createUserMessage({
+        content: [{ type: 'text', text: 'event' }], source: { kind: 'user' },
+      }),
       { surfaceOp: 'append' },
     )
     const secret = `event missing beside hidden-${toolName}-secret`
@@ -900,7 +904,9 @@ describe('workspace authority and lineage redaction', () => {
     const target = createSession(mounted.ctx, `cancelled-${toolName}`, '/work')
     target.append(
       'user/message',
-      { content: [{ type: 'text', text: 'pending exact read' }], source: { kind: 'user' } },
+      createUserMessage({
+        content: [{ type: 'text', text: 'pending exact read' }], source: { kind: 'user' },
+      }),
       { surfaceOp: 'append' },
     )
     const controller = new AbortController()
@@ -1130,7 +1136,9 @@ describe('workspace authority and lineage redaction', () => {
     const target = createSession(mounted.ctx, 'moving-target', '/work')
     target.append(
       'user/message',
-      { content: [{ type: 'text', text: 'authorized payload' }], source: { kind: 'user' } },
+      createUserMessage({
+        content: [{ type: 'text', text: 'authorized payload' }], source: { kind: 'user' },
+      }),
       { surfaceOp: 'append' },
     )
     const movedHeader = header(target.id, '/outside')
@@ -1947,7 +1955,9 @@ describe('trace and exact read rendering', () => {
     const session = createSession(mounted.ctx, 'relationships', '/work')
     session.append(
       'user/message',
-      { content: [{ type: 'text', text: 'source' }], source: { kind: 'user' } },
+      createUserMessage({
+        content: [{ type: 'text', text: 'source' }], source: { kind: 'user' },
+      }),
       { surfaceOp: 'append' },
     )
     session.append(
@@ -1955,8 +1965,14 @@ describe('trace and exact read rendering', () => {
       {
         turn: 1,
         step: 1,
-        content: [{ type: 'text', text: 'replacement' }],
-        provenance: { provider: 'test', model: 'test' },
+        message: createMessage({
+          role: 'assistant',
+          content: [{ type: 'text', text: 'replacement' }],
+          source: {
+            kind: 'model',
+            ...{ provider: 'test', model: 'test' },
+          },
+        }),
       },
       { surfaceOp: { op: 'replace', start: 0, end: 0 }, sourceEventSeqs: [0] },
     )
@@ -1971,7 +1987,9 @@ describe('trace and exact read rendering', () => {
     const session = createSession(mounted.ctx, 'read', '/work')
     session.append(
       'user/message',
-      { content: [{ type: 'text', text: 'before semantic text' }], source: { kind: 'user' } },
+      createUserMessage({
+        content: [{ type: 'text', text: 'before semantic text' }], source: { kind: 'user' },
+      }),
       { surfaceOp: 'append' },
     )
     session.append(
@@ -1979,8 +1997,14 @@ describe('trace and exact read rendering', () => {
       {
         turn: 1,
         step: 1,
-        content: [{ type: 'text', text: 'target full text' }],
-        provenance: { provider: 'test', model: 'test' },
+        message: createMessage({
+          role: 'assistant',
+          content: [{ type: 'text', text: 'target full text' }],
+          source: {
+            kind: 'model',
+            ...{ provider: 'test', model: 'test' },
+          },
+        }),
       },
       { surfaceOp: 'append' },
     )