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