58 lines
2.1 KiB
TypeScript
58 lines
2.1 KiB
TypeScript
import { mkdtemp, rm } from 'node:fs/promises'
|
|
import { tmpdir } from 'node:os'
|
|
import { join } from 'node:path'
|
|
import { afterEach, describe, expect, it } from 'vitest'
|
|
import { Context } from 'cordis'
|
|
import SessionStore, { SessionId } from '@deepseek-ai/dsh-session'
|
|
import SessionPersistenceJsonl from '@deepseek-ai/dsh-session-persistence-jsonl'
|
|
import SessionPersistenceSqlite from '@deepseek-ai/dsh-session-persistence-sqlite'
|
|
import type {} from '../src/index.ts'
|
|
|
|
const dirs: string[] = []
|
|
|
|
afterEach(async () => {
|
|
for (const dir of dirs.splice(0)) await rm(dir, { recursive: true, force: true })
|
|
})
|
|
|
|
async function backend(kind: 'jsonl' | 'sqlite'): Promise<Context> {
|
|
const ctx = new Context()
|
|
await ctx.plugin(SessionStore)
|
|
if (kind === 'jsonl') {
|
|
const root = await mkdtemp(join(tmpdir(), 'dsh-llm-retry-jsonl-'))
|
|
dirs.push(root)
|
|
await ctx.plugin(SessionPersistenceJsonl, { root })
|
|
} else {
|
|
await ctx.plugin(SessionPersistenceSqlite, { path: ':memory:' })
|
|
}
|
|
return ctx
|
|
}
|
|
|
|
describe.each(['jsonl', 'sqlite'] as const)('%s retry-event persistence', (kind) => {
|
|
it('round-trips the event losslessly without adding a model message', async () => {
|
|
const ctx = await backend(kind)
|
|
try {
|
|
const session = ctx.sessions.create(SessionId(`retry-${kind}`))
|
|
session.append('turn/start', { turn: 1, trigger: { kind: 'message', source: { kind: 'user' } } })
|
|
session.append('step/start', { turn: 1, step: 1 })
|
|
session.append('step/end', { turn: 1, step: 1 })
|
|
const event = session.append('llm/retry', {
|
|
turn: 1,
|
|
step: 1,
|
|
retry: 1,
|
|
maxRetries: 2,
|
|
delayMs: 750,
|
|
failure: { message: 'provider busy', code: 'RATE_LIMIT', status: 429 },
|
|
})
|
|
session.append('turn/end', { turn: 1, reason: { kind: 'aborted', reason: 'cancelled in backoff' } })
|
|
|
|
expect(session.deriveMessages()).toEqual([])
|
|
await ctx.sessions.flush(session)
|
|
const loaded = await ctx.sessionPersistence.load(session.id)
|
|
|
|
expect(loaded.events.find(item => item.type === 'llm/retry')).toEqual(event)
|
|
} finally {
|
|
await ctx.fiber.dispose()
|
|
}
|
|
})
|
|
})
|