Skip to content

Commit a259d22

Browse files
Missing-Identitytombeckenham
authored andcommitted
fix: address Gemini structured stream review
1 parent 9e5da37 commit a259d22

3 files changed

Lines changed: 7 additions & 3 deletions

File tree

packages/ai-gemini/src/adapters/text.ts

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -226,7 +226,11 @@ export class GeminiTextAdapter<
226226
async *structuredOutputStream(
227227
options: StructuredOutputOptions<GeminiTextProviderOptions>,
228228
): AsyncIterable<StreamChunk> {
229-
const { chatOptions, outputSchema } = options
229+
const { chatOptions: requestedChatOptions, outputSchema } = options
230+
const chatOptions = {
231+
...requestedChatOptions,
232+
runId: requestedChatOptions.runId ?? generateId(this.name),
233+
}
230234
const mappedOptions = this.mapCommonOptionsToGemini(chatOptions)
231235

232236
try {

packages/ai-gemini/tests/gemini-adapter.test.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import { describe, it, expect, beforeEach, vi } from 'vitest'
22
import { z } from 'zod'
3-
import { chat, EventType, summarize } from '@tanstack/ai'
3+
import { EventType, chat, summarize } from '@tanstack/ai'
44
import type { Tool, StreamChunk } from '@tanstack/ai'
55
import {
66
Type,

packages/ai-gemini/tests/text-interactions-adapter.test.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import { beforeEach, describe, expect, it, vi } from 'vitest'
22
import { z } from 'zod'
3-
import { chat, EventType } from '@tanstack/ai'
3+
import { EventType, chat } from '@tanstack/ai'
44
import type { StreamChunk, Tool } from '@tanstack/ai'
55
import { GeminiTextInteractionsAdapter } from '../src/experimental/text-interactions/adapter'
66
import type { GeminiTextInteractionsProviderOptions } from '../src/experimental/text-interactions/adapter'

0 commit comments

Comments
 (0)