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