Turtle
bec6805d6a
The persistence seam is now create/open/stat/list returning per-session SessionHandles (read/append/flush/close); every log read and write flows through the owning handle. The seam package exports only the service and handle contracts, consumer-visible errors, and pure durable-data validation helpers; each backend owns its complete storage runtime, and the shared contract suites pin equivalent observable behavior. The backend routes published sessions' live events by id into the active write handle; agent-loop only acquires, seeds, and closes the handle. Resume appends interruptedTurnClosers through its write handle; session-query owns the revision-keyed cold cache. Legacy-only surfaces are removed in the same swap: locate/readRaw/supportsRawArtifacts, the legacy event-shape read migration, zstd torn-frame salvage, DSH_SESSION_JSONL, and hook transcript_path population; a torn final zstd frame is discarded whole; the session-list cold blank probe returns on stat metadata (eventCount derived from the last physical row, sizeBytes). The WebUI ZIP export serializes the logical log from a read handle, so both backends export identically. Refs #3245
84 lines
2.8 KiB
TypeScript
84 lines
2.8 KiB
TypeScript
import { createUserMessage } from '@deepseek-ai/dsh-llm'
|
|
import { afterEach, describe, expect, it } from 'vitest'
|
|
import { Context } from '@deepseek-ai/cordis'
|
|
import { mkdtemp, rm } from 'node:fs/promises'
|
|
import { tmpdir } from 'node:os'
|
|
import { join } from 'node:path'
|
|
import SessionStore, { SessionId } from '@deepseek-ai/dsh-session'
|
|
import SessionProjectionRegistry from '@deepseek-ai/dsh-session-projection'
|
|
import JsonlSessionPersistence from '@deepseek-ai/dsh-session-persistence-jsonl'
|
|
import SessionTitleService, { foldSessionTitle } from '@deepseek-ai/dsh-session-title'
|
|
|
|
const CONFIG = {
|
|
fallbackMaxWords: 5,
|
|
fallbackMaxBytes: 40,
|
|
maxTitleBytes: 80,
|
|
} as const
|
|
|
|
const roots: string[] = []
|
|
|
|
afterEach(async () => {
|
|
for (const root of roots.splice(0)) await rm(root, { recursive: true, force: true })
|
|
})
|
|
|
|
async function appendPersistedTitle(ctx: Context, id: ReturnType<typeof SessionId>): Promise<void> {
|
|
const session = ctx.sessions.create(id)
|
|
const handle = await ctx.sessionPersistence.create(session.header)
|
|
try {
|
|
session.append('turn/start', {
|
|
turn: 1,
|
|
})
|
|
session.append('user/message', createUserMessage({
|
|
content: [{ type: 'text', text: 'Persist this session title' }],
|
|
source: { kind: 'user' },
|
|
}), { surfaceOp: 'append' })
|
|
session.append('turn/end', { turn: 1, reason: { kind: 'completed' } })
|
|
await ctx.sessionTitle.refresh(session)
|
|
} finally {
|
|
await handle.close()
|
|
}
|
|
}
|
|
|
|
async function expectPersistedTitle(ctx: Context, id: ReturnType<typeof SessionId>): Promise<void> {
|
|
const handle = await ctx.sessionPersistence.open(id, 'read')
|
|
try {
|
|
const events = await handle.read()
|
|
expect(foldSessionTitle(events)).toMatchObject({
|
|
title: 'Persist this session title',
|
|
messageSeqs: [1],
|
|
source: { kind: 'fallback' },
|
|
eventSeq: 3,
|
|
})
|
|
expect(events.map(event => event.type)).toEqual([
|
|
'turn/start',
|
|
'user/message',
|
|
'turn/end',
|
|
'session/title',
|
|
])
|
|
} finally {
|
|
await handle.close()
|
|
}
|
|
}
|
|
|
|
describe('session title persistence round trips', () => {
|
|
it('round-trips through a remounted JSONL backend', async () => {
|
|
const root = await mkdtemp(join(tmpdir(), 'dsh-title-jsonl-'))
|
|
roots.push(root)
|
|
const id = SessionId('title-jsonl')
|
|
const writer = new Context()
|
|
await writer.plugin(SessionStore)
|
|
await writer.plugin(SessionProjectionRegistry)
|
|
await writer.plugin(JsonlSessionPersistence, { root, compression: 'none' })
|
|
await writer.plugin(SessionTitleService, CONFIG)
|
|
await appendPersistedTitle(writer, id)
|
|
await writer.fiber.dispose()
|
|
|
|
const reader = new Context()
|
|
await reader.plugin(SessionStore)
|
|
await reader.plugin(SessionProjectionRegistry)
|
|
await reader.plugin(JsonlSessionPersistence, { root, compression: 'none' })
|
|
await expectPersistedTitle(reader, id)
|
|
await reader.fiber.dispose()
|
|
})
|
|
})
|