36 lines
1.4 KiB
JavaScript
36 lines
1.4 KiB
JavaScript
|
import assert from 'node:assert/strict';
|
||
|
|
import test from 'node:test';
|
||
|
|
import { ReadableStream } from 'node:stream/web';
|
||
|
|
import {
|
||
|
|
consumeSessionEventsUntilFinish,
|
||
|
|
eventMatchesRequest,
|
||
|
|
parseSessionStreamEvent,
|
||
|
|
} from './session-reply-wait.mjs';
|
||
|
|
|
||
|
|
test('parseSessionStreamEvent parses SSE data payload', () => {
|
||
|
|
const event = parseSessionStreamEvent('id: 1\ndata: {"type":"Finish","token_state":null}\n');
|
||
|
|
assert.equal(event?.type, 'Finish');
|
||
|
|
});
|
||
|
|
|
||
|
|
test('eventMatchesRequest scopes by request id when present', () => {
|
||
|
|
assert.equal(eventMatchesRequest({ request_id: 'req-1' }, 'req-1'), true);
|
||
|
|
assert.equal(eventMatchesRequest({ request_id: 'req-2' }, 'req-1'), false);
|
||
|
|
assert.equal(eventMatchesRequest({ type: 'Finish' }, 'req-1'), true);
|
||
|
|
});
|
||
|
|
|
||
|
|
test('consumeSessionEventsUntilFinish resolves on Finish', async () => {
|
||
|
|
const frames = [
|
||
|
|
'data: {"type":"Message","request_id":"req-1","message":{"role":"assistant"}}\n\n',
|
||
|
|
'data: {"type":"Finish","request_id":"req-1","token_state":{"totalTokens":12}}\n\n',
|
||
|
|
];
|
||
|
|
const stream = new ReadableStream({
|
||
|
|
start(controller) {
|
||
|
|
for (const frame of frames) controller.enqueue(new TextEncoder().encode(frame));
|
||
|
|
controller.close();
|
||
|
|
},
|
||
|
|
});
|
||
|
|
const result = await consumeSessionEventsUntilFinish(stream, { requestId: 'req-1', timeoutMs: 5000 });
|
||
|
|
assert.equal(result.finishEvent.type, 'Finish');
|
||
|
|
assert.equal(result.tokenState.totalTokens, 12);
|
||
|
|
});
|