({
messages,
systemPrompt,
thinkingConfig,
tools,
signal,
options,
}: {
messages: Message[]
systemPrompt: SystemPrompt
thinkingConfig: ThinkingConfig
tools: Tools
signal: AbortSignal
options: Options
})
| 750 | } |
| 751 | |
| 752 | export async function* queryModelWithStreaming({ |
| 753 | messages, |
| 754 | systemPrompt, |
| 755 | thinkingConfig, |
| 756 | tools, |
| 757 | signal, |
| 758 | options, |
| 759 | }: { |
| 760 | messages: Message[] |
| 761 | systemPrompt: SystemPrompt |
| 762 | thinkingConfig: ThinkingConfig |
| 763 | tools: Tools |
| 764 | signal: AbortSignal |
| 765 | options: Options |
| 766 | }): AsyncGenerator< |
| 767 | StreamEvent | AssistantMessage | SystemAPIErrorMessage, |
| 768 | void |
| 769 | > { |
| 770 | return yield* withStreamingVCR(messages, async function* () { |
| 771 | yield* queryModel( |
| 772 | messages, |
| 773 | systemPrompt, |
| 774 | thinkingConfig, |
| 775 | tools, |
| 776 | signal, |
| 777 | options, |
| 778 | ) |
| 779 | }) |
| 780 | } |
| 781 | |
| 782 | /** |
| 783 | * Determines if an LSP tool should be deferred (tool appears with defer_loading: true) |
no test coverage detected