LZH-YS1998
430 lines
13 KiB
TypeScript
430 lines
13 KiB
TypeScript
import assert from 'node:assert/strict'
|
|
|
|
import type { ChatMessage } from '../types/chat'
|
|
import { mapBackendMessage } from '../lib/collabSync'
|
|
import { analyzeCheckpointMessages } from './checkpointUtils'
|
|
import { __chatStoreTestUtils } from './ChatStore'
|
|
|
|
const persistentTimestamps = __chatStoreTestUtils.latestPersistentMessageTimestamps([
|
|
{
|
|
id: 'db-assistant-1',
|
|
channelId: 'session:read-test',
|
|
sender: 'assistant',
|
|
senderName: 'OPC',
|
|
content: 'First persisted reply',
|
|
timestamp: 10,
|
|
mentions: [],
|
|
},
|
|
{
|
|
id: 'msg-local-only',
|
|
channelId: 'session:read-test',
|
|
sender: 'user',
|
|
senderName: 'You',
|
|
content: 'Optimistic message',
|
|
timestamp: 30,
|
|
mentions: [],
|
|
metadata: { ui_message_id: 'ui-local-only' },
|
|
},
|
|
{
|
|
id: 'db-assistant-2',
|
|
channelId: 'session:read-test',
|
|
sender: 'assistant',
|
|
senderName: 'OPC',
|
|
content: 'Latest persisted reply',
|
|
timestamp: 20,
|
|
mentions: [],
|
|
},
|
|
])
|
|
|
|
assert.equal(persistentTimestamps['session:read-test'], 20)
|
|
|
|
const unreadState = { 'session:read-test': 10 }
|
|
const advancedReadState = __chatStoreTestUtils.advanceReadTimestamp(
|
|
unreadState,
|
|
'session:read-test',
|
|
persistentTimestamps['session:read-test'],
|
|
)
|
|
assert.notEqual(advancedReadState, unreadState)
|
|
assert.equal(advancedReadState['session:read-test'], 20)
|
|
assert.equal(
|
|
__chatStoreTestUtils.advanceReadTimestamp(advancedReadState, 'session:read-test', 20),
|
|
advancedReadState,
|
|
'marking an already-read channel must preserve state identity',
|
|
)
|
|
assert.equal(
|
|
__chatStoreTestUtils.advanceReadTimestamp(advancedReadState, 'session:read-test', 15),
|
|
advancedReadState,
|
|
'an older snapshot must not move the read cursor backwards',
|
|
)
|
|
|
|
const syntheticCheckpoint: ChatMessage = {
|
|
id: 'checkpoint::cp-delivery',
|
|
channelId: 'session:task-1',
|
|
sender: 'assistant',
|
|
senderName: 'Company Member',
|
|
content: 'Human review requested.',
|
|
timestamp: 1,
|
|
mentions: [],
|
|
metadata: {
|
|
checkpoint_type: 'company_delivery_feedback',
|
|
checkpoint_id: 'cp-delivery',
|
|
summary: 'Pending review',
|
|
},
|
|
}
|
|
|
|
const backendCheckpointUpdate: ChatMessage = {
|
|
id: 'db-message-1',
|
|
channelId: 'session:task-1',
|
|
sender: 'assistant',
|
|
senderName: 'Company Member',
|
|
content: 'Human review requested.',
|
|
timestamp: 2,
|
|
mentions: [],
|
|
metadata: {
|
|
checkpoint_type: 'company_delivery_feedback',
|
|
checkpoint_id: 'cp-delivery',
|
|
checkpoint_status: 'ignored',
|
|
checkpoint_reply_kind: 'ignore',
|
|
},
|
|
}
|
|
|
|
const mergedCheckpoint = __chatStoreTestUtils.dedupeMessages([
|
|
syntheticCheckpoint,
|
|
backendCheckpointUpdate,
|
|
])
|
|
|
|
assert.equal(mergedCheckpoint.length, 1)
|
|
assert.equal(mergedCheckpoint[0].id, 'db-message-1')
|
|
assert.equal(mergedCheckpoint[0].timestamp, 1, 'checkpoint status updates must keep their original timeline position')
|
|
assert.equal(mergedCheckpoint[0].metadata?.checkpoint_status, 'ignored')
|
|
assert.deepEqual([...analyzeCheckpointMessages(mergedCheckpoint).pendingMessageIds], [])
|
|
assert.deepEqual([...analyzeCheckpointMessages(mergedCheckpoint).respondedMessageIds], ['db-message-1'])
|
|
|
|
const terminalSyntheticCheckpoint: ChatMessage = {
|
|
...syntheticCheckpoint,
|
|
timestamp: 2,
|
|
metadata: {
|
|
...syntheticCheckpoint.metadata,
|
|
checkpoint_status: 'ignored',
|
|
checkpoint_reply_kind: 'ignore',
|
|
},
|
|
}
|
|
|
|
const mergedSameIdCheckpoint = __chatStoreTestUtils.dedupeMessages([
|
|
syntheticCheckpoint,
|
|
terminalSyntheticCheckpoint,
|
|
])
|
|
|
|
assert.equal(mergedSameIdCheckpoint.length, 1)
|
|
assert.equal(mergedSameIdCheckpoint[0].id, 'checkpoint::cp-delivery')
|
|
assert.equal(mergedSameIdCheckpoint[0].metadata?.checkpoint_status, 'ignored')
|
|
assert.deepEqual([...analyzeCheckpointMessages(mergedSameIdCheckpoint).pendingMessageIds], [])
|
|
|
|
const optimisticUserMessage: ChatMessage = {
|
|
id: 'msg-local',
|
|
channelId: 'session:task-1',
|
|
sender: 'user',
|
|
senderName: 'You',
|
|
content: 'New requirement',
|
|
timestamp: 3,
|
|
mentions: [],
|
|
metadata: {
|
|
ui_message_id: 'ui-1',
|
|
},
|
|
}
|
|
|
|
const backendUserMessage: ChatMessage = {
|
|
id: 'db-user-1',
|
|
channelId: 'session:task-1',
|
|
sender: 'user',
|
|
senderName: 'You',
|
|
content: 'New requirement',
|
|
timestamp: 4,
|
|
mentions: [],
|
|
metadata: {
|
|
ui_message_id: 'ui-1',
|
|
},
|
|
}
|
|
|
|
const mergedUserMessage = __chatStoreTestUtils.dedupeMessages([
|
|
optimisticUserMessage,
|
|
backendUserMessage,
|
|
])
|
|
|
|
assert.equal(mergedUserMessage.length, 1)
|
|
assert.equal(mergedUserMessage[0].metadata?.ui_message_id, 'ui-1')
|
|
assert.equal(mergedUserMessage[0].timestamp, 4, 'backend acknowledgement must replace the optimistic client clock')
|
|
|
|
const mirroredUserMessages = __chatStoreTestUtils.dedupeMessages([
|
|
backendUserMessage,
|
|
{ ...backendUserMessage, id: 'db-user-mirror', channelId: 'session:child-task' },
|
|
])
|
|
assert.equal(
|
|
mirroredUserMessages.length,
|
|
2,
|
|
'ui_message_id mirrors in different channels must remain available to each channel projection',
|
|
)
|
|
|
|
assert.deepEqual(
|
|
__chatStoreTestUtils.unreadMessageCounts([
|
|
{
|
|
id: 'msg-local-system',
|
|
channelId: 'session:task-1',
|
|
sender: 'system',
|
|
senderName: 'System',
|
|
content: 'Local task assignment notice',
|
|
timestamp: 100,
|
|
mentions: [],
|
|
},
|
|
{
|
|
id: 'db-assistant-unread',
|
|
channelId: 'session:task-1',
|
|
sender: 'assistant',
|
|
senderName: 'OPC',
|
|
content: 'Persisted reply',
|
|
timestamp: 90,
|
|
mentions: [],
|
|
},
|
|
], {}),
|
|
{ 'session:task-1': 1 },
|
|
'local-only system rows must not become unread entries that markRead can never cover',
|
|
)
|
|
|
|
const nativeCompanyRawTurn: ChatMessage = {
|
|
id: 'native-raw-1',
|
|
channelId: 'session:task-1',
|
|
sender: 'assistant',
|
|
senderName: 'Task Generalist',
|
|
content: '最终分析已经完成，结论如下。',
|
|
timestamp: 5,
|
|
mentions: [],
|
|
metadata: {
|
|
source: 'engine',
|
|
transcript_kind: 'runtime_v2_assistant',
|
|
},
|
|
}
|
|
|
|
const companyRoleResult: ChatMessage = {
|
|
id: 'role-result-1',
|
|
channelId: 'session:task-1',
|
|
sender: 'chao',
|
|
senderName: 'Chao',
|
|
content: '最终分析已经完成，结论如下。',
|
|
timestamp: 6,
|
|
mentions: [],
|
|
metadata: {
|
|
source: 'engine',
|
|
transcript_kind: 'company_role_result',
|
|
},
|
|
}
|
|
|
|
const mergedNativeCompanyDuplicate = __chatStoreTestUtils.dedupeMessages([
|
|
nativeCompanyRawTurn,
|
|
companyRoleResult,
|
|
])
|
|
|
|
assert.equal(mergedNativeCompanyDuplicate.length, 1)
|
|
assert.equal(mergedNativeCompanyDuplicate[0].id, 'role-result-1')
|
|
assert.equal(mergedNativeCompanyDuplicate[0].senderName, 'Chao')
|
|
assert.equal(mergedNativeCompanyDuplicate[0].timestamp, 5, 'semantic result replacement must retain its original timeline position')
|
|
assert.equal(
|
|
mergedNativeCompanyDuplicate[0].metadata?.ui_timeline_id,
|
|
'message:native-raw-1',
|
|
'semantic result replacement must retain the already-mounted row identity',
|
|
)
|
|
const repeatedNativeCompanySync = __chatStoreTestUtils.dedupeMessages([
|
|
...mergedNativeCompanyDuplicate,
|
|
nativeCompanyRawTurn,
|
|
companyRoleResult,
|
|
])
|
|
assert.equal(repeatedNativeCompanySync.length, 1)
|
|
assert.equal(repeatedNativeCompanySync[0].metadata?.ui_timeline_id, 'message:native-raw-1')
|
|
assert.equal(repeatedNativeCompanySync[0].timestamp, 5)
|
|
|
|
const structuredDeliveryMerge = __chatStoreTestUtils.dedupeMessages([
|
|
{
|
|
...nativeCompanyRawTurn,
|
|
content: 'Raw runtime wording before the canonical result wrapper.',
|
|
metadata: {
|
|
...nativeCompanyRawTurn.metadata,
|
|
result_delivery_id: 'result:company-turn-1:attempt:0',
|
|
},
|
|
},
|
|
{
|
|
...companyRoleResult,
|
|
content: 'Canonical committed wording may differ without creating another row.',
|
|
metadata: {
|
|
...companyRoleResult.metadata,
|
|
result_delivery_id: 'result:company-turn-1:attempt:0',
|
|
},
|
|
},
|
|
])
|
|
assert.equal(structuredDeliveryMerge.length, 1)
|
|
assert.equal(structuredDeliveryMerge[0].id, 'role-result-1')
|
|
assert.equal(
|
|
structuredDeliveryMerge[0].content,
|
|
'Canonical committed wording may differ without creating another row.',
|
|
)
|
|
|
|
const deliveryMirrorWithLongerWrapper = __chatStoreTestUtils.dedupeMessages([
|
|
{
|
|
...nativeCompanyRawTurn,
|
|
content: 'Runtime wrapper that must not outrank the committed surface.\n\nCanonical result body.',
|
|
metadata: {
|
|
...nativeCompanyRawTurn.metadata,
|
|
result_delivery_id: 'result:company-turn-wrapper:attempt:0',
|
|
},
|
|
},
|
|
{
|
|
...companyRoleResult,
|
|
content: 'Canonical result body.',
|
|
metadata: {
|
|
...companyRoleResult.metadata,
|
|
result_delivery_id: 'result:company-turn-wrapper:attempt:0',
|
|
},
|
|
},
|
|
])
|
|
assert.equal(deliveryMirrorWithLongerWrapper.length, 1)
|
|
assert.equal(
|
|
deliveryMirrorWithLongerWrapper[0].content,
|
|
'Canonical result body.',
|
|
'delivery mirrors follow surface authority; suffix repair only applies to one persistent message id',
|
|
)
|
|
|
|
const ctoDispatchContent = `Both work items have been successfully dispatched to my senior engineer. Here's the status:
|
|
|
|
## Dispatch Summary
|
|
|
|
**Work Item 1: OpenOPC Source Code Architecture Deep-Dive Analysis**
|
|
- ID: \`1ed5f5f1-ac41-49a1-b1fa-23bbc9adab82\`
|
|
- Owner: senior_engineer
|
|
- Scope: \`openopc-source-analysis\`
|
|
- Output: \`/data2/bjdwhzzh/project-hku/OpenOPC_workplace/0009/openopc-architecture-analysis.md\`
|
|
- Covers: Layered architecture, the work-item state machine, collaboration policy, and seat executor mechanisms.
|
|
|
|
**Work Item 2: External Multi-Agent Frameworks Architecture Research**
|
|
- ID: \`d0307208-6b95-44c1-9b51-6bf073bbdcef\`
|
|
- Owner: senior_engineer
|
|
|
|
Both work items are independent and can execute in parallel.`
|
|
|
|
const ctoCompanyFinal: ChatMessage = {
|
|
id: 'runtime-v2-company-assistant-final:cto-turn-9',
|
|
channelId: 'session:cto-work-item',
|
|
sender: 'cto',
|
|
senderName: 'CTO',
|
|
content: ctoDispatchContent,
|
|
timestamp: 20,
|
|
mentions: [],
|
|
metadata: {
|
|
source: 'engine',
|
|
transcript_kind: 'runtime_v2_company_assistant',
|
|
canonical_turn_id: 'cto-turn-9',
|
|
ui_message_id: 'runtime-v2-company-assistant-final:cto-turn-9',
|
|
},
|
|
}
|
|
|
|
const repeatedCtoSnapshot = __chatStoreTestUtils.dedupeMessages([
|
|
ctoCompanyFinal,
|
|
{ ...ctoCompanyFinal, metadata: { ...ctoCompanyFinal.metadata } },
|
|
])
|
|
assert.equal(repeatedCtoSnapshot.length, 1)
|
|
assert.equal(
|
|
repeatedCtoSnapshot[0].content,
|
|
ctoDispatchContent,
|
|
'comparison normalization must never peel Work Item, ID, or Owner fields from rendered content',
|
|
)
|
|
|
|
let repeatedCtoMerge = [ctoCompanyFinal]
|
|
for (let replay = 0; replay < 6; replay += 1) {
|
|
repeatedCtoMerge = __chatStoreTestUtils.mergeMessagesIntoExisting(
|
|
repeatedCtoMerge,
|
|
[{ ...ctoCompanyFinal, metadata: { ...ctoCompanyFinal.metadata } }],
|
|
)
|
|
assert.equal(repeatedCtoMerge.length, 1)
|
|
assert.equal(
|
|
repeatedCtoMerge[0].content,
|
|
ctoDispatchContent,
|
|
`identical MERGE replay ${replay + 1} must preserve the complete source text`,
|
|
)
|
|
}
|
|
|
|
const truncatedCtoContent = ctoDispatchContent.slice(
|
|
ctoDispatchContent.indexOf('OpenOPC Source Code Architecture Deep-Dive Analysis**'),
|
|
)
|
|
const truncatedCtoFinal: ChatMessage = {
|
|
...ctoCompanyFinal,
|
|
content: truncatedCtoContent,
|
|
}
|
|
|
|
let interleavedCtoReplay = [ctoCompanyFinal]
|
|
for (const replayedMessage of [
|
|
truncatedCtoFinal,
|
|
ctoCompanyFinal,
|
|
truncatedCtoFinal,
|
|
ctoCompanyFinal,
|
|
]) {
|
|
interleavedCtoReplay = __chatStoreTestUtils.mergeMessagesIntoExisting(
|
|
interleavedCtoReplay,
|
|
[{ ...replayedMessage, metadata: { ...replayedMessage.metadata } }],
|
|
)
|
|
assert.equal(interleavedCtoReplay.length, 1)
|
|
assert.equal(
|
|
interleavedCtoReplay[0].content,
|
|
ctoDispatchContent,
|
|
'a same-identity truncated cache replay must neither replace nor duplicate the complete message',
|
|
)
|
|
}
|
|
|
|
const repairedCtoReplay = __chatStoreTestUtils.mergeMessagesIntoExisting(
|
|
[truncatedCtoFinal],
|
|
[ctoCompanyFinal],
|
|
)
|
|
assert.equal(repairedCtoReplay.length, 1)
|
|
assert.equal(repairedCtoReplay[0].content, ctoDispatchContent)
|
|
assert.equal(
|
|
__chatStoreTestUtils.mergeMessagesIntoExisting(repairedCtoReplay, [truncatedCtoFinal])[0].content,
|
|
ctoDispatchContent,
|
|
'once a complete same-identity source arrives, later truncated replays must not regress it',
|
|
)
|
|
|
|
const mountedHighPriorityResult: ChatMessage = {
|
|
...companyRoleResult,
|
|
id: 'mounted-high-result',
|
|
timestamp: 10,
|
|
metadata: { source: 'engine', transcript_kind: 'child_task_result' },
|
|
}
|
|
const olderLowPrioritySurface: ChatMessage = {
|
|
...nativeCompanyRawTurn,
|
|
id: 'older-low-result',
|
|
timestamp: 4,
|
|
}
|
|
const historyBackfillMerge = __chatStoreTestUtils.mergeMessagesIntoExisting(
|
|
[mountedHighPriorityResult],
|
|
[olderLowPrioritySurface],
|
|
)
|
|
assert.equal(historyBackfillMerge.length, 1)
|
|
assert.equal(historyBackfillMerge[0].id, 'mounted-high-result')
|
|
assert.equal(historyBackfillMerge[0].timestamp, 10, 'history backfill must not move an already-mounted result row')
|
|
assert.equal(
|
|
historyBackfillMerge[0].metadata?.ui_timeline_id,
|
|
'message:mounted-high-result',
|
|
'history backfill must retain the mounted high-priority result key',
|
|
)
|
|
|
|
const mappedTaskGeneralistMessage = mapBackendMessage({
|
|
message_id: 'legacy-task-generalist',
|
|
channel_id: 'session:task-1',
|
|
sender: 'task_generalist',
|
|
sender_name: 'Task Generalist',
|
|
content: 'Legacy native task result.',
|
|
created_at: 10,
|
|
metadata: {
|
|
transcript_kind: 'runtime_v2_company_assistant',
|
|
},
|
|
})
|
|
|
|
assert.equal(mappedTaskGeneralistMessage.senderName, 'OPC')
|
|
|
|
console.log('ChatStore.test.ts: OK (read cursors, optimistic, checkpoint, and company result identity merging)')
|