deepseek-harness/packages/api/session-controller/tests/assistant-stream.host.spec.ts

import { describe, expect, it } from 'vitest'
import { LlmAttemptId } from '@deepseek-ai/dsh-llm'
import { SessionAssistantStreamAccumulator } from '../src/assistant-stream.ts'

describe('SessionAssistantStreamAccumulator', () => {
  it('replaces stale lifecycles, rejects frame gaps, and caches each baseline', () => {
    const accumulator = new SessionAssistantStreamAccumulator()
    const empty = accumulator.snapshot()
    expect(accumulator.snapshot()).toBe(empty)

    accumulator.accept({
      type: 'start', attemptId: LlmAttemptId('stale'), revision: 2,
      startedTime: 1, turn: 1, step: 1,
    })
    expect(accumulator.snapshot()).toEqual({ revision: 2 })

    accumulator.accept({
      type: 'start', attemptId: LlmAttemptId('current'), revision: 1,
      startedTime: 2, turn: 2, step: 3,
    })
    expect(accumulator.snapshot()).toMatchObject({
      revision: 1,
      activeAttempt: { attemptId: 'current', turn: 2, step: 3, nextIndex: 0, stream: [] },
    })

    accumulator.accept({
      type: 'chunk', attemptId: LlmAttemptId('other'), revision: 2, index: 0,
      time: 4, chunk: { type: 'text-delta', index: 0, text: 'lost' },
    })
    expect(accumulator.snapshot()).toEqual({ revision: 2 })

    accumulator.accept({
      type: 'start', attemptId: LlmAttemptId('settled'), revision: 3,
      startedTime: 3, turn: 2, step: 4,
    })
    accumulator.accept({
      type: 'chunk', attemptId: LlmAttemptId('settled'), revision: 4, index: 0,
      time: 5, chunk: { type: 'text-delta', index: 0, text: 'ok' },
    })
    const active = accumulator.snapshot()
    expect(active).toMatchObject({
      revision: 4,
      activeAttempt: {
        attemptId: 'settled', nextIndex: 1,
        stream: [{ type: 'text-chunks', time0: 5, index: 0, dt: [], texts: ['ok'] }],
      },
    })
    expect(accumulator.snapshot()).toBe(active)

    accumulator.accept({
      type: 'end', attemptId: LlmAttemptId('settled'), revision: 5, index: 1,
      outcome: { kind: 'abandoned' },
    })
    expect(accumulator.snapshot()).toEqual({ revision: 5 })
  })
})