| 12 | import type { TokenUsage } from '#/usage'; |
| 13 | import { describe, expect, it } from 'vitest'; |
| 14 | function createMockStream( |
| 15 | parts: StreamedMessagePart[], |
| 16 | opts?: { id?: string; usage?: TokenUsage }, |
| 17 | ): StreamedMessage { |
| 18 | return { |
| 19 | get id(): string | null { |
| 20 | return opts?.id ?? null; |
| 21 | }, |
| 22 | get usage(): TokenUsage | null { |
| 23 | return opts?.usage ?? null; |
| 24 | }, |
| 25 | finishReason: null, |
| 26 | rawFinishReason: null, |
| 27 | async *[Symbol.asyncIterator](): AsyncIterator<StreamedMessagePart> { |
| 28 | for (const part of parts) { |
| 29 | yield part; |
| 30 | } |
| 31 | }, |
| 32 | }; |
| 33 | } |
| 34 | |
| 35 | function createMockProvider(stream: StreamedMessage): ChatProvider { |
| 36 | return { |