@tanstack/ai-gemini 0.3.2 → 0.4.1

package/README.md

@@ -45,6 +45,7 @@ A powerful, type-safe AI SDK for building AI-powered applications.
 - Headless chat state management with adapters (SSE, HTTP stream, custom)
 - Isomorphic type-safe tools with server/client execution
 - **Enhanced integration with TanStack Start** - Share implementations between AI tools and server functions
+- **Observability events** - Structured, typed events for text, tools, image, speech, transcription, and video ([docs](./docs/guides/observability.md))
 
 ### <a href="https://tanstack.com/ai">Read the docs →</b></a>
 

package/dist/esm/adapters/summarize.d.ts

@@ -1,6 +1,11 @@
-import { GoogleGenAI } from '@google/genai';
+import { GeminiClientConfig } from '../utils.js';
 import { SummarizeAdapter } from '@tanstack/ai/adapters';
 import { StreamChunk, SummarizationOptions, SummarizationResult } from '@tanstack/ai';
+/**
+ * Configuration for Gemini summarize adapter
+ */
+export interface GeminiSummarizeConfig extends GeminiClientConfig {
+}
 /**
  * Available Gemini models for summarization
  */
@@ -38,7 +43,7 @@ export declare class GeminiSummarizeAdapter<TModel extends GeminiSummarizeModel>
         providerOptions: GeminiSummarizeProviderOptions;
     };
     private client;
-    constructor(apiKeyOrClient: string | GoogleGenAI, model: TModel, _options?: GeminiSummarizeAdapterOptions);
+    constructor(config: GeminiSummarizeConfig, model: TModel);
     summarize(options: SummarizationOptions): Promise<SummarizationResult>;
     summarizeStream(options: SummarizationOptions): AsyncIterable<StreamChunk>;
     private getFormatInstructions;
@@ -46,8 +51,8 @@ export declare class GeminiSummarizeAdapter<TModel extends GeminiSummarizeModel>
 /**
  * Creates a Gemini summarize adapter with explicit API key and model
  */
-export declare function createGeminiSummarize<TModel extends GeminiSummarizeModel>(apiKey: string, model: TModel, options?: GeminiSummarizeAdapterOptions): GeminiSummarizeAdapter<TModel>;
+export declare function createGeminiSummarize<TModel extends GeminiSummarizeModel>(apiKey: string, model: TModel, config?: Omit<GeminiSummarizeConfig, 'apiKey'>): GeminiSummarizeAdapter<TModel>;
 /**
  * Creates a Gemini summarize adapter with API key from environment and required model
  */
-export declare function geminiSummarize<TModel extends GeminiSummarizeModel>(model: TModel, options?: GeminiSummarizeAdapterOptions): GeminiSummarizeAdapter<TModel>;
+export declare function geminiSummarize<TModel extends GeminiSummarizeModel>(model: TModel, config?: Omit<GeminiSummarizeConfig, 'apiKey'>): GeminiSummarizeAdapter<TModel>;

package/dist/esm/adapters/summarize.js

@@ -7,10 +7,10 @@ const GeminiSummarizeModels = [
   "gemini-2.0-flash-lite"
 ];
 class GeminiSummarizeAdapter {
-  constructor(apiKeyOrClient, model, _options = {}) {
+  constructor(config, model) {
     this.kind = "summarize";
     this.name = "gemini";
-    this.client = typeof apiKeyOrClient === "string" ? createGeminiClient({ apiKey: apiKeyOrClient }) : apiKeyOrClient;
+    this.client = createGeminiClient(config);
     this.model = model;
   }
   async summarize(options) {
@@ -83,13 +83,12 @@ ${options.text}` }
           if (part.text) {
             accumulatedContent += part.text;
             yield {
-              type: "content",
-              id,
+              type: "TEXT_MESSAGE_CONTENT",
+              messageId: id,
               model,
               timestamp: Date.now(),
               delta: part.text,
-              content: accumulatedContent,
-              role: "assistant"
+              content: accumulatedContent
             };
           }
         }
@@ -97,8 +96,8 @@ ${options.text}` }
       const finishReason = chunk.candidates?.[0]?.finishReason;
       if (finishReason === FinishReason.STOP || finishReason === FinishReason.MAX_TOKENS || finishReason === FinishReason.SAFETY) {
         yield {
-          type: "done",
-          id,
+          type: "RUN_FINISHED",
+          runId: id,
           model,
           timestamp: Date.now(),
           finishReason: finishReason === FinishReason.STOP ? "stop" : finishReason === FinishReason.MAX_TOKENS ? "length" : "content_filter",
@@ -123,12 +122,12 @@ ${options.text}` }
     }
   }
 }
-function createGeminiSummarize(apiKey, model, options) {
-  return new GeminiSummarizeAdapter(apiKey, model, options);
+function createGeminiSummarize(apiKey, model, config) {
+  return new GeminiSummarizeAdapter({ ...config, apiKey }, model);
 }
-function geminiSummarize(model, options) {
+function geminiSummarize(model, config) {
   const apiKey = getGeminiApiKeyFromEnv();
-  return new GeminiSummarizeAdapter(apiKey, model, options);
+  return new GeminiSummarizeAdapter({ ...config, apiKey }, model);
 }
 export {
   GeminiSummarizeAdapter,

package/dist/esm/adapters/summarize.js.map

@@ -1 +1 @@
-{"version":3,"file":"summarize.js","sources":["../../../src/adapters/summarize.ts"],"sourcesContent":["import { FinishReason } from '@google/genai'\nimport {\n  createGeminiClient,\n  generateId,\n  getGeminiApiKeyFromEnv,\n} from '../utils'\n\nimport type { GoogleGenAI } from '@google/genai'\nimport type { SummarizeAdapter } from '@tanstack/ai/adapters'\nimport type {\n  StreamChunk,\n  SummarizationOptions,\n  SummarizationResult,\n} from '@tanstack/ai'\n\n/**\n * Available Gemini models for summarization\n */\nexport const GeminiSummarizeModels = [\n  'gemini-2.0-flash',\n  'gemini-1.5-flash',\n  'gemini-1.5-pro',\n  'gemini-2.0-flash-lite',\n] as const\n\nexport type GeminiSummarizeModel = (typeof GeminiSummarizeModels)[number]\n\n/**\n * Provider-specific options for Gemini summarization\n */\nexport interface GeminiSummarizeProviderOptions {\n  /** Generation configuration */\n  generationConfig?: {\n    temperature?: number\n    topP?: number\n    topK?: number\n    maxOutputTokens?: number\n    stopSequences?: Array<string>\n  }\n  /** Safety settings */\n  safetySettings?: Array<{\n    category: string\n    threshold: string\n  }>\n}\n\nexport interface GeminiSummarizeAdapterOptions {\n  // Additional adapter options can be added here\n}\n\n/**\n * Gemini Summarize Adapter\n * A tree-shakeable summarization adapter for Google Gemini\n */\nexport class GeminiSummarizeAdapter<\n  TModel extends GeminiSummarizeModel,\n> implements SummarizeAdapter<TModel, GeminiSummarizeProviderOptions> {\n  readonly kind = 'summarize' as const\n  readonly name = 'gemini' as const\n  readonly model: TModel\n\n  // Type-only property - never assigned at runtime\n  declare '~types': {\n    providerOptions: GeminiSummarizeProviderOptions\n  }\n\n  private client: GoogleGenAI\n\n  constructor(\n    apiKeyOrClient: string | GoogleGenAI,\n    model: TModel,\n    _options: GeminiSummarizeAdapterOptions = {},\n  ) {\n    this.client =\n      typeof apiKeyOrClient === 'string'\n        ? createGeminiClient({ apiKey: apiKeyOrClient })\n        : apiKeyOrClient\n    this.model = model\n  }\n\n  async summarize(options: SummarizationOptions): Promise<SummarizationResult> {\n    const model = options.model\n\n    // Build the system prompt based on format\n    const formatInstructions = this.getFormatInstructions(options.style)\n    const lengthInstructions = options.maxLength\n      ? ` Keep the summary under ${options.maxLength} tokens.`\n      : ''\n\n    const systemPrompt = `You are a helpful assistant that summarizes text. ${formatInstructions}${lengthInstructions}`\n\n    const response = await this.client.models.generateContent({\n      model,\n      contents: [\n        {\n          role: 'user',\n          parts: [\n            { text: `Please summarize the following:\\n\\n${options.text}` },\n          ],\n        },\n      ],\n      config: {\n        systemInstruction: systemPrompt,\n      },\n    })\n\n    const summary = response.text ?? ''\n    const inputTokens = response.usageMetadata?.promptTokenCount ?? 0\n    const outputTokens = response.usageMetadata?.candidatesTokenCount ?? 0\n\n    return {\n      id: generateId('sum'),\n      model,\n      summary,\n      usage: {\n        promptTokens: inputTokens,\n        completionTokens: outputTokens,\n        totalTokens: inputTokens + outputTokens,\n      },\n    }\n  }\n\n  async *summarizeStream(\n    options: SummarizationOptions,\n  ): AsyncIterable<StreamChunk> {\n    const model = options.model\n    const id = generateId('sum')\n    let accumulatedContent = ''\n    let inputTokens = 0\n    let outputTokens = 0\n\n    // Build the system prompt based on format\n    const formatInstructions = this.getFormatInstructions(options.style)\n    const lengthInstructions = options.maxLength\n      ? ` Keep the summary under ${options.maxLength} words.`\n      : ''\n\n    const systemPrompt = `You are a helpful assistant that summarizes text. ${formatInstructions}${lengthInstructions}`\n\n    const result = await this.client.models.generateContentStream({\n      model,\n      contents: [\n        {\n          role: 'user',\n          parts: [\n            { text: `Please summarize the following:\\n\\n${options.text}` },\n          ],\n        },\n      ],\n      config: {\n        systemInstruction: systemPrompt,\n      },\n    })\n\n    for await (const chunk of result) {\n      // Track usage metadata\n      if (chunk.usageMetadata) {\n        inputTokens = chunk.usageMetadata.promptTokenCount ?? inputTokens\n        outputTokens = chunk.usageMetadata.candidatesTokenCount ?? outputTokens\n      }\n\n      if (chunk.candidates?.[0]?.content?.parts) {\n        for (const part of chunk.candidates[0].content.parts) {\n          if (part.text) {\n            accumulatedContent += part.text\n            yield {\n              type: 'content',\n              id,\n              model,\n              timestamp: Date.now(),\n              delta: part.text,\n              content: accumulatedContent,\n              role: 'assistant',\n            }\n          }\n        }\n      }\n\n      // Check for finish reason\n      const finishReason = chunk.candidates?.[0]?.finishReason\n      if (\n        finishReason === FinishReason.STOP ||\n        finishReason === FinishReason.MAX_TOKENS ||\n        finishReason === FinishReason.SAFETY\n      ) {\n        yield {\n          type: 'done',\n          id,\n          model,\n          timestamp: Date.now(),\n          finishReason:\n            finishReason === FinishReason.STOP\n              ? 'stop'\n              : finishReason === FinishReason.MAX_TOKENS\n                ? 'length'\n                : 'content_filter',\n          usage: {\n            promptTokens: inputTokens,\n            completionTokens: outputTokens,\n            totalTokens: inputTokens + outputTokens,\n          },\n        }\n      }\n    }\n  }\n\n  private getFormatInstructions(\n    style?: 'paragraph' | 'bullet-points' | 'concise',\n  ): string {\n    switch (style) {\n      case 'bullet-points':\n        return 'Provide the summary as bullet points.'\n      case 'concise':\n        return 'Provide a very brief one or two sentence summary.'\n      case 'paragraph':\n      default:\n        return 'Provide the summary in paragraph form.'\n    }\n  }\n}\n\n/**\n * Creates a Gemini summarize adapter with explicit API key and model\n */\nexport function createGeminiSummarize<TModel extends GeminiSummarizeModel>(\n  apiKey: string,\n  model: TModel,\n  options?: GeminiSummarizeAdapterOptions,\n): GeminiSummarizeAdapter<TModel> {\n  return new GeminiSummarizeAdapter(apiKey, model, options)\n}\n\n/**\n * Creates a Gemini summarize adapter with API key from environment and required model\n */\nexport function geminiSummarize<TModel extends GeminiSummarizeModel>(\n  model: TModel,\n  options?: GeminiSummarizeAdapterOptions,\n): GeminiSummarizeAdapter<TModel> {\n  const apiKey = getGeminiApiKeyFromEnv()\n  return new GeminiSummarizeAdapter(apiKey, model, options)\n}\n"],"names":[],"mappings":";;AAkBO,MAAM,wBAAwB;AAAA,EACnC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF;AA+BO,MAAM,uBAEyD;AAAA,EAYpE,YACE,gBACA,OACA,WAA0C,CAAA,GAC1C;AAfF,SAAS,OAAO;AAChB,SAAS,OAAO;AAed,SAAK,SACH,OAAO,mBAAmB,WACtB,mBAAmB,EAAE,QAAQ,eAAA,CAAgB,IAC7C;AACN,SAAK,QAAQ;AAAA,EACf;AAAA,EAEA,MAAM,UAAU,SAA6D;AAC3E,UAAM,QAAQ,QAAQ;AAGtB,UAAM,qBAAqB,KAAK,sBAAsB,QAAQ,KAAK;AACnE,UAAM,qBAAqB,QAAQ,YAC/B,2BAA2B,QAAQ,SAAS,aAC5C;AAEJ,UAAM,eAAe,qDAAqD,kBAAkB,GAAG,kBAAkB;AAEjH,UAAM,WAAW,MAAM,KAAK,OAAO,OAAO,gBAAgB;AAAA,MACxD;AAAA,MACA,UAAU;AAAA,QACR;AAAA,UACE,MAAM;AAAA,UACN,OAAO;AAAA,YACL,EAAE,MAAM;AAAA;AAAA,EAAsC,QAAQ,IAAI,GAAA;AAAA,UAAG;AAAA,QAC/D;AAAA,MACF;AAAA,MAEF,QAAQ;AAAA,QACN,mBAAmB;AAAA,MAAA;AAAA,IACrB,CACD;AAED,UAAM,UAAU,SAAS,QAAQ;AACjC,UAAM,cAAc,SAAS,eAAe,oBAAoB;AAChE,UAAM,eAAe,SAAS,eAAe,wBAAwB;AAErE,WAAO;AAAA,MACL,IAAI,WAAW,KAAK;AAAA,MACpB;AAAA,MACA;AAAA,MACA,OAAO;AAAA,QACL,cAAc;AAAA,QACd,kBAAkB;AAAA,QAClB,aAAa,cAAc;AAAA,MAAA;AAAA,IAC7B;AAAA,EAEJ;AAAA,EAEA,OAAO,gBACL,SAC4B;AAC5B,UAAM,QAAQ,QAAQ;AACtB,UAAM,KAAK,WAAW,KAAK;AAC3B,QAAI,qBAAqB;AACzB,QAAI,cAAc;AAClB,QAAI,eAAe;AAGnB,UAAM,qBAAqB,KAAK,sBAAsB,QAAQ,KAAK;AACnE,UAAM,qBAAqB,QAAQ,YAC/B,2BAA2B,QAAQ,SAAS,YAC5C;AAEJ,UAAM,eAAe,qDAAqD,kBAAkB,GAAG,kBAAkB;AAEjH,UAAM,SAAS,MAAM,KAAK,OAAO,OAAO,sBAAsB;AAAA,MAC5D;AAAA,MACA,UAAU;AAAA,QACR;AAAA,UACE,MAAM;AAAA,UACN,OAAO;AAAA,YACL,EAAE,MAAM;AAAA;AAAA,EAAsC,QAAQ,IAAI,GAAA;AAAA,UAAG;AAAA,QAC/D;AAAA,MACF;AAAA,MAEF,QAAQ;AAAA,QACN,mBAAmB;AAAA,MAAA;AAAA,IACrB,CACD;AAED,qBAAiB,SAAS,QAAQ;AAEhC,UAAI,MAAM,eAAe;AACvB,sBAAc,MAAM,cAAc,oBAAoB;AACtD,uBAAe,MAAM,cAAc,wBAAwB;AAAA,MAC7D;AAEA,UAAI,MAAM,aAAa,CAAC,GAAG,SAAS,OAAO;AACzC,mBAAW,QAAQ,MAAM,WAAW,CAAC,EAAE,QAAQ,OAAO;AACpD,cAAI,KAAK,MAAM;AACb,kCAAsB,KAAK;AAC3B,kBAAM;AAAA,cACJ,MAAM;AAAA,cACN;AAAA,cACA;AAAA,cACA,WAAW,KAAK,IAAA;AAAA,cAChB,OAAO,KAAK;AAAA,cACZ,SAAS;AAAA,cACT,MAAM;AAAA,YAAA;AAAA,UAEV;AAAA,QACF;AAAA,MACF;AAGA,YAAM,eAAe,MAAM,aAAa,CAAC,GAAG;AAC5C,UACE,iBAAiB,aAAa,QAC9B,iBAAiB,aAAa,cAC9B,iBAAiB,aAAa,QAC9B;AACA,cAAM;AAAA,UACJ,MAAM;AAAA,UACN;AAAA,UACA;AAAA,UACA,WAAW,KAAK,IAAA;AAAA,UAChB,cACE,iBAAiB,aAAa,OAC1B,SACA,iBAAiB,aAAa,aAC5B,WACA;AAAA,UACR,OAAO;AAAA,YACL,cAAc;AAAA,YACd,kBAAkB;AAAA,YAClB,aAAa,cAAc;AAAA,UAAA;AAAA,QAC7B;AAAA,MAEJ;AAAA,IACF;AAAA,EACF;AAAA,EAEQ,sBACN,OACQ;AACR,YAAQ,OAAA;AAAA,MACN,KAAK;AACH,eAAO;AAAA,MACT,KAAK;AACH,eAAO;AAAA,MACT,KAAK;AAAA,MACL;AACE,eAAO;AAAA,IAAA;AAAA,EAEb;AACF;AAKO,SAAS,sBACd,QACA,OACA,SACgC;AAChC,SAAO,IAAI,uBAAuB,QAAQ,OAAO,OAAO;AAC1D;AAKO,SAAS,gBACd,OACA,SACgC;AAChC,QAAM,SAAS,uBAAA;AACf,SAAO,IAAI,uBAAuB,QAAQ,OAAO,OAAO;AAC1D;"}
+{"version":3,"file":"summarize.js","sources":["../../../src/adapters/summarize.ts"],"sourcesContent":["import { FinishReason } from '@google/genai'\nimport {\n  createGeminiClient,\n  generateId,\n  getGeminiApiKeyFromEnv,\n} from '../utils'\nimport type { GoogleGenAI } from '@google/genai'\nimport type { GeminiClientConfig } from '../utils'\nimport type { SummarizeAdapter } from '@tanstack/ai/adapters'\nimport type {\n  StreamChunk,\n  SummarizationOptions,\n  SummarizationResult,\n} from '@tanstack/ai'\n\n/**\n * Configuration for Gemini summarize adapter\n */\nexport interface GeminiSummarizeConfig extends GeminiClientConfig {}\n/**\n * Available Gemini models for summarization\n */\nexport const GeminiSummarizeModels = [\n  'gemini-2.0-flash',\n  'gemini-1.5-flash',\n  'gemini-1.5-pro',\n  'gemini-2.0-flash-lite',\n] as const\n\nexport type GeminiSummarizeModel = (typeof GeminiSummarizeModels)[number]\n\n/**\n * Provider-specific options for Gemini summarization\n */\nexport interface GeminiSummarizeProviderOptions {\n  /** Generation configuration */\n  generationConfig?: {\n    temperature?: number\n    topP?: number\n    topK?: number\n    maxOutputTokens?: number\n    stopSequences?: Array<string>\n  }\n  /** Safety settings */\n  safetySettings?: Array<{\n    category: string\n    threshold: string\n  }>\n}\n\nexport interface GeminiSummarizeAdapterOptions {\n  // Additional adapter options can be added here\n}\n\n/**\n * Gemini Summarize Adapter\n * A tree-shakeable summarization adapter for Google Gemini\n */\nexport class GeminiSummarizeAdapter<\n  TModel extends GeminiSummarizeModel,\n> implements SummarizeAdapter<TModel, GeminiSummarizeProviderOptions> {\n  readonly kind = 'summarize' as const\n  readonly name = 'gemini' as const\n  readonly model: TModel\n\n  // Type-only property - never assigned at runtime\n  declare '~types': {\n    providerOptions: GeminiSummarizeProviderOptions\n  }\n\n  private client: GoogleGenAI\n\n  constructor(config: GeminiSummarizeConfig, model: TModel) {\n    this.client = createGeminiClient(config)\n    this.model = model\n  }\n\n  async summarize(options: SummarizationOptions): Promise<SummarizationResult> {\n    const model = options.model\n\n    // Build the system prompt based on format\n    const formatInstructions = this.getFormatInstructions(options.style)\n    const lengthInstructions = options.maxLength\n      ? ` Keep the summary under ${options.maxLength} tokens.`\n      : ''\n\n    const systemPrompt = `You are a helpful assistant that summarizes text. ${formatInstructions}${lengthInstructions}`\n\n    const response = await this.client.models.generateContent({\n      model,\n      contents: [\n        {\n          role: 'user',\n          parts: [\n            { text: `Please summarize the following:\\n\\n${options.text}` },\n          ],\n        },\n      ],\n      config: {\n        systemInstruction: systemPrompt,\n      },\n    })\n\n    const summary = response.text ?? ''\n    const inputTokens = response.usageMetadata?.promptTokenCount ?? 0\n    const outputTokens = response.usageMetadata?.candidatesTokenCount ?? 0\n\n    return {\n      id: generateId('sum'),\n      model,\n      summary,\n      usage: {\n        promptTokens: inputTokens,\n        completionTokens: outputTokens,\n        totalTokens: inputTokens + outputTokens,\n      },\n    }\n  }\n\n  async *summarizeStream(\n    options: SummarizationOptions,\n  ): AsyncIterable<StreamChunk> {\n    const model = options.model\n    const id = generateId('sum')\n    let accumulatedContent = ''\n    let inputTokens = 0\n    let outputTokens = 0\n\n    // Build the system prompt based on format\n    const formatInstructions = this.getFormatInstructions(options.style)\n    const lengthInstructions = options.maxLength\n      ? ` Keep the summary under ${options.maxLength} words.`\n      : ''\n\n    const systemPrompt = `You are a helpful assistant that summarizes text. ${formatInstructions}${lengthInstructions}`\n\n    const result = await this.client.models.generateContentStream({\n      model,\n      contents: [\n        {\n          role: 'user',\n          parts: [\n            { text: `Please summarize the following:\\n\\n${options.text}` },\n          ],\n        },\n      ],\n      config: {\n        systemInstruction: systemPrompt,\n      },\n    })\n\n    for await (const chunk of result) {\n      // Track usage metadata\n      if (chunk.usageMetadata) {\n        inputTokens = chunk.usageMetadata.promptTokenCount ?? inputTokens\n        outputTokens = chunk.usageMetadata.candidatesTokenCount ?? outputTokens\n      }\n\n      if (chunk.candidates?.[0]?.content?.parts) {\n        for (const part of chunk.candidates[0].content.parts) {\n          if (part.text) {\n            accumulatedContent += part.text\n            yield {\n              type: 'TEXT_MESSAGE_CONTENT',\n              messageId: id,\n              model,\n              timestamp: Date.now(),\n              delta: part.text,\n              content: accumulatedContent,\n            }\n          }\n        }\n      }\n\n      // Check for finish reason\n      const finishReason = chunk.candidates?.[0]?.finishReason\n      if (\n        finishReason === FinishReason.STOP ||\n        finishReason === FinishReason.MAX_TOKENS ||\n        finishReason === FinishReason.SAFETY\n      ) {\n        yield {\n          type: 'RUN_FINISHED',\n          runId: id,\n          model,\n          timestamp: Date.now(),\n          finishReason:\n            finishReason === FinishReason.STOP\n              ? 'stop'\n              : finishReason === FinishReason.MAX_TOKENS\n                ? 'length'\n                : 'content_filter',\n          usage: {\n            promptTokens: inputTokens,\n            completionTokens: outputTokens,\n            totalTokens: inputTokens + outputTokens,\n          },\n        }\n      }\n    }\n  }\n\n  private getFormatInstructions(\n    style?: 'paragraph' | 'bullet-points' | 'concise',\n  ): string {\n    switch (style) {\n      case 'bullet-points':\n        return 'Provide the summary as bullet points.'\n      case 'concise':\n        return 'Provide a very brief one or two sentence summary.'\n      case 'paragraph':\n      default:\n        return 'Provide the summary in paragraph form.'\n    }\n  }\n}\n\n/**\n * Creates a Gemini summarize adapter with explicit API key and model\n */\nexport function createGeminiSummarize<TModel extends GeminiSummarizeModel>(\n  apiKey: string,\n  model: TModel,\n  config?: Omit<GeminiSummarizeConfig, 'apiKey'>,\n): GeminiSummarizeAdapter<TModel> {\n  return new GeminiSummarizeAdapter({ ...config, apiKey }, model)\n}\n\n/**\n * Creates a Gemini summarize adapter with API key from environment and required model\n */\nexport function geminiSummarize<TModel extends GeminiSummarizeModel>(\n  model: TModel,\n  config?: Omit<GeminiSummarizeConfig, 'apiKey'>,\n): GeminiSummarizeAdapter<TModel> {\n  const apiKey = getGeminiApiKeyFromEnv()\n  return new GeminiSummarizeAdapter({ ...config, apiKey }, model)\n}\n"],"names":[],"mappings":";;AAsBO,MAAM,wBAAwB;AAAA,EACnC;AAAA,EACA;AAAA,EACA;AAAA,EACA;AACF;AA+BO,MAAM,uBAEyD;AAAA,EAYpE,YAAY,QAA+B,OAAe;AAX1D,SAAS,OAAO;AAChB,SAAS,OAAO;AAWd,SAAK,SAAS,mBAAmB,MAAM;AACvC,SAAK,QAAQ;AAAA,EACf;AAAA,EAEA,MAAM,UAAU,SAA6D;AAC3E,UAAM,QAAQ,QAAQ;AAGtB,UAAM,qBAAqB,KAAK,sBAAsB,QAAQ,KAAK;AACnE,UAAM,qBAAqB,QAAQ,YAC/B,2BAA2B,QAAQ,SAAS,aAC5C;AAEJ,UAAM,eAAe,qDAAqD,kBAAkB,GAAG,kBAAkB;AAEjH,UAAM,WAAW,MAAM,KAAK,OAAO,OAAO,gBAAgB;AAAA,MACxD;AAAA,MACA,UAAU;AAAA,QACR;AAAA,UACE,MAAM;AAAA,UACN,OAAO;AAAA,YACL,EAAE,MAAM;AAAA;AAAA,EAAsC,QAAQ,IAAI,GAAA;AAAA,UAAG;AAAA,QAC/D;AAAA,MACF;AAAA,MAEF,QAAQ;AAAA,QACN,mBAAmB;AAAA,MAAA;AAAA,IACrB,CACD;AAED,UAAM,UAAU,SAAS,QAAQ;AACjC,UAAM,cAAc,SAAS,eAAe,oBAAoB;AAChE,UAAM,eAAe,SAAS,eAAe,wBAAwB;AAErE,WAAO;AAAA,MACL,IAAI,WAAW,KAAK;AAAA,MACpB;AAAA,MACA;AAAA,MACA,OAAO;AAAA,QACL,cAAc;AAAA,QACd,kBAAkB;AAAA,QAClB,aAAa,cAAc;AAAA,MAAA;AAAA,IAC7B;AAAA,EAEJ;AAAA,EAEA,OAAO,gBACL,SAC4B;AAC5B,UAAM,QAAQ,QAAQ;AACtB,UAAM,KAAK,WAAW,KAAK;AAC3B,QAAI,qBAAqB;AACzB,QAAI,cAAc;AAClB,QAAI,eAAe;AAGnB,UAAM,qBAAqB,KAAK,sBAAsB,QAAQ,KAAK;AACnE,UAAM,qBAAqB,QAAQ,YAC/B,2BAA2B,QAAQ,SAAS,YAC5C;AAEJ,UAAM,eAAe,qDAAqD,kBAAkB,GAAG,kBAAkB;AAEjH,UAAM,SAAS,MAAM,KAAK,OAAO,OAAO,sBAAsB;AAAA,MAC5D;AAAA,MACA,UAAU;AAAA,QACR;AAAA,UACE,MAAM;AAAA,UACN,OAAO;AAAA,YACL,EAAE,MAAM;AAAA;AAAA,EAAsC,QAAQ,IAAI,GAAA;AAAA,UAAG;AAAA,QAC/D;AAAA,MACF;AAAA,MAEF,QAAQ;AAAA,QACN,mBAAmB;AAAA,MAAA;AAAA,IACrB,CACD;AAED,qBAAiB,SAAS,QAAQ;AAEhC,UAAI,MAAM,eAAe;AACvB,sBAAc,MAAM,cAAc,oBAAoB;AACtD,uBAAe,MAAM,cAAc,wBAAwB;AAAA,MAC7D;AAEA,UAAI,MAAM,aAAa,CAAC,GAAG,SAAS,OAAO;AACzC,mBAAW,QAAQ,MAAM,WAAW,CAAC,EAAE,QAAQ,OAAO;AACpD,cAAI,KAAK,MAAM;AACb,kCAAsB,KAAK;AAC3B,kBAAM;AAAA,cACJ,MAAM;AAAA,cACN,WAAW;AAAA,cACX;AAAA,cACA,WAAW,KAAK,IAAA;AAAA,cAChB,OAAO,KAAK;AAAA,cACZ,SAAS;AAAA,YAAA;AAAA,UAEb;AAAA,QACF;AAAA,MACF;AAGA,YAAM,eAAe,MAAM,aAAa,CAAC,GAAG;AAC5C,UACE,iBAAiB,aAAa,QAC9B,iBAAiB,aAAa,cAC9B,iBAAiB,aAAa,QAC9B;AACA,cAAM;AAAA,UACJ,MAAM;AAAA,UACN,OAAO;AAAA,UACP;AAAA,UACA,WAAW,KAAK,IAAA;AAAA,UAChB,cACE,iBAAiB,aAAa,OAC1B,SACA,iBAAiB,aAAa,aAC5B,WACA;AAAA,UACR,OAAO;AAAA,YACL,cAAc;AAAA,YACd,kBAAkB;AAAA,YAClB,aAAa,cAAc;AAAA,UAAA;AAAA,QAC7B;AAAA,MAEJ;AAAA,IACF;AAAA,EACF;AAAA,EAEQ,sBACN,OACQ;AACR,YAAQ,OAAA;AAAA,MACN,KAAK;AACH,eAAO;AAAA,MACT,KAAK;AACH,eAAO;AAAA,MACT,KAAK;AAAA,MACL;AACE,eAAO;AAAA,IAAA;AAAA,EAEb;AACF;AAKO,SAAS,sBACd,QACA,OACA,QACgC;AAChC,SAAO,IAAI,uBAAuB,EAAE,GAAG,QAAQ,OAAA,GAAU,KAAK;AAChE;AAKO,SAAS,gBACd,OACA,QACgC;AAChC,QAAM,SAAS,uBAAA;AACf,SAAO,IAAI,uBAAuB,EAAE,GAAG,QAAQ,OAAA,GAAU,KAAK;AAChE;"}

package/dist/esm/adapters/text.js

@@ -17,8 +17,7 @@ class GeminiTextAdapter extends BaseTextAdapter {
     } catch (error) {
       const timestamp = Date.now();
       yield {
-        type: "error",
-        id: generateId(this.name),
+        type: "RUN_ERROR",
         model: options.model,
         timestamp,
         error: {
@@ -82,30 +81,65 @@ class GeminiTextAdapter extends BaseTextAdapter {
     let accumulatedContent = "";
     const toolCallMap = /* @__PURE__ */ new Map();
     let nextToolIndex = 0;
+    const runId = generateId(this.name);
+    const messageId = generateId(this.name);
+    let stepId = null;
+    let hasEmittedRunStarted = false;
+    let hasEmittedTextMessageStart = false;
+    let hasEmittedStepStarted = false;
     for await (const chunk of result) {
+      if (!hasEmittedRunStarted) {
+        hasEmittedRunStarted = true;
+        yield {
+          type: "RUN_STARTED",
+          runId,
+          model,
+          timestamp
+        };
+      }
       if (chunk.candidates?.[0]?.content?.parts) {
         const parts = chunk.candidates[0].content.parts;
         for (const part of parts) {
           if (part.text) {
             if (part.thought) {
+              if (!hasEmittedStepStarted) {
+                hasEmittedStepStarted = true;
+                stepId = generateId(this.name);
+                yield {
+                  type: "STEP_STARTED",
+                  stepId,
+                  model,
+                  timestamp,
+                  stepType: "thinking"
+                };
+              }
               yield {
-                type: "thinking",
-                content: part.text,
-                delta: part.text,
-                id: generateId(this.name),
+                type: "STEP_FINISHED",
+                stepId: stepId || generateId(this.name),
                 model,
-                timestamp
+                timestamp,
+                delta: part.text,
+                content: part.text
               };
             } else {
+              if (!hasEmittedTextMessageStart) {
+                hasEmittedTextMessageStart = true;
+                yield {
+                  type: "TEXT_MESSAGE_START",
+                  messageId,
+                  model,
+                  timestamp,
+                  role: "assistant"
+                };
+              }
               accumulatedContent += part.text;
               yield {
-                type: "content",
-                id: generateId(this.name),
+                type: "TEXT_MESSAGE_CONTENT",
+                messageId,
                 model,
                 timestamp,
                 delta: part.text,
-                content: accumulatedContent,
-                role: "assistant"
+                content: accumulatedContent
               };
             }
           }
@@ -118,7 +152,8 @@ class GeminiTextAdapter extends BaseTextAdapter {
               toolCallData = {
                 name: functionCall.name || "",
                 args: typeof functionArgs === "string" ? functionArgs : JSON.stringify(functionArgs),
-                index: nextToolIndex++
+                index: nextToolIndex++,
+                started: false
               };
               toolCallMap.set(toolCallId, toolCallData);
             } else {
@@ -131,33 +166,46 @@ class GeminiTextAdapter extends BaseTextAdapter {
                 toolCallData.args = typeof functionArgs === "string" ? functionArgs : JSON.stringify(functionArgs);
               }
             }
+            if (!toolCallData.started) {
+              toolCallData.started = true;
+              yield {
+                type: "TOOL_CALL_START",
+                toolCallId,
+                toolName: toolCallData.name,
+                model,
+                timestamp,
+                index: toolCallData.index
+              };
+            }
             yield {
-              type: "tool_call",
-              id: generateId(this.name),
+              type: "TOOL_CALL_ARGS",
+              toolCallId,
               model,
               timestamp,
-              toolCall: {
-                id: toolCallId,
-                type: "function",
-                function: {
-                  name: toolCallData.name,
-                  arguments: toolCallData.args
-                }
-              },
-              index: toolCallData.index
+              delta: toolCallData.args,
+              args: toolCallData.args
             };
           }
         }
       } else if (chunk.data) {
+        if (!hasEmittedTextMessageStart) {
+          hasEmittedTextMessageStart = true;
+          yield {
+            type: "TEXT_MESSAGE_START",
+            messageId,
+            model,
+            timestamp,
+            role: "assistant"
+          };
+        }
         accumulatedContent += chunk.data;
         yield {
-          type: "content",
-          id: generateId(this.name),
+          type: "TEXT_MESSAGE_CONTENT",
+          messageId,
           model,
           timestamp,
           delta: chunk.data,
-          content: accumulatedContent,
-          role: "assistant"
+          content: accumulatedContent
         };
       }
       if (chunk.candidates?.[0]?.finishReason) {
@@ -169,44 +217,78 @@ class GeminiTextAdapter extends BaseTextAdapter {
               if (functionCall) {
                 const toolCallId = functionCall.id || `${functionCall.name}_${Date.now()}_${nextToolIndex}`;
                 const functionArgs = functionCall.args || {};
+                const argsString = typeof functionArgs === "string" ? functionArgs : JSON.stringify(functionArgs);
                 toolCallMap.set(toolCallId, {
                   name: functionCall.name || "",
-                  args: typeof functionArgs === "string" ? functionArgs : JSON.stringify(functionArgs),
-                  index: nextToolIndex++
+                  args: argsString,
+                  index: nextToolIndex++,
+                  started: true
                 });
                 yield {
-                  type: "tool_call",
-                  id: generateId(this.name),
+                  type: "TOOL_CALL_START",
+                  toolCallId,
+                  toolName: functionCall.name || "",
                   model,
                   timestamp,
-                  toolCall: {
-                    id: toolCallId,
-                    type: "function",
-                    function: {
-                      name: functionCall.name || "",
-                      arguments: typeof functionArgs === "string" ? functionArgs : JSON.stringify(functionArgs)
-                    }
-                  },
                   index: nextToolIndex - 1
                 };
+                let parsedInput = {};
+                try {
+                  parsedInput = typeof functionArgs === "string" ? JSON.parse(functionArgs) : functionArgs;
+                } catch {
+                  parsedInput = {};
+                }
+                yield {
+                  type: "TOOL_CALL_END",
+                  toolCallId,
+                  toolName: functionCall.name || "",
+                  model,
+                  timestamp,
+                  input: parsedInput
+                };
               }
             }
           }
         }
+        for (const [toolCallId, toolCallData] of toolCallMap.entries()) {
+          let parsedInput = {};
+          try {
+            parsedInput = JSON.parse(toolCallData.args);
+          } catch {
+            parsedInput = {};
+          }
+          yield {
+            type: "TOOL_CALL_END",
+            toolCallId,
+            toolName: toolCallData.name,
+            model,
+            timestamp,
+            input: parsedInput
+          };
+        }
         if (finishReason === FinishReason.MAX_TOKENS) {
           yield {
-            type: "error",
-            id: generateId(this.name),
+            type: "RUN_ERROR",
+            runId,
             model,
             timestamp,
             error: {
-              message: "The response was cut off because the maximum token limit was reached."
+              message: "The response was cut off because the maximum token limit was reached.",
+              code: "max_tokens"
             }
           };
         }
+        if (hasEmittedTextMessageStart) {
+          yield {
+            type: "TEXT_MESSAGE_END",
+            messageId,
+            model,
+            timestamp
+          };
+        }
         yield {
-          type: "done",
-          id: generateId(this.name),
+          type: "RUN_FINISHED",
+          runId,
           model,
           timestamp,
           finishReason: toolCallMap.size > 0 ? "tool_calls" : "stop",
@@ -220,18 +302,6 @@ class GeminiTextAdapter extends BaseTextAdapter {
     }
   }
   convertContentPartToGemini(part) {
-    const getDefaultFileType = (part2) => {
-      switch (part2) {
-        case "image":
-          return "image/jpeg";
-        case "audio":
-          return "audio/mp3";
-        case "video":
-          return "video/mp4";
-        case "document":
-          return "application/pdf";
-      }
-    };
     switch (part.type) {
       case "text":
         return { text: part.content };
@@ -239,19 +309,24 @@ class GeminiTextAdapter extends BaseTextAdapter {
       case "audio":
       case "video":
       case "document": {
-        const metadata = part.metadata;
         if (part.source.type === "data") {
           return {
             inlineData: {
               data: part.source.value,
-              mimeType: metadata?.mimeType ?? getDefaultFileType(part.type)
+              mimeType: part.source.mimeType
             }
           };
         } else {
+          const defaultMimeType = {
+            image: "image/jpeg",
+            audio: "audio/mp3",
+            video: "video/mp4",
+            document: "application/pdf"
+          }[part.type];
           return {
             fileData: {
               fileUri: part.source.value,
-              mimeType: metadata?.mimeType ?? getDefaultFileType(part.type)
+              mimeType: part.source.mimeType ?? defaultMimeType
             }
           };
         }

package/dist/esm/adapters/text.js.map

@@ -1 +1 @@
-{"version":3,"file":"text.js","sources":["../../../src/adapters/text.ts"],"sourcesContent":["import { FinishReason } from '@google/genai'\nimport { BaseTextAdapter } from '@tanstack/ai/adapters'\nimport { convertToolsToProviderFormat } from '../tools/tool-converter'\nimport {\n  createGeminiClient,\n  generateId,\n  getGeminiApiKeyFromEnv,\n} from '../utils'\nimport type {\n  GEMINI_MODELS,\n  GeminiChatModelProviderOptionsByName,\n  GeminiModelInputModalitiesByName,\n} from '../model-meta'\nimport type {\n  StructuredOutputOptions,\n  StructuredOutputResult,\n} from '@tanstack/ai/adapters'\nimport type {\n  GenerateContentParameters,\n  GenerateContentResponse,\n  GoogleGenAI,\n  Part,\n  ThinkingLevel,\n} from '@google/genai'\nimport type {\n  ContentPart,\n  Modality,\n  ModelMessage,\n  StreamChunk,\n  TextOptions,\n} from '@tanstack/ai'\nimport type { ExternalTextProviderOptions } from '../text/text-provider-options'\nimport type {\n  GeminiAudioMetadata,\n  GeminiDocumentMetadata,\n  GeminiImageMetadata,\n  GeminiMessageMetadataByModality,\n  GeminiVideoMetadata,\n} from '../message-types'\nimport type { GeminiClientConfig } from '../utils'\n\n/**\n * Configuration for Gemini text adapter\n */\nexport interface GeminiTextConfig extends GeminiClientConfig {}\n\n/**\n * Gemini-specific provider options for text/chat\n */\nexport type GeminiTextProviderOptions = ExternalTextProviderOptions\n\n// ===========================\n// Type Resolution Helpers\n// ===========================\n\n/**\n * Resolve provider options for a specific model.\n * If the model has explicit options in the map, use those; otherwise use base options.\n */\ntype ResolveProviderOptions<TModel extends string> =\n  TModel extends keyof GeminiChatModelProviderOptionsByName\n    ? GeminiChatModelProviderOptionsByName[TModel]\n    : GeminiTextProviderOptions\n\n/**\n * Resolve input modalities for a specific model.\n * If the model has explicit modalities in the map, use those; otherwise use all modalities.\n */\ntype ResolveInputModalities<TModel extends string> =\n  TModel extends keyof GeminiModelInputModalitiesByName\n    ? GeminiModelInputModalitiesByName[TModel]\n    : readonly ['text', 'image', 'audio', 'video', 'document']\n\n// ===========================\n// Adapter Implementation\n// ===========================\n\n/**\n * Gemini Text (Chat) Adapter\n *\n * Tree-shakeable adapter for Gemini chat/text completion functionality.\n * Import only what you need for smaller bundle sizes.\n */\nexport class GeminiTextAdapter<\n  TModel extends (typeof GEMINI_MODELS)[number],\n  TProviderOptions extends object = ResolveProviderOptions<TModel>,\n  TInputModalities extends ReadonlyArray<Modality> =\n    ResolveInputModalities<TModel>,\n> extends BaseTextAdapter<\n  TModel,\n  TProviderOptions,\n  TInputModalities,\n  GeminiMessageMetadataByModality\n> {\n  readonly kind = 'text' as const\n  readonly name = 'gemini' as const\n\n  private client: GoogleGenAI\n\n  constructor(config: GeminiTextConfig, model: TModel) {\n    super({}, model)\n    this.client = createGeminiClient(config)\n  }\n\n  async *chatStream(\n    options: TextOptions<GeminiTextProviderOptions>,\n  ): AsyncIterable<StreamChunk> {\n    const mappedOptions = this.mapCommonOptionsToGemini(options)\n\n    try {\n      const result =\n        await this.client.models.generateContentStream(mappedOptions)\n\n      yield* this.processStreamChunks(result, options.model)\n    } catch (error) {\n      const timestamp = Date.now()\n      yield {\n        type: 'error',\n        id: generateId(this.name),\n        model: options.model,\n        timestamp,\n        error: {\n          message:\n            error instanceof Error\n              ? error.message\n              : 'An unknown error occurred during the chat stream.',\n        },\n      }\n    }\n  }\n\n  /**\n   * Generate structured output using Gemini's native JSON response format.\n   * Uses responseMimeType: 'application/json' and responseSchema for structured output.\n   * The outputSchema is already JSON Schema (converted in the ai layer).\n   */\n  async structuredOutput(\n    options: StructuredOutputOptions<GeminiTextProviderOptions>,\n  ): Promise<StructuredOutputResult<unknown>> {\n    const { chatOptions, outputSchema } = options\n\n    const mappedOptions = this.mapCommonOptionsToGemini(chatOptions)\n\n    try {\n      // Add structured output configuration\n      const result = await this.client.models.generateContent({\n        ...mappedOptions,\n        config: {\n          ...mappedOptions.config,\n          responseMimeType: 'application/json',\n          responseSchema: outputSchema,\n        },\n      })\n\n      // Extract text content from the response\n      const rawText = this.extractTextFromResponse(result)\n\n      // Parse the JSON response\n      let parsed: unknown\n      try {\n        parsed = JSON.parse(rawText)\n      } catch {\n        throw new Error(\n          `Failed to parse structured output as JSON. Content: ${rawText.slice(0, 200)}${rawText.length > 200 ? '...' : ''}`,\n        )\n      }\n\n      return {\n        data: parsed,\n        rawText,\n      }\n    } catch (error) {\n      throw new Error(\n        error instanceof Error\n          ? error.message\n          : 'An unknown error occurred during structured output generation.',\n      )\n    }\n  }\n\n  /**\n   * Extract text content from a non-streaming response\n   */\n  private extractTextFromResponse(response: GenerateContentResponse): string {\n    let textContent = ''\n\n    if (response.candidates?.[0]?.content?.parts) {\n      for (const part of response.candidates[0].content.parts) {\n        if (part.text) {\n          textContent += part.text\n        }\n      }\n    }\n\n    return textContent\n  }\n\n  private async *processStreamChunks(\n    result: AsyncGenerator<GenerateContentResponse, unknown, unknown>,\n    model: string,\n  ): AsyncIterable<StreamChunk> {\n    const timestamp = Date.now()\n    let accumulatedContent = ''\n    const toolCallMap = new Map<\n      string,\n      { name: string; args: string; index: number }\n    >()\n    let nextToolIndex = 0\n\n    for await (const chunk of result) {\n      if (chunk.candidates?.[0]?.content?.parts) {\n        const parts = chunk.candidates[0].content.parts\n\n        for (const part of parts) {\n          if (part.text) {\n            if (part.thought) {\n              yield {\n                type: 'thinking',\n                content: part.text,\n                delta: part.text,\n                id: generateId(this.name),\n                model,\n                timestamp,\n              }\n            } else {\n              accumulatedContent += part.text\n              yield {\n                type: 'content',\n                id: generateId(this.name),\n                model,\n                timestamp,\n                delta: part.text,\n                content: accumulatedContent,\n                role: 'assistant',\n              }\n            }\n          }\n\n          const functionCall = part.functionCall\n          if (functionCall) {\n            const toolCallId =\n              functionCall.id ||\n              `${functionCall.name}_${Date.now()}_${nextToolIndex}`\n            const functionArgs = functionCall.args || {}\n\n            let toolCallData = toolCallMap.get(toolCallId)\n            if (!toolCallData) {\n              toolCallData = {\n                name: functionCall.name || '',\n                args:\n                  typeof functionArgs === 'string'\n                    ? functionArgs\n                    : JSON.stringify(functionArgs),\n                index: nextToolIndex++,\n              }\n              toolCallMap.set(toolCallId, toolCallData)\n            } else {\n              try {\n                const existingArgs = JSON.parse(toolCallData.args)\n                const newArgs =\n                  typeof functionArgs === 'string'\n                    ? JSON.parse(functionArgs)\n                    : functionArgs\n                const mergedArgs = { ...existingArgs, ...newArgs }\n                toolCallData.args = JSON.stringify(mergedArgs)\n              } catch {\n                toolCallData.args =\n                  typeof functionArgs === 'string'\n                    ? functionArgs\n                    : JSON.stringify(functionArgs)\n              }\n            }\n\n            yield {\n              type: 'tool_call',\n              id: generateId(this.name),\n              model,\n              timestamp,\n              toolCall: {\n                id: toolCallId,\n                type: 'function',\n                function: {\n                  name: toolCallData.name,\n                  arguments: toolCallData.args,\n                },\n              },\n              index: toolCallData.index,\n            }\n          }\n        }\n      } else if (chunk.data) {\n        accumulatedContent += chunk.data\n        yield {\n          type: 'content',\n          id: generateId(this.name),\n          model,\n          timestamp,\n          delta: chunk.data,\n          content: accumulatedContent,\n          role: 'assistant',\n        }\n      }\n\n      if (chunk.candidates?.[0]?.finishReason) {\n        const finishReason = chunk.candidates[0].finishReason\n\n        if (finishReason === FinishReason.UNEXPECTED_TOOL_CALL) {\n          if (chunk.candidates[0].content?.parts) {\n            for (const part of chunk.candidates[0].content.parts) {\n              const functionCall = part.functionCall\n              if (functionCall) {\n                const toolCallId =\n                  functionCall.id ||\n                  `${functionCall.name}_${Date.now()}_${nextToolIndex}`\n                const functionArgs = functionCall.args || {}\n\n                toolCallMap.set(toolCallId, {\n                  name: functionCall.name || '',\n                  args:\n                    typeof functionArgs === 'string'\n                      ? functionArgs\n                      : JSON.stringify(functionArgs),\n                  index: nextToolIndex++,\n                })\n\n                yield {\n                  type: 'tool_call',\n                  id: generateId(this.name),\n                  model,\n                  timestamp,\n                  toolCall: {\n                    id: toolCallId,\n                    type: 'function',\n                    function: {\n                      name: functionCall.name || '',\n                      arguments:\n                        typeof functionArgs === 'string'\n                          ? functionArgs\n                          : JSON.stringify(functionArgs),\n                    },\n                  },\n                  index: nextToolIndex - 1,\n                }\n              }\n            }\n          }\n        }\n        if (finishReason === FinishReason.MAX_TOKENS) {\n          yield {\n            type: 'error',\n            id: generateId(this.name),\n            model,\n            timestamp,\n            error: {\n              message:\n                'The response was cut off because the maximum token limit was reached.',\n            },\n          }\n        }\n\n        yield {\n          type: 'done',\n          id: generateId(this.name),\n          model,\n          timestamp,\n          finishReason: toolCallMap.size > 0 ? 'tool_calls' : 'stop',\n          usage: chunk.usageMetadata\n            ? {\n                promptTokens: chunk.usageMetadata.promptTokenCount ?? 0,\n                completionTokens: chunk.usageMetadata.candidatesTokenCount ?? 0,\n                totalTokens: chunk.usageMetadata.totalTokenCount ?? 0,\n              }\n            : undefined,\n        }\n      }\n    }\n  }\n\n  private convertContentPartToGemini(part: ContentPart): Part {\n    const getDefaultFileType = (\n      part: 'image' | 'audio' | 'video' | 'document',\n    ) => {\n      switch (part) {\n        case 'image':\n          return 'image/jpeg'\n        case 'audio':\n          return 'audio/mp3'\n        case 'video':\n          return 'video/mp4'\n        case 'document':\n          return 'application/pdf'\n      }\n    }\n    switch (part.type) {\n      case 'text':\n        return { text: part.content }\n      case 'image':\n      case 'audio':\n      case 'video':\n      case 'document': {\n        const metadata = part.metadata as\n          | GeminiDocumentMetadata\n          | GeminiImageMetadata\n          | GeminiVideoMetadata\n          | GeminiAudioMetadata\n          | undefined\n        if (part.source.type === 'data') {\n          return {\n            inlineData: {\n              data: part.source.value,\n              mimeType: metadata?.mimeType ?? getDefaultFileType(part.type),\n            },\n          }\n        } else {\n          return {\n            fileData: {\n              fileUri: part.source.value,\n              mimeType: metadata?.mimeType ?? getDefaultFileType(part.type),\n            },\n          }\n        }\n      }\n      default: {\n        const _exhaustiveCheck: never = part\n        throw new Error(\n          `Unsupported content part type: ${(_exhaustiveCheck as ContentPart).type}`,\n        )\n      }\n    }\n  }\n\n  private formatMessages(\n    messages: Array<ModelMessage>,\n  ): GenerateContentParameters['contents'] {\n    return messages.map((msg) => {\n      const role: 'user' | 'model' = msg.role === 'assistant' ? 'model' : 'user'\n      const parts: Array<Part> = []\n\n      if (Array.isArray(msg.content)) {\n        for (const contentPart of msg.content) {\n          parts.push(this.convertContentPartToGemini(contentPart))\n        }\n      } else if (msg.content) {\n        parts.push({ text: msg.content })\n      }\n\n      if (msg.role === 'assistant' && msg.toolCalls?.length) {\n        for (const toolCall of msg.toolCalls) {\n          let parsedArgs: Record<string, unknown> = {}\n          try {\n            parsedArgs = toolCall.function.arguments\n              ? (JSON.parse(toolCall.function.arguments) as Record<\n                  string,\n                  unknown\n                >)\n              : {}\n          } catch {\n            parsedArgs = toolCall.function.arguments as unknown as Record<\n              string,\n              unknown\n            >\n          }\n\n          parts.push({\n            functionCall: {\n              name: toolCall.function.name,\n              args: parsedArgs,\n            },\n          })\n        }\n      }\n\n      if (msg.role === 'tool' && msg.toolCallId) {\n        parts.push({\n          functionResponse: {\n            name: msg.toolCallId,\n            response: {\n              content: msg.content || '',\n            },\n          },\n        })\n      }\n\n      return {\n        role,\n        parts: parts.length > 0 ? parts : [{ text: '' }],\n      }\n    })\n  }\n\n  private mapCommonOptionsToGemini(\n    options: TextOptions<GeminiTextProviderOptions>,\n  ) {\n    const modelOpts = options.modelOptions\n    const thinkingConfig = modelOpts?.thinkingConfig\n    const requestOptions: GenerateContentParameters = {\n      model: options.model,\n      contents: this.formatMessages(options.messages),\n      config: {\n        ...modelOpts,\n        temperature: options.temperature,\n        topP: options.topP,\n        maxOutputTokens: options.maxTokens,\n        thinkingConfig: thinkingConfig\n          ? {\n              ...thinkingConfig,\n              thinkingLevel: thinkingConfig.thinkingLevel\n                ? // Enum is provided by the SDK, we use it for the type but cast it to string constants, here we just cast them back\n                  (thinkingConfig.thinkingLevel as ThinkingLevel)\n                : undefined,\n            }\n          : undefined,\n        systemInstruction: options.systemPrompts?.join('\\n'),\n        tools: convertToolsToProviderFormat(options.tools),\n      },\n    }\n\n    return requestOptions\n  }\n}\n\n/**\n * Creates a Gemini text adapter with explicit API key.\n * Type resolution happens here at the call site.\n */\nexport function createGeminiChat<TModel extends (typeof GEMINI_MODELS)[number]>(\n  model: TModel,\n  apiKey: string,\n  config?: Omit<GeminiTextConfig, 'apiKey'>,\n): GeminiTextAdapter<\n  TModel,\n  ResolveProviderOptions<TModel>,\n  ResolveInputModalities<TModel>\n> {\n  return new GeminiTextAdapter({ apiKey, ...config }, model)\n}\n\n/**\n * Creates a Gemini text adapter with automatic API key detection.\n * Type resolution happens here at the call site.\n */\nexport function geminiText<TModel extends (typeof GEMINI_MODELS)[number]>(\n  model: TModel,\n  config?: Omit<GeminiTextConfig, 'apiKey'>,\n): GeminiTextAdapter<\n  TModel,\n  ResolveProviderOptions<TModel>,\n  ResolveInputModalities<TModel>\n> {\n  const apiKey = getGeminiApiKeyFromEnv()\n  return createGeminiChat(model, apiKey, config)\n}\n"],"names":["part"],"mappings":";;;;AAmFO,MAAM,0BAKH,gBAKR;AAAA,EAMA,YAAY,QAA0B,OAAe;AACnD,UAAM,CAAA,GAAI,KAAK;AANjB,SAAS,OAAO;AAChB,SAAS,OAAO;AAMd,SAAK,SAAS,mBAAmB,MAAM;AAAA,EACzC;AAAA,EAEA,OAAO,WACL,SAC4B;AAC5B,UAAM,gBAAgB,KAAK,yBAAyB,OAAO;AAE3D,QAAI;AACF,YAAM,SACJ,MAAM,KAAK,OAAO,OAAO,sBAAsB,aAAa;AAE9D,aAAO,KAAK,oBAAoB,QAAQ,QAAQ,KAAK;AAAA,IACvD,SAAS,OAAO;AACd,YAAM,YAAY,KAAK,IAAA;AACvB,YAAM;AAAA,QACJ,MAAM;AAAA,QACN,IAAI,WAAW,KAAK,IAAI;AAAA,QACxB,OAAO,QAAQ;AAAA,QACf;AAAA,QACA,OAAO;AAAA,UACL,SACE,iBAAiB,QACb,MAAM,UACN;AAAA,QAAA;AAAA,MACR;AAAA,IAEJ;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,iBACJ,SAC0C;AAC1C,UAAM,EAAE,aAAa,aAAA,IAAiB;AAEtC,UAAM,gBAAgB,KAAK,yBAAyB,WAAW;AAE/D,QAAI;AAEF,YAAM,SAAS,MAAM,KAAK,OAAO,OAAO,gBAAgB;AAAA,QACtD,GAAG;AAAA,QACH,QAAQ;AAAA,UACN,GAAG,cAAc;AAAA,UACjB,kBAAkB;AAAA,UAClB,gBAAgB;AAAA,QAAA;AAAA,MAClB,CACD;AAGD,YAAM,UAAU,KAAK,wBAAwB,MAAM;AAGnD,UAAI;AACJ,UAAI;AACF,iBAAS,KAAK,MAAM,OAAO;AAAA,MAC7B,QAAQ;AACN,cAAM,IAAI;AAAA,UACR,uDAAuD,QAAQ,MAAM,GAAG,GAAG,CAAC,GAAG,QAAQ,SAAS,MAAM,QAAQ,EAAE;AAAA,QAAA;AAAA,MAEpH;AAEA,aAAO;AAAA,QACL,MAAM;AAAA,QACN;AAAA,MAAA;AAAA,IAEJ,SAAS,OAAO;AACd,YAAM,IAAI;AAAA,QACR,iBAAiB,QACb,MAAM,UACN;AAAA,MAAA;AAAA,IAER;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKQ,wBAAwB,UAA2C;AACzE,QAAI,cAAc;AAElB,QAAI,SAAS,aAAa,CAAC,GAAG,SAAS,OAAO;AAC5C,iBAAW,QAAQ,SAAS,WAAW,CAAC,EAAE,QAAQ,OAAO;AACvD,YAAI,KAAK,MAAM;AACb,yBAAe,KAAK;AAAA,QACtB;AAAA,MACF;AAAA,IACF;AAEA,WAAO;AAAA,EACT;AAAA,EAEA,OAAe,oBACb,QACA,OAC4B;AAC5B,UAAM,YAAY,KAAK,IAAA;AACvB,QAAI,qBAAqB;AACzB,UAAM,kCAAkB,IAAA;AAIxB,QAAI,gBAAgB;AAEpB,qBAAiB,SAAS,QAAQ;AAChC,UAAI,MAAM,aAAa,CAAC,GAAG,SAAS,OAAO;AACzC,cAAM,QAAQ,MAAM,WAAW,CAAC,EAAE,QAAQ;AAE1C,mBAAW,QAAQ,OAAO;AACxB,cAAI,KAAK,MAAM;AACb,gBAAI,KAAK,SAAS;AAChB,oBAAM;AAAA,gBACJ,MAAM;AAAA,gBACN,SAAS,KAAK;AAAA,gBACd,OAAO,KAAK;AAAA,gBACZ,IAAI,WAAW,KAAK,IAAI;AAAA,gBACxB;AAAA,gBACA;AAAA,cAAA;AAAA,YAEJ,OAAO;AACL,oCAAsB,KAAK;AAC3B,oBAAM;AAAA,gBACJ,MAAM;AAAA,gBACN,IAAI,WAAW,KAAK,IAAI;AAAA,gBACxB;AAAA,gBACA;AAAA,gBACA,OAAO,KAAK;AAAA,gBACZ,SAAS;AAAA,gBACT,MAAM;AAAA,cAAA;AAAA,YAEV;AAAA,UACF;AAEA,gBAAM,eAAe,KAAK;AAC1B,cAAI,cAAc;AAChB,kBAAM,aACJ,aAAa,MACb,GAAG,aAAa,IAAI,IAAI,KAAK,IAAA,CAAK,IAAI,aAAa;AACrD,kBAAM,eAAe,aAAa,QAAQ,CAAA;AAE1C,gBAAI,eAAe,YAAY,IAAI,UAAU;AAC7C,gBAAI,CAAC,cAAc;AACjB,6BAAe;AAAA,gBACb,MAAM,aAAa,QAAQ;AAAA,gBAC3B,MACE,OAAO,iBAAiB,WACpB,eACA,KAAK,UAAU,YAAY;AAAA,gBACjC,OAAO;AAAA,cAAA;AAET,0BAAY,IAAI,YAAY,YAAY;AAAA,YAC1C,OAAO;AACL,kBAAI;AACF,sBAAM,eAAe,KAAK,MAAM,aAAa,IAAI;AACjD,sBAAM,UACJ,OAAO,iBAAiB,WACpB,KAAK,MAAM,YAAY,IACvB;AACN,sBAAM,aAAa,EAAE,GAAG,cAAc,GAAG,QAAA;AACzC,6BAAa,OAAO,KAAK,UAAU,UAAU;AAAA,cAC/C,QAAQ;AACN,6BAAa,OACX,OAAO,iBAAiB,WACpB,eACA,KAAK,UAAU,YAAY;AAAA,cACnC;AAAA,YACF;AAEA,kBAAM;AAAA,cACJ,MAAM;AAAA,cACN,IAAI,WAAW,KAAK,IAAI;AAAA,cACxB;AAAA,cACA;AAAA,cACA,UAAU;AAAA,gBACR,IAAI;AAAA,gBACJ,MAAM;AAAA,gBACN,UAAU;AAAA,kBACR,MAAM,aAAa;AAAA,kBACnB,WAAW,aAAa;AAAA,gBAAA;AAAA,cAC1B;AAAA,cAEF,OAAO,aAAa;AAAA,YAAA;AAAA,UAExB;AAAA,QACF;AAAA,MACF,WAAW,MAAM,MAAM;AACrB,8BAAsB,MAAM;AAC5B,cAAM;AAAA,UACJ,MAAM;AAAA,UACN,IAAI,WAAW,KAAK,IAAI;AAAA,UACxB;AAAA,UACA;AAAA,UACA,OAAO,MAAM;AAAA,UACb,SAAS;AAAA,UACT,MAAM;AAAA,QAAA;AAAA,MAEV;AAEA,UAAI,MAAM,aAAa,CAAC,GAAG,cAAc;AACvC,cAAM,eAAe,MAAM,WAAW,CAAC,EAAE;AAEzC,YAAI,iBAAiB,aAAa,sBAAsB;AACtD,cAAI,MAAM,WAAW,CAAC,EAAE,SAAS,OAAO;AACtC,uBAAW,QAAQ,MAAM,WAAW,CAAC,EAAE,QAAQ,OAAO;AACpD,oBAAM,eAAe,KAAK;AAC1B,kBAAI,cAAc;AAChB,sBAAM,aACJ,aAAa,MACb,GAAG,aAAa,IAAI,IAAI,KAAK,IAAA,CAAK,IAAI,aAAa;AACrD,sBAAM,eAAe,aAAa,QAAQ,CAAA;AAE1C,4BAAY,IAAI,YAAY;AAAA,kBAC1B,MAAM,aAAa,QAAQ;AAAA,kBAC3B,MACE,OAAO,iBAAiB,WACpB,eACA,KAAK,UAAU,YAAY;AAAA,kBACjC,OAAO;AAAA,gBAAA,CACR;AAED,sBAAM;AAAA,kBACJ,MAAM;AAAA,kBACN,IAAI,WAAW,KAAK,IAAI;AAAA,kBACxB;AAAA,kBACA;AAAA,kBACA,UAAU;AAAA,oBACR,IAAI;AAAA,oBACJ,MAAM;AAAA,oBACN,UAAU;AAAA,sBACR,MAAM,aAAa,QAAQ;AAAA,sBAC3B,WACE,OAAO,iBAAiB,WACpB,eACA,KAAK,UAAU,YAAY;AAAA,oBAAA;AAAA,kBACnC;AAAA,kBAEF,OAAO,gBAAgB;AAAA,gBAAA;AAAA,cAE3B;AAAA,YACF;AAAA,UACF;AAAA,QACF;AACA,YAAI,iBAAiB,aAAa,YAAY;AAC5C,gBAAM;AAAA,YACJ,MAAM;AAAA,YACN,IAAI,WAAW,KAAK,IAAI;AAAA,YACxB;AAAA,YACA;AAAA,YACA,OAAO;AAAA,cACL,SACE;AAAA,YAAA;AAAA,UACJ;AAAA,QAEJ;AAEA,cAAM;AAAA,UACJ,MAAM;AAAA,UACN,IAAI,WAAW,KAAK,IAAI;AAAA,UACxB;AAAA,UACA;AAAA,UACA,cAAc,YAAY,OAAO,IAAI,eAAe;AAAA,UACpD,OAAO,MAAM,gBACT;AAAA,YACE,cAAc,MAAM,cAAc,oBAAoB;AAAA,YACtD,kBAAkB,MAAM,cAAc,wBAAwB;AAAA,YAC9D,aAAa,MAAM,cAAc,mBAAmB;AAAA,UAAA,IAEtD;AAAA,QAAA;AAAA,MAER;AAAA,IACF;AAAA,EACF;AAAA,EAEQ,2BAA2B,MAAyB;AAC1D,UAAM,qBAAqB,CACzBA,UACG;AACH,cAAQA,OAAAA;AAAAA,QACN,KAAK;AACH,iBAAO;AAAA,QACT,KAAK;AACH,iBAAO;AAAA,QACT,KAAK;AACH,iBAAO;AAAA,QACT,KAAK;AACH,iBAAO;AAAA,MAAA;AAAA,IAEb;AACA,YAAQ,KAAK,MAAA;AAAA,MACX,KAAK;AACH,eAAO,EAAE,MAAM,KAAK,QAAA;AAAA,MACtB,KAAK;AAAA,MACL,KAAK;AAAA,MACL,KAAK;AAAA,MACL,KAAK,YAAY;AACf,cAAM,WAAW,KAAK;AAMtB,YAAI,KAAK,OAAO,SAAS,QAAQ;AAC/B,iBAAO;AAAA,YACL,YAAY;AAAA,cACV,MAAM,KAAK,OAAO;AAAA,cAClB,UAAU,UAAU,YAAY,mBAAmB,KAAK,IAAI;AAAA,YAAA;AAAA,UAC9D;AAAA,QAEJ,OAAO;AACL,iBAAO;AAAA,YACL,UAAU;AAAA,cACR,SAAS,KAAK,OAAO;AAAA,cACrB,UAAU,UAAU,YAAY,mBAAmB,KAAK,IAAI;AAAA,YAAA;AAAA,UAC9D;AAAA,QAEJ;AAAA,MACF;AAAA,MACA,SAAS;AACP,cAAM,mBAA0B;AAChC,cAAM,IAAI;AAAA,UACR,kCAAmC,iBAAiC,IAAI;AAAA,QAAA;AAAA,MAE5E;AAAA,IAAA;AAAA,EAEJ;AAAA,EAEQ,eACN,UACuC;AACvC,WAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,YAAM,OAAyB,IAAI,SAAS,cAAc,UAAU;AACpE,YAAM,QAAqB,CAAA;AAE3B,UAAI,MAAM,QAAQ,IAAI,OAAO,GAAG;AAC9B,mBAAW,eAAe,IAAI,SAAS;AACrC,gBAAM,KAAK,KAAK,2BAA2B,WAAW,CAAC;AAAA,QACzD;AAAA,MACF,WAAW,IAAI,SAAS;AACtB,cAAM,KAAK,EAAE,MAAM,IAAI,SAAS;AAAA,MAClC;AAEA,UAAI,IAAI,SAAS,eAAe,IAAI,WAAW,QAAQ;AACrD,mBAAW,YAAY,IAAI,WAAW;AACpC,cAAI,aAAsC,CAAA;AAC1C,cAAI;AACF,yBAAa,SAAS,SAAS,YAC1B,KAAK,MAAM,SAAS,SAAS,SAAS,IAIvC,CAAA;AAAA,UACN,QAAQ;AACN,yBAAa,SAAS,SAAS;AAAA,UAIjC;AAEA,gBAAM,KAAK;AAAA,YACT,cAAc;AAAA,cACZ,MAAM,SAAS,SAAS;AAAA,cACxB,MAAM;AAAA,YAAA;AAAA,UACR,CACD;AAAA,QACH;AAAA,MACF;AAEA,UAAI,IAAI,SAAS,UAAU,IAAI,YAAY;AACzC,cAAM,KAAK;AAAA,UACT,kBAAkB;AAAA,YAChB,MAAM,IAAI;AAAA,YACV,UAAU;AAAA,cACR,SAAS,IAAI,WAAW;AAAA,YAAA;AAAA,UAC1B;AAAA,QACF,CACD;AAAA,MACH;AAEA,aAAO;AAAA,QACL;AAAA,QACA,OAAO,MAAM,SAAS,IAAI,QAAQ,CAAC,EAAE,MAAM,GAAA,CAAI;AAAA,MAAA;AAAA,IAEnD,CAAC;AAAA,EACH;AAAA,EAEQ,yBACN,SACA;AACA,UAAM,YAAY,QAAQ;AAC1B,UAAM,iBAAiB,WAAW;AAClC,UAAM,iBAA4C;AAAA,MAChD,OAAO,QAAQ;AAAA,MACf,UAAU,KAAK,eAAe,QAAQ,QAAQ;AAAA,MAC9C,QAAQ;AAAA,QACN,GAAG;AAAA,QACH,aAAa,QAAQ;AAAA,QACrB,MAAM,QAAQ;AAAA,QACd,iBAAiB,QAAQ;AAAA,QACzB,gBAAgB,iBACZ;AAAA,UACE,GAAG;AAAA,UACH,eAAe,eAAe;AAAA;AAAA,YAEzB,eAAe;AAAA,cAChB;AAAA,QAAA,IAEN;AAAA,QACJ,mBAAmB,QAAQ,eAAe,KAAK,IAAI;AAAA,QACnD,OAAO,6BAA6B,QAAQ,KAAK;AAAA,MAAA;AAAA,IACnD;AAGF,WAAO;AAAA,EACT;AACF;AAMO,SAAS,iBACd,OACA,QACA,QAKA;AACA,SAAO,IAAI,kBAAkB,EAAE,QAAQ,GAAG,OAAA,GAAU,KAAK;AAC3D;AAMO,SAAS,WACd,OACA,QAKA;AACA,QAAM,SAAS,uBAAA;AACf,SAAO,iBAAiB,OAAO,QAAQ,MAAM;AAC/C;"}
+{"version":3,"file":"text.js","sources":["../../../src/adapters/text.ts"],"sourcesContent":["import { FinishReason } from '@google/genai'\nimport { BaseTextAdapter } from '@tanstack/ai/adapters'\nimport { convertToolsToProviderFormat } from '../tools/tool-converter'\nimport {\n  createGeminiClient,\n  generateId,\n  getGeminiApiKeyFromEnv,\n} from '../utils'\nimport type {\n  GEMINI_MODELS,\n  GeminiChatModelProviderOptionsByName,\n  GeminiModelInputModalitiesByName,\n} from '../model-meta'\nimport type {\n  StructuredOutputOptions,\n  StructuredOutputResult,\n} from '@tanstack/ai/adapters'\nimport type {\n  GenerateContentParameters,\n  GenerateContentResponse,\n  GoogleGenAI,\n  Part,\n  ThinkingLevel,\n} from '@google/genai'\nimport type {\n  ContentPart,\n  Modality,\n  ModelMessage,\n  StreamChunk,\n  TextOptions,\n} from '@tanstack/ai'\nimport type { ExternalTextProviderOptions } from '../text/text-provider-options'\nimport type { GeminiMessageMetadataByModality } from '../message-types'\nimport type { GeminiClientConfig } from '../utils'\n\n/**\n * Configuration for Gemini text adapter\n */\nexport interface GeminiTextConfig extends GeminiClientConfig {}\n\n/**\n * Gemini-specific provider options for text/chat\n */\nexport type GeminiTextProviderOptions = ExternalTextProviderOptions\n\n// ===========================\n// Type Resolution Helpers\n// ===========================\n\n/**\n * Resolve provider options for a specific model.\n * If the model has explicit options in the map, use those; otherwise use base options.\n */\ntype ResolveProviderOptions<TModel extends string> =\n  TModel extends keyof GeminiChatModelProviderOptionsByName\n    ? GeminiChatModelProviderOptionsByName[TModel]\n    : GeminiTextProviderOptions\n\n/**\n * Resolve input modalities for a specific model.\n * If the model has explicit modalities in the map, use those; otherwise use all modalities.\n */\ntype ResolveInputModalities<TModel extends string> =\n  TModel extends keyof GeminiModelInputModalitiesByName\n    ? GeminiModelInputModalitiesByName[TModel]\n    : readonly ['text', 'image', 'audio', 'video', 'document']\n\n// ===========================\n// Adapter Implementation\n// ===========================\n\n/**\n * Gemini Text (Chat) Adapter\n *\n * Tree-shakeable adapter for Gemini chat/text completion functionality.\n * Import only what you need for smaller bundle sizes.\n */\nexport class GeminiTextAdapter<\n  TModel extends (typeof GEMINI_MODELS)[number],\n  TProviderOptions extends object = ResolveProviderOptions<TModel>,\n  TInputModalities extends ReadonlyArray<Modality> =\n    ResolveInputModalities<TModel>,\n> extends BaseTextAdapter<\n  TModel,\n  TProviderOptions,\n  TInputModalities,\n  GeminiMessageMetadataByModality\n> {\n  readonly kind = 'text' as const\n  readonly name = 'gemini' as const\n\n  private client: GoogleGenAI\n\n  constructor(config: GeminiTextConfig, model: TModel) {\n    super({}, model)\n    this.client = createGeminiClient(config)\n  }\n\n  async *chatStream(\n    options: TextOptions<GeminiTextProviderOptions>,\n  ): AsyncIterable<StreamChunk> {\n    const mappedOptions = this.mapCommonOptionsToGemini(options)\n\n    try {\n      const result =\n        await this.client.models.generateContentStream(mappedOptions)\n\n      yield* this.processStreamChunks(result, options.model)\n    } catch (error) {\n      const timestamp = Date.now()\n      yield {\n        type: 'RUN_ERROR',\n        model: options.model,\n        timestamp,\n        error: {\n          message:\n            error instanceof Error\n              ? error.message\n              : 'An unknown error occurred during the chat stream.',\n        },\n      }\n    }\n  }\n\n  /**\n   * Generate structured output using Gemini's native JSON response format.\n   * Uses responseMimeType: 'application/json' and responseSchema for structured output.\n   * The outputSchema is already JSON Schema (converted in the ai layer).\n   */\n  async structuredOutput(\n    options: StructuredOutputOptions<GeminiTextProviderOptions>,\n  ): Promise<StructuredOutputResult<unknown>> {\n    const { chatOptions, outputSchema } = options\n\n    const mappedOptions = this.mapCommonOptionsToGemini(chatOptions)\n\n    try {\n      // Add structured output configuration\n      const result = await this.client.models.generateContent({\n        ...mappedOptions,\n        config: {\n          ...mappedOptions.config,\n          responseMimeType: 'application/json',\n          responseSchema: outputSchema,\n        },\n      })\n\n      // Extract text content from the response\n      const rawText = this.extractTextFromResponse(result)\n\n      // Parse the JSON response\n      let parsed: unknown\n      try {\n        parsed = JSON.parse(rawText)\n      } catch {\n        throw new Error(\n          `Failed to parse structured output as JSON. Content: ${rawText.slice(0, 200)}${rawText.length > 200 ? '...' : ''}`,\n        )\n      }\n\n      return {\n        data: parsed,\n        rawText,\n      }\n    } catch (error) {\n      throw new Error(\n        error instanceof Error\n          ? error.message\n          : 'An unknown error occurred during structured output generation.',\n      )\n    }\n  }\n\n  /**\n   * Extract text content from a non-streaming response\n   */\n  private extractTextFromResponse(response: GenerateContentResponse): string {\n    let textContent = ''\n\n    if (response.candidates?.[0]?.content?.parts) {\n      for (const part of response.candidates[0].content.parts) {\n        if (part.text) {\n          textContent += part.text\n        }\n      }\n    }\n\n    return textContent\n  }\n\n  private async *processStreamChunks(\n    result: AsyncGenerator<GenerateContentResponse, unknown, unknown>,\n    model: string,\n  ): AsyncIterable<StreamChunk> {\n    const timestamp = Date.now()\n    let accumulatedContent = ''\n    const toolCallMap = new Map<\n      string,\n      { name: string; args: string; index: number; started: boolean }\n    >()\n    let nextToolIndex = 0\n\n    // AG-UI lifecycle tracking\n    const runId = generateId(this.name)\n    const messageId = generateId(this.name)\n    let stepId: string | null = null\n    let hasEmittedRunStarted = false\n    let hasEmittedTextMessageStart = false\n    let hasEmittedStepStarted = false\n\n    for await (const chunk of result) {\n      // Emit RUN_STARTED on first chunk\n      if (!hasEmittedRunStarted) {\n        hasEmittedRunStarted = true\n        yield {\n          type: 'RUN_STARTED',\n          runId,\n          model,\n          timestamp,\n        }\n      }\n\n      if (chunk.candidates?.[0]?.content?.parts) {\n        const parts = chunk.candidates[0].content.parts\n\n        for (const part of parts) {\n          if (part.text) {\n            if (part.thought) {\n              // Emit STEP_STARTED on first thinking content\n              if (!hasEmittedStepStarted) {\n                hasEmittedStepStarted = true\n                stepId = generateId(this.name)\n                yield {\n                  type: 'STEP_STARTED',\n                  stepId,\n                  model,\n                  timestamp,\n                  stepType: 'thinking',\n                }\n              }\n\n              yield {\n                type: 'STEP_FINISHED',\n                stepId: stepId || generateId(this.name),\n                model,\n                timestamp,\n                delta: part.text,\n                content: part.text,\n              }\n            } else {\n              // Emit TEXT_MESSAGE_START on first text content\n              if (!hasEmittedTextMessageStart) {\n                hasEmittedTextMessageStart = true\n                yield {\n                  type: 'TEXT_MESSAGE_START',\n                  messageId,\n                  model,\n                  timestamp,\n                  role: 'assistant',\n                }\n              }\n\n              accumulatedContent += part.text\n              yield {\n                type: 'TEXT_MESSAGE_CONTENT',\n                messageId,\n                model,\n                timestamp,\n                delta: part.text,\n                content: accumulatedContent,\n              }\n            }\n          }\n\n          const functionCall = part.functionCall\n          if (functionCall) {\n            const toolCallId =\n              functionCall.id ||\n              `${functionCall.name}_${Date.now()}_${nextToolIndex}`\n            const functionArgs = functionCall.args || {}\n\n            let toolCallData = toolCallMap.get(toolCallId)\n            if (!toolCallData) {\n              toolCallData = {\n                name: functionCall.name || '',\n                args:\n                  typeof functionArgs === 'string'\n                    ? functionArgs\n                    : JSON.stringify(functionArgs),\n                index: nextToolIndex++,\n                started: false,\n              }\n              toolCallMap.set(toolCallId, toolCallData)\n            } else {\n              try {\n                const existingArgs = JSON.parse(toolCallData.args)\n                const newArgs =\n                  typeof functionArgs === 'string'\n                    ? JSON.parse(functionArgs)\n                    : functionArgs\n                const mergedArgs = { ...existingArgs, ...newArgs }\n                toolCallData.args = JSON.stringify(mergedArgs)\n              } catch {\n                toolCallData.args =\n                  typeof functionArgs === 'string'\n                    ? functionArgs\n                    : JSON.stringify(functionArgs)\n              }\n            }\n\n            // Emit TOOL_CALL_START if not already started\n            if (!toolCallData.started) {\n              toolCallData.started = true\n              yield {\n                type: 'TOOL_CALL_START',\n                toolCallId,\n                toolName: toolCallData.name,\n                model,\n                timestamp,\n                index: toolCallData.index,\n              }\n            }\n\n            // Emit TOOL_CALL_ARGS\n            yield {\n              type: 'TOOL_CALL_ARGS',\n              toolCallId,\n              model,\n              timestamp,\n              delta: toolCallData.args,\n              args: toolCallData.args,\n            }\n          }\n        }\n      } else if (chunk.data) {\n        // Emit TEXT_MESSAGE_START on first text content\n        if (!hasEmittedTextMessageStart) {\n          hasEmittedTextMessageStart = true\n          yield {\n            type: 'TEXT_MESSAGE_START',\n            messageId,\n            model,\n            timestamp,\n            role: 'assistant',\n          }\n        }\n\n        accumulatedContent += chunk.data\n        yield {\n          type: 'TEXT_MESSAGE_CONTENT',\n          messageId,\n          model,\n          timestamp,\n          delta: chunk.data,\n          content: accumulatedContent,\n        }\n      }\n\n      if (chunk.candidates?.[0]?.finishReason) {\n        const finishReason = chunk.candidates[0].finishReason\n\n        if (finishReason === FinishReason.UNEXPECTED_TOOL_CALL) {\n          if (chunk.candidates[0].content?.parts) {\n            for (const part of chunk.candidates[0].content.parts) {\n              const functionCall = part.functionCall\n              if (functionCall) {\n                const toolCallId =\n                  functionCall.id ||\n                  `${functionCall.name}_${Date.now()}_${nextToolIndex}`\n                const functionArgs = functionCall.args || {}\n\n                const argsString =\n                  typeof functionArgs === 'string'\n                    ? functionArgs\n                    : JSON.stringify(functionArgs)\n\n                toolCallMap.set(toolCallId, {\n                  name: functionCall.name || '',\n                  args: argsString,\n                  index: nextToolIndex++,\n                  started: true,\n                })\n\n                // Emit TOOL_CALL_START\n                yield {\n                  type: 'TOOL_CALL_START',\n                  toolCallId,\n                  toolName: functionCall.name || '',\n                  model,\n                  timestamp,\n                  index: nextToolIndex - 1,\n                }\n\n                // Emit TOOL_CALL_END with parsed input\n                let parsedInput: unknown = {}\n                try {\n                  parsedInput =\n                    typeof functionArgs === 'string'\n                      ? JSON.parse(functionArgs)\n                      : functionArgs\n                } catch {\n                  parsedInput = {}\n                }\n\n                yield {\n                  type: 'TOOL_CALL_END',\n                  toolCallId,\n                  toolName: functionCall.name || '',\n                  model,\n                  timestamp,\n                  input: parsedInput,\n                }\n              }\n            }\n          }\n        }\n\n        // Emit TOOL_CALL_END for all tracked tool calls\n        for (const [toolCallId, toolCallData] of toolCallMap.entries()) {\n          let parsedInput: unknown = {}\n          try {\n            parsedInput = JSON.parse(toolCallData.args)\n          } catch {\n            parsedInput = {}\n          }\n\n          yield {\n            type: 'TOOL_CALL_END',\n            toolCallId,\n            toolName: toolCallData.name,\n            model,\n            timestamp,\n            input: parsedInput,\n          }\n        }\n\n        if (finishReason === FinishReason.MAX_TOKENS) {\n          yield {\n            type: 'RUN_ERROR',\n            runId,\n            model,\n            timestamp,\n            error: {\n              message:\n                'The response was cut off because the maximum token limit was reached.',\n              code: 'max_tokens',\n            },\n          }\n        }\n\n        // Emit TEXT_MESSAGE_END if we had text content\n        if (hasEmittedTextMessageStart) {\n          yield {\n            type: 'TEXT_MESSAGE_END',\n            messageId,\n            model,\n            timestamp,\n          }\n        }\n\n        yield {\n          type: 'RUN_FINISHED',\n          runId,\n          model,\n          timestamp,\n          finishReason: toolCallMap.size > 0 ? 'tool_calls' : 'stop',\n          usage: chunk.usageMetadata\n            ? {\n                promptTokens: chunk.usageMetadata.promptTokenCount ?? 0,\n                completionTokens: chunk.usageMetadata.candidatesTokenCount ?? 0,\n                totalTokens: chunk.usageMetadata.totalTokenCount ?? 0,\n              }\n            : undefined,\n        }\n      }\n    }\n  }\n\n  private convertContentPartToGemini(part: ContentPart): Part {\n    switch (part.type) {\n      case 'text':\n        return { text: part.content }\n      case 'image':\n      case 'audio':\n      case 'video':\n      case 'document': {\n        if (part.source.type === 'data') {\n          return {\n            inlineData: {\n              data: part.source.value,\n              mimeType: part.source.mimeType,\n            },\n          }\n        } else {\n          // For URL sources, use provided mimeType or fall back to reasonable defaults\n          const defaultMimeType = {\n            image: 'image/jpeg',\n            audio: 'audio/mp3',\n            video: 'video/mp4',\n            document: 'application/pdf',\n          }[part.type]\n\n          return {\n            fileData: {\n              fileUri: part.source.value,\n              mimeType: part.source.mimeType ?? defaultMimeType,\n            },\n          }\n        }\n      }\n      default: {\n        const _exhaustiveCheck: never = part\n        throw new Error(\n          `Unsupported content part type: ${(_exhaustiveCheck as ContentPart).type}`,\n        )\n      }\n    }\n  }\n\n  private formatMessages(\n    messages: Array<ModelMessage>,\n  ): GenerateContentParameters['contents'] {\n    return messages.map((msg) => {\n      const role: 'user' | 'model' = msg.role === 'assistant' ? 'model' : 'user'\n      const parts: Array<Part> = []\n\n      if (Array.isArray(msg.content)) {\n        for (const contentPart of msg.content) {\n          parts.push(this.convertContentPartToGemini(contentPart))\n        }\n      } else if (msg.content) {\n        parts.push({ text: msg.content })\n      }\n\n      if (msg.role === 'assistant' && msg.toolCalls?.length) {\n        for (const toolCall of msg.toolCalls) {\n          let parsedArgs: Record<string, unknown> = {}\n          try {\n            parsedArgs = toolCall.function.arguments\n              ? (JSON.parse(toolCall.function.arguments) as Record<\n                  string,\n                  unknown\n                >)\n              : {}\n          } catch {\n            parsedArgs = toolCall.function.arguments as unknown as Record<\n              string,\n              unknown\n            >\n          }\n\n          parts.push({\n            functionCall: {\n              name: toolCall.function.name,\n              args: parsedArgs,\n            },\n          })\n        }\n      }\n\n      if (msg.role === 'tool' && msg.toolCallId) {\n        parts.push({\n          functionResponse: {\n            name: msg.toolCallId,\n            response: {\n              content: msg.content || '',\n            },\n          },\n        })\n      }\n\n      return {\n        role,\n        parts: parts.length > 0 ? parts : [{ text: '' }],\n      }\n    })\n  }\n\n  private mapCommonOptionsToGemini(\n    options: TextOptions<GeminiTextProviderOptions>,\n  ) {\n    const modelOpts = options.modelOptions\n    const thinkingConfig = modelOpts?.thinkingConfig\n    const requestOptions: GenerateContentParameters = {\n      model: options.model,\n      contents: this.formatMessages(options.messages),\n      config: {\n        ...modelOpts,\n        temperature: options.temperature,\n        topP: options.topP,\n        maxOutputTokens: options.maxTokens,\n        thinkingConfig: thinkingConfig\n          ? {\n              ...thinkingConfig,\n              thinkingLevel: thinkingConfig.thinkingLevel\n                ? // Enum is provided by the SDK, we use it for the type but cast it to string constants, here we just cast them back\n                  (thinkingConfig.thinkingLevel as ThinkingLevel)\n                : undefined,\n            }\n          : undefined,\n        systemInstruction: options.systemPrompts?.join('\\n'),\n        tools: convertToolsToProviderFormat(options.tools),\n      },\n    }\n\n    return requestOptions\n  }\n}\n\n/**\n * Creates a Gemini text adapter with explicit API key.\n * Type resolution happens here at the call site.\n */\nexport function createGeminiChat<TModel extends (typeof GEMINI_MODELS)[number]>(\n  model: TModel,\n  apiKey: string,\n  config?: Omit<GeminiTextConfig, 'apiKey'>,\n): GeminiTextAdapter<\n  TModel,\n  ResolveProviderOptions<TModel>,\n  ResolveInputModalities<TModel>\n> {\n  return new GeminiTextAdapter({ apiKey, ...config }, model)\n}\n\n/**\n * Creates a Gemini text adapter with automatic API key detection.\n * Type resolution happens here at the call site.\n */\nexport function geminiText<TModel extends (typeof GEMINI_MODELS)[number]>(\n  model: TModel,\n  config?: Omit<GeminiTextConfig, 'apiKey'>,\n): GeminiTextAdapter<\n  TModel,\n  ResolveProviderOptions<TModel>,\n  ResolveInputModalities<TModel>\n> {\n  const apiKey = getGeminiApiKeyFromEnv()\n  return createGeminiChat(model, apiKey, config)\n}\n"],"names":[],"mappings":";;;;AA6EO,MAAM,0BAKH,gBAKR;AAAA,EAMA,YAAY,QAA0B,OAAe;AACnD,UAAM,CAAA,GAAI,KAAK;AANjB,SAAS,OAAO;AAChB,SAAS,OAAO;AAMd,SAAK,SAAS,mBAAmB,MAAM;AAAA,EACzC;AAAA,EAEA,OAAO,WACL,SAC4B;AAC5B,UAAM,gBAAgB,KAAK,yBAAyB,OAAO;AAE3D,QAAI;AACF,YAAM,SACJ,MAAM,KAAK,OAAO,OAAO,sBAAsB,aAAa;AAE9D,aAAO,KAAK,oBAAoB,QAAQ,QAAQ,KAAK;AAAA,IACvD,SAAS,OAAO;AACd,YAAM,YAAY,KAAK,IAAA;AACvB,YAAM;AAAA,QACJ,MAAM;AAAA,QACN,OAAO,QAAQ;AAAA,QACf;AAAA,QACA,OAAO;AAAA,UACL,SACE,iBAAiB,QACb,MAAM,UACN;AAAA,QAAA;AAAA,MACR;AAAA,IAEJ;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,iBACJ,SAC0C;AAC1C,UAAM,EAAE,aAAa,aAAA,IAAiB;AAEtC,UAAM,gBAAgB,KAAK,yBAAyB,WAAW;AAE/D,QAAI;AAEF,YAAM,SAAS,MAAM,KAAK,OAAO,OAAO,gBAAgB;AAAA,QACtD,GAAG;AAAA,QACH,QAAQ;AAAA,UACN,GAAG,cAAc;AAAA,UACjB,kBAAkB;AAAA,UAClB,gBAAgB;AAAA,QAAA;AAAA,MAClB,CACD;AAGD,YAAM,UAAU,KAAK,wBAAwB,MAAM;AAGnD,UAAI;AACJ,UAAI;AACF,iBAAS,KAAK,MAAM,OAAO;AAAA,MAC7B,QAAQ;AACN,cAAM,IAAI;AAAA,UACR,uDAAuD,QAAQ,MAAM,GAAG,GAAG,CAAC,GAAG,QAAQ,SAAS,MAAM,QAAQ,EAAE;AAAA,QAAA;AAAA,MAEpH;AAEA,aAAO;AAAA,QACL,MAAM;AAAA,QACN;AAAA,MAAA;AAAA,IAEJ,SAAS,OAAO;AACd,YAAM,IAAI;AAAA,QACR,iBAAiB,QACb,MAAM,UACN;AAAA,MAAA;AAAA,IAER;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKQ,wBAAwB,UAA2C;AACzE,QAAI,cAAc;AAElB,QAAI,SAAS,aAAa,CAAC,GAAG,SAAS,OAAO;AAC5C,iBAAW,QAAQ,SAAS,WAAW,CAAC,EAAE,QAAQ,OAAO;AACvD,YAAI,KAAK,MAAM;AACb,yBAAe,KAAK;AAAA,QACtB;AAAA,MACF;AAAA,IACF;AAEA,WAAO;AAAA,EACT;AAAA,EAEA,OAAe,oBACb,QACA,OAC4B;AAC5B,UAAM,YAAY,KAAK,IAAA;AACvB,QAAI,qBAAqB;AACzB,UAAM,kCAAkB,IAAA;AAIxB,QAAI,gBAAgB;AAGpB,UAAM,QAAQ,WAAW,KAAK,IAAI;AAClC,UAAM,YAAY,WAAW,KAAK,IAAI;AACtC,QAAI,SAAwB;AAC5B,QAAI,uBAAuB;AAC3B,QAAI,6BAA6B;AACjC,QAAI,wBAAwB;AAE5B,qBAAiB,SAAS,QAAQ;AAEhC,UAAI,CAAC,sBAAsB;AACzB,+BAAuB;AACvB,cAAM;AAAA,UACJ,MAAM;AAAA,UACN;AAAA,UACA;AAAA,UACA;AAAA,QAAA;AAAA,MAEJ;AAEA,UAAI,MAAM,aAAa,CAAC,GAAG,SAAS,OAAO;AACzC,cAAM,QAAQ,MAAM,WAAW,CAAC,EAAE,QAAQ;AAE1C,mBAAW,QAAQ,OAAO;AACxB,cAAI,KAAK,MAAM;AACb,gBAAI,KAAK,SAAS;AAEhB,kBAAI,CAAC,uBAAuB;AAC1B,wCAAwB;AACxB,yBAAS,WAAW,KAAK,IAAI;AAC7B,sBAAM;AAAA,kBACJ,MAAM;AAAA,kBACN;AAAA,kBACA;AAAA,kBACA;AAAA,kBACA,UAAU;AAAA,gBAAA;AAAA,cAEd;AAEA,oBAAM;AAAA,gBACJ,MAAM;AAAA,gBACN,QAAQ,UAAU,WAAW,KAAK,IAAI;AAAA,gBACtC;AAAA,gBACA;AAAA,gBACA,OAAO,KAAK;AAAA,gBACZ,SAAS,KAAK;AAAA,cAAA;AAAA,YAElB,OAAO;AAEL,kBAAI,CAAC,4BAA4B;AAC/B,6CAA6B;AAC7B,sBAAM;AAAA,kBACJ,MAAM;AAAA,kBACN;AAAA,kBACA;AAAA,kBACA;AAAA,kBACA,MAAM;AAAA,gBAAA;AAAA,cAEV;AAEA,oCAAsB,KAAK;AAC3B,oBAAM;AAAA,gBACJ,MAAM;AAAA,gBACN;AAAA,gBACA;AAAA,gBACA;AAAA,gBACA,OAAO,KAAK;AAAA,gBACZ,SAAS;AAAA,cAAA;AAAA,YAEb;AAAA,UACF;AAEA,gBAAM,eAAe,KAAK;AAC1B,cAAI,cAAc;AAChB,kBAAM,aACJ,aAAa,MACb,GAAG,aAAa,IAAI,IAAI,KAAK,IAAA,CAAK,IAAI,aAAa;AACrD,kBAAM,eAAe,aAAa,QAAQ,CAAA;AAE1C,gBAAI,eAAe,YAAY,IAAI,UAAU;AAC7C,gBAAI,CAAC,cAAc;AACjB,6BAAe;AAAA,gBACb,MAAM,aAAa,QAAQ;AAAA,gBAC3B,MACE,OAAO,iBAAiB,WACpB,eACA,KAAK,UAAU,YAAY;AAAA,gBACjC,OAAO;AAAA,gBACP,SAAS;AAAA,cAAA;AAEX,0BAAY,IAAI,YAAY,YAAY;AAAA,YAC1C,OAAO;AACL,kBAAI;AACF,sBAAM,eAAe,KAAK,MAAM,aAAa,IAAI;AACjD,sBAAM,UACJ,OAAO,iBAAiB,WACpB,KAAK,MAAM,YAAY,IACvB;AACN,sBAAM,aAAa,EAAE,GAAG,cAAc,GAAG,QAAA;AACzC,6BAAa,OAAO,KAAK,UAAU,UAAU;AAAA,cAC/C,QAAQ;AACN,6BAAa,OACX,OAAO,iBAAiB,WACpB,eACA,KAAK,UAAU,YAAY;AAAA,cACnC;AAAA,YACF;AAGA,gBAAI,CAAC,aAAa,SAAS;AACzB,2BAAa,UAAU;AACvB,oBAAM;AAAA,gBACJ,MAAM;AAAA,gBACN;AAAA,gBACA,UAAU,aAAa;AAAA,gBACvB;AAAA,gBACA;AAAA,gBACA,OAAO,aAAa;AAAA,cAAA;AAAA,YAExB;AAGA,kBAAM;AAAA,cACJ,MAAM;AAAA,cACN;AAAA,cACA;AAAA,cACA;AAAA,cACA,OAAO,aAAa;AAAA,cACpB,MAAM,aAAa;AAAA,YAAA;AAAA,UAEvB;AAAA,QACF;AAAA,MACF,WAAW,MAAM,MAAM;AAErB,YAAI,CAAC,4BAA4B;AAC/B,uCAA6B;AAC7B,gBAAM;AAAA,YACJ,MAAM;AAAA,YACN;AAAA,YACA;AAAA,YACA;AAAA,YACA,MAAM;AAAA,UAAA;AAAA,QAEV;AAEA,8BAAsB,MAAM;AAC5B,cAAM;AAAA,UACJ,MAAM;AAAA,UACN;AAAA,UACA;AAAA,UACA;AAAA,UACA,OAAO,MAAM;AAAA,UACb,SAAS;AAAA,QAAA;AAAA,MAEb;AAEA,UAAI,MAAM,aAAa,CAAC,GAAG,cAAc;AACvC,cAAM,eAAe,MAAM,WAAW,CAAC,EAAE;AAEzC,YAAI,iBAAiB,aAAa,sBAAsB;AACtD,cAAI,MAAM,WAAW,CAAC,EAAE,SAAS,OAAO;AACtC,uBAAW,QAAQ,MAAM,WAAW,CAAC,EAAE,QAAQ,OAAO;AACpD,oBAAM,eAAe,KAAK;AAC1B,kBAAI,cAAc;AAChB,sBAAM,aACJ,aAAa,MACb,GAAG,aAAa,IAAI,IAAI,KAAK,IAAA,CAAK,IAAI,aAAa;AACrD,sBAAM,eAAe,aAAa,QAAQ,CAAA;AAE1C,sBAAM,aACJ,OAAO,iBAAiB,WACpB,eACA,KAAK,UAAU,YAAY;AAEjC,4BAAY,IAAI,YAAY;AAAA,kBAC1B,MAAM,aAAa,QAAQ;AAAA,kBAC3B,MAAM;AAAA,kBACN,OAAO;AAAA,kBACP,SAAS;AAAA,gBAAA,CACV;AAGD,sBAAM;AAAA,kBACJ,MAAM;AAAA,kBACN;AAAA,kBACA,UAAU,aAAa,QAAQ;AAAA,kBAC/B;AAAA,kBACA;AAAA,kBACA,OAAO,gBAAgB;AAAA,gBAAA;AAIzB,oBAAI,cAAuB,CAAA;AAC3B,oBAAI;AACF,gCACE,OAAO,iBAAiB,WACpB,KAAK,MAAM,YAAY,IACvB;AAAA,gBACR,QAAQ;AACN,gCAAc,CAAA;AAAA,gBAChB;AAEA,sBAAM;AAAA,kBACJ,MAAM;AAAA,kBACN;AAAA,kBACA,UAAU,aAAa,QAAQ;AAAA,kBAC/B;AAAA,kBACA;AAAA,kBACA,OAAO;AAAA,gBAAA;AAAA,cAEX;AAAA,YACF;AAAA,UACF;AAAA,QACF;AAGA,mBAAW,CAAC,YAAY,YAAY,KAAK,YAAY,WAAW;AAC9D,cAAI,cAAuB,CAAA;AAC3B,cAAI;AACF,0BAAc,KAAK,MAAM,aAAa,IAAI;AAAA,UAC5C,QAAQ;AACN,0BAAc,CAAA;AAAA,UAChB;AAEA,gBAAM;AAAA,YACJ,MAAM;AAAA,YACN;AAAA,YACA,UAAU,aAAa;AAAA,YACvB;AAAA,YACA;AAAA,YACA,OAAO;AAAA,UAAA;AAAA,QAEX;AAEA,YAAI,iBAAiB,aAAa,YAAY;AAC5C,gBAAM;AAAA,YACJ,MAAM;AAAA,YACN;AAAA,YACA;AAAA,YACA;AAAA,YACA,OAAO;AAAA,cACL,SACE;AAAA,cACF,MAAM;AAAA,YAAA;AAAA,UACR;AAAA,QAEJ;AAGA,YAAI,4BAA4B;AAC9B,gBAAM;AAAA,YACJ,MAAM;AAAA,YACN;AAAA,YACA;AAAA,YACA;AAAA,UAAA;AAAA,QAEJ;AAEA,cAAM;AAAA,UACJ,MAAM;AAAA,UACN;AAAA,UACA;AAAA,UACA;AAAA,UACA,cAAc,YAAY,OAAO,IAAI,eAAe;AAAA,UACpD,OAAO,MAAM,gBACT;AAAA,YACE,cAAc,MAAM,cAAc,oBAAoB;AAAA,YACtD,kBAAkB,MAAM,cAAc,wBAAwB;AAAA,YAC9D,aAAa,MAAM,cAAc,mBAAmB;AAAA,UAAA,IAEtD;AAAA,QAAA;AAAA,MAER;AAAA,IACF;AAAA,EACF;AAAA,EAEQ,2BAA2B,MAAyB;AAC1D,YAAQ,KAAK,MAAA;AAAA,MACX,KAAK;AACH,eAAO,EAAE,MAAM,KAAK,QAAA;AAAA,MACtB,KAAK;AAAA,MACL,KAAK;AAAA,MACL,KAAK;AAAA,MACL,KAAK,YAAY;AACf,YAAI,KAAK,OAAO,SAAS,QAAQ;AAC/B,iBAAO;AAAA,YACL,YAAY;AAAA,cACV,MAAM,KAAK,OAAO;AAAA,cAClB,UAAU,KAAK,OAAO;AAAA,YAAA;AAAA,UACxB;AAAA,QAEJ,OAAO;AAEL,gBAAM,kBAAkB;AAAA,YACtB,OAAO;AAAA,YACP,OAAO;AAAA,YACP,OAAO;AAAA,YACP,UAAU;AAAA,UAAA,EACV,KAAK,IAAI;AAEX,iBAAO;AAAA,YACL,UAAU;AAAA,cACR,SAAS,KAAK,OAAO;AAAA,cACrB,UAAU,KAAK,OAAO,YAAY;AAAA,YAAA;AAAA,UACpC;AAAA,QAEJ;AAAA,MACF;AAAA,MACA,SAAS;AACP,cAAM,mBAA0B;AAChC,cAAM,IAAI;AAAA,UACR,kCAAmC,iBAAiC,IAAI;AAAA,QAAA;AAAA,MAE5E;AAAA,IAAA;AAAA,EAEJ;AAAA,EAEQ,eACN,UACuC;AACvC,WAAO,SAAS,IAAI,CAAC,QAAQ;AAC3B,YAAM,OAAyB,IAAI,SAAS,cAAc,UAAU;AACpE,YAAM,QAAqB,CAAA;AAE3B,UAAI,MAAM,QAAQ,IAAI,OAAO,GAAG;AAC9B,mBAAW,eAAe,IAAI,SAAS;AACrC,gBAAM,KAAK,KAAK,2BAA2B,WAAW,CAAC;AAAA,QACzD;AAAA,MACF,WAAW,IAAI,SAAS;AACtB,cAAM,KAAK,EAAE,MAAM,IAAI,SAAS;AAAA,MAClC;AAEA,UAAI,IAAI,SAAS,eAAe,IAAI,WAAW,QAAQ;AACrD,mBAAW,YAAY,IAAI,WAAW;AACpC,cAAI,aAAsC,CAAA;AAC1C,cAAI;AACF,yBAAa,SAAS,SAAS,YAC1B,KAAK,MAAM,SAAS,SAAS,SAAS,IAIvC,CAAA;AAAA,UACN,QAAQ;AACN,yBAAa,SAAS,SAAS;AAAA,UAIjC;AAEA,gBAAM,KAAK;AAAA,YACT,cAAc;AAAA,cACZ,MAAM,SAAS,SAAS;AAAA,cACxB,MAAM;AAAA,YAAA;AAAA,UACR,CACD;AAAA,QACH;AAAA,MACF;AAEA,UAAI,IAAI,SAAS,UAAU,IAAI,YAAY;AACzC,cAAM,KAAK;AAAA,UACT,kBAAkB;AAAA,YAChB,MAAM,IAAI;AAAA,YACV,UAAU;AAAA,cACR,SAAS,IAAI,WAAW;AAAA,YAAA;AAAA,UAC1B;AAAA,QACF,CACD;AAAA,MACH;AAEA,aAAO;AAAA,QACL;AAAA,QACA,OAAO,MAAM,SAAS,IAAI,QAAQ,CAAC,EAAE,MAAM,GAAA,CAAI;AAAA,MAAA;AAAA,IAEnD,CAAC;AAAA,EACH;AAAA,EAEQ,yBACN,SACA;AACA,UAAM,YAAY,QAAQ;AAC1B,UAAM,iBAAiB,WAAW;AAClC,UAAM,iBAA4C;AAAA,MAChD,OAAO,QAAQ;AAAA,MACf,UAAU,KAAK,eAAe,QAAQ,QAAQ;AAAA,MAC9C,QAAQ;AAAA,QACN,GAAG;AAAA,QACH,aAAa,QAAQ;AAAA,QACrB,MAAM,QAAQ;AAAA,QACd,iBAAiB,QAAQ;AAAA,QACzB,gBAAgB,iBACZ;AAAA,UACE,GAAG;AAAA,UACH,eAAe,eAAe;AAAA;AAAA,YAEzB,eAAe;AAAA,cAChB;AAAA,QAAA,IAEN;AAAA,QACJ,mBAAmB,QAAQ,eAAe,KAAK,IAAI;AAAA,QACnD,OAAO,6BAA6B,QAAQ,KAAK;AAAA,MAAA;AAAA,IACnD;AAGF,WAAO;AAAA,EACT;AACF;AAMO,SAAS,iBACd,OACA,QACA,QAKA;AACA,SAAO,IAAI,kBAAkB,EAAE,QAAQ,GAAG,OAAA,GAAU,KAAK;AAC3D;AAMO,SAAS,WACd,OACA,QAKA;AACA,QAAM,SAAS,uBAAA;AACf,SAAO,iBAAiB,OAAO,QAAQ,MAAM;AAC/C;"}

package/dist/esm/index.d.ts

@@ -16,4 +16,4 @@ export type { GeminiTTSVoice } from './model-meta.js';
 export type { GeminiChatModelProviderOptionsByName, GeminiModelInputModalitiesByName, } from './model-meta.js';
 export type { GeminiStructuredOutputOptions, GeminiThinkingOptions, } from './text/text-provider-options.js';
 export type { GoogleGeminiTool } from './tools/index.js';
-export type { GeminiTextMetadata, GeminiImageMetadata, GeminiAudioMetadata, GeminiVideoMetadata, GeminiDocumentMetadata, GeminiImageMimeType, GeminiAudioMimeType, GeminiVideoMimeType, GeminiDocumentMimeType, GeminiMessageMetadataByModality, } from './message-types.js';
+export type { GeminiTextMetadata, GeminiImageMetadata, GeminiAudioMetadata, GeminiVideoMetadata, GeminiDocumentMetadata, GeminiMessageMetadataByModality, } from './message-types.js';

package/dist/esm/utils/client.d.ts

@@ -1,5 +1,5 @@
-import { GoogleGenAI } from '@google/genai';
-export interface GeminiClientConfig {
+import { GoogleGenAI, GoogleGenAIOptions } from '@google/genai';
+export interface GeminiClientConfig extends GoogleGenAIOptions {
     apiKey: string;
 }
 /**

package/dist/esm/utils/client.js

@@ -1,6 +1,7 @@
 import { GoogleGenAI } from "@google/genai";
 function createGeminiClient(config) {
   return new GoogleGenAI({
+    ...config,
     apiKey: config.apiKey
   });
 }

package/dist/esm/utils/client.js.map

@@ -1 +1 @@
-{"version":3,"file":"client.js","sources":["../../../src/utils/client.ts"],"sourcesContent":["import { GoogleGenAI } from '@google/genai'\n\nexport interface GeminiClientConfig {\n  apiKey: string\n}\n\n/**\n * Creates a Google Generative AI client instance\n */\nexport function createGeminiClient(config: GeminiClientConfig): GoogleGenAI {\n  return new GoogleGenAI({\n    apiKey: config.apiKey,\n  })\n}\n\n/**\n * Gets Google API key from environment variables\n * @throws Error if GOOGLE_API_KEY or GEMINI_API_KEY is not found\n */\nexport function getGeminiApiKeyFromEnv(): string {\n  const env =\n    typeof globalThis !== 'undefined' && (globalThis as any).window?.env\n      ? (globalThis as any).window.env\n      : typeof process !== 'undefined'\n        ? process.env\n        : undefined\n  const key = env?.GOOGLE_API_KEY || env?.GEMINI_API_KEY\n\n  if (!key) {\n    throw new Error(\n      'GOOGLE_API_KEY or GEMINI_API_KEY is required. Please set it in your environment variables or use the factory function with an explicit API key.',\n    )\n  }\n\n  return key\n}\n\n/**\n * Generates a unique ID with a prefix\n */\nexport function generateId(prefix: string): string {\n  return `${prefix}-${Date.now()}-${Math.random().toString(36).substring(7)}`\n}\n"],"names":[],"mappings":";AASO,SAAS,mBAAmB,QAAyC;AAC1E,SAAO,IAAI,YAAY;AAAA,IACrB,QAAQ,OAAO;AAAA,EAAA,CAChB;AACH;AAMO,SAAS,yBAAiC;AAC/C,QAAM,MACJ,OAAO,eAAe,eAAgB,WAAmB,QAAQ,MAC5D,WAAmB,OAAO,MAC3B,OAAO,YAAY,cACjB,QAAQ,MACR;AACR,QAAM,MAAM,KAAK,kBAAkB,KAAK;AAExC,MAAI,CAAC,KAAK;AACR,UAAM,IAAI;AAAA,MACR;AAAA,IAAA;AAAA,EAEJ;AAEA,SAAO;AACT;AAKO,SAAS,WAAW,QAAwB;AACjD,SAAO,GAAG,MAAM,IAAI,KAAK,KAAK,IAAI,KAAK,OAAA,EAAS,SAAS,EAAE,EAAE,UAAU,CAAC,CAAC;AAC3E;"}
+{"version":3,"file":"client.js","sources":["../../../src/utils/client.ts"],"sourcesContent":["import { GoogleGenAI } from '@google/genai'\nimport type { GoogleGenAIOptions } from '@google/genai'\n\nexport interface GeminiClientConfig extends GoogleGenAIOptions {\n  apiKey: string\n}\n\n/**\n * Creates a Google Generative AI client instance\n */\nexport function createGeminiClient(config: GeminiClientConfig): GoogleGenAI {\n  return new GoogleGenAI({\n    ...config,\n    apiKey: config.apiKey,\n  })\n}\n\n/**\n * Gets Google API key from environment variables\n * @throws Error if GOOGLE_API_KEY or GEMINI_API_KEY is not found\n */\nexport function getGeminiApiKeyFromEnv(): string {\n  const env =\n    typeof globalThis !== 'undefined' && (globalThis as any).window?.env\n      ? (globalThis as any).window.env\n      : typeof process !== 'undefined'\n        ? process.env\n        : undefined\n  const key = env?.GOOGLE_API_KEY || env?.GEMINI_API_KEY\n\n  if (!key) {\n    throw new Error(\n      'GOOGLE_API_KEY or GEMINI_API_KEY is required. Please set it in your environment variables or use the factory function with an explicit API key.',\n    )\n  }\n\n  return key\n}\n\n/**\n * Generates a unique ID with a prefix\n */\nexport function generateId(prefix: string): string {\n  return `${prefix}-${Date.now()}-${Math.random().toString(36).substring(7)}`\n}\n"],"names":[],"mappings":";AAUO,SAAS,mBAAmB,QAAyC;AAC1E,SAAO,IAAI,YAAY;AAAA,IACrB,GAAG;AAAA,IACH,QAAQ,OAAO;AAAA,EAAA,CAChB;AACH;AAMO,SAAS,yBAAiC;AAC/C,QAAM,MACJ,OAAO,eAAe,eAAgB,WAAmB,QAAQ,MAC5D,WAAmB,OAAO,MAC3B,OAAO,YAAY,cACjB,QAAQ,MACR;AACR,QAAM,MAAM,KAAK,kBAAkB,KAAK;AAExC,MAAI,CAAC,KAAK;AACR,UAAM,IAAI;AAAA,MACR;AAAA,IAAA;AAAA,EAEJ;AAEA,SAAO;AACT;AAKO,SAAS,WAAW,QAAwB;AACjD,SAAO,GAAG,MAAM,IAAI,KAAK,KAAK,IAAI,KAAK,OAAA,EAAS,SAAS,EAAE,EAAE,UAAU,CAAC,CAAC;AAC3E;"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tanstack/ai-gemini",
-  "version": "0.3.2",
+  "version": "0.4.1",
   "description": "Google Gemini adapter for TanStack AI",
   "author": "",
   "license": "MIT",
@@ -33,12 +33,12 @@
     "@google/genai": "^1.30.0"
   },
   "peerDependencies": {
-    "@tanstack/ai": "^0.2.2"
+    "@tanstack/ai": "^0.4.2"
   },
   "devDependencies": {
     "@vitest/coverage-v8": "4.0.14",
     "vite": "^7.2.7",
-    "@tanstack/ai": "0.2.2"
+    "@tanstack/ai": "0.4.2"
   },
   "scripts": {
     "build": "vite build",

package/src/adapters/summarize.ts

@@ -4,8 +4,8 @@ import {
   generateId,
   getGeminiApiKeyFromEnv,
 } from '../utils'
-
 import type { GoogleGenAI } from '@google/genai'
+import type { GeminiClientConfig } from '../utils'
 import type { SummarizeAdapter } from '@tanstack/ai/adapters'
 import type {
   StreamChunk,
@@ -13,6 +13,10 @@ import type {
   SummarizationResult,
 } from '@tanstack/ai'
 
+/**
+ * Configuration for Gemini summarize adapter
+ */
+export interface GeminiSummarizeConfig extends GeminiClientConfig {}
 /**
  * Available Gemini models for summarization
  */
@@ -66,15 +70,8 @@ export class GeminiSummarizeAdapter<
 
   private client: GoogleGenAI
 
-  constructor(
-    apiKeyOrClient: string | GoogleGenAI,
-    model: TModel,
-    _options: GeminiSummarizeAdapterOptions = {},
-  ) {
-    this.client =
-      typeof apiKeyOrClient === 'string'
-        ? createGeminiClient({ apiKey: apiKeyOrClient })
-        : apiKeyOrClient
+  constructor(config: GeminiSummarizeConfig, model: TModel) {
+    this.client = createGeminiClient(config)
     this.model = model
   }
 
@@ -164,13 +161,12 @@ export class GeminiSummarizeAdapter<
           if (part.text) {
             accumulatedContent += part.text
             yield {
-              type: 'content',
-              id,
+              type: 'TEXT_MESSAGE_CONTENT',
+              messageId: id,
               model,
               timestamp: Date.now(),
               delta: part.text,
               content: accumulatedContent,
-              role: 'assistant',
             }
           }
         }
@@ -184,8 +180,8 @@ export class GeminiSummarizeAdapter<
         finishReason === FinishReason.SAFETY
       ) {
         yield {
-          type: 'done',
-          id,
+          type: 'RUN_FINISHED',
+          runId: id,
           model,
           timestamp: Date.now(),
           finishReason:
@@ -225,9 +221,9 @@ export class GeminiSummarizeAdapter<
 export function createGeminiSummarize<TModel extends GeminiSummarizeModel>(
   apiKey: string,
   model: TModel,
-  options?: GeminiSummarizeAdapterOptions,
+  config?: Omit<GeminiSummarizeConfig, 'apiKey'>,
 ): GeminiSummarizeAdapter<TModel> {
-  return new GeminiSummarizeAdapter(apiKey, model, options)
+  return new GeminiSummarizeAdapter({ ...config, apiKey }, model)
 }
 
 /**
@@ -235,8 +231,8 @@ export function createGeminiSummarize<TModel extends GeminiSummarizeModel>(
  */
 export function geminiSummarize<TModel extends GeminiSummarizeModel>(
   model: TModel,
-  options?: GeminiSummarizeAdapterOptions,
+  config?: Omit<GeminiSummarizeConfig, 'apiKey'>,
 ): GeminiSummarizeAdapter<TModel> {
   const apiKey = getGeminiApiKeyFromEnv()
-  return new GeminiSummarizeAdapter(apiKey, model, options)
+  return new GeminiSummarizeAdapter({ ...config, apiKey }, model)
 }

package/src/adapters/text.ts

@@ -30,13 +30,7 @@ import type {
   TextOptions,
 } from '@tanstack/ai'
 import type { ExternalTextProviderOptions } from '../text/text-provider-options'
-import type {
-  GeminiAudioMetadata,
-  GeminiDocumentMetadata,
-  GeminiImageMetadata,
-  GeminiMessageMetadataByModality,
-  GeminiVideoMetadata,
-} from '../message-types'
+import type { GeminiMessageMetadataByModality } from '../message-types'
 import type { GeminiClientConfig } from '../utils'
 
 /**
@@ -115,8 +109,7 @@ export class GeminiTextAdapter<
     } catch (error) {
       const timestamp = Date.now()
       yield {
-        type: 'error',
-        id: generateId(this.name),
+        type: 'RUN_ERROR',
         model: options.model,
         timestamp,
         error: {
@@ -203,35 +196,78 @@ export class GeminiTextAdapter<
     let accumulatedContent = ''
     const toolCallMap = new Map<
       string,
-      { name: string; args: string; index: number }
+      { name: string; args: string; index: number; started: boolean }
     >()
     let nextToolIndex = 0
 
+    // AG-UI lifecycle tracking
+    const runId = generateId(this.name)
+    const messageId = generateId(this.name)
+    let stepId: string | null = null
+    let hasEmittedRunStarted = false
+    let hasEmittedTextMessageStart = false
+    let hasEmittedStepStarted = false
+
     for await (const chunk of result) {
+      // Emit RUN_STARTED on first chunk
+      if (!hasEmittedRunStarted) {
+        hasEmittedRunStarted = true
+        yield {
+          type: 'RUN_STARTED',
+          runId,
+          model,
+          timestamp,
+        }
+      }
+
       if (chunk.candidates?.[0]?.content?.parts) {
         const parts = chunk.candidates[0].content.parts
 
         for (const part of parts) {
           if (part.text) {
             if (part.thought) {
+              // Emit STEP_STARTED on first thinking content
+              if (!hasEmittedStepStarted) {
+                hasEmittedStepStarted = true
+                stepId = generateId(this.name)
+                yield {
+                  type: 'STEP_STARTED',
+                  stepId,
+                  model,
+                  timestamp,
+                  stepType: 'thinking',
+                }
+              }
+
               yield {
-                type: 'thinking',
-                content: part.text,
-                delta: part.text,
-                id: generateId(this.name),
+                type: 'STEP_FINISHED',
+                stepId: stepId || generateId(this.name),
                 model,
                 timestamp,
+                delta: part.text,
+                content: part.text,
               }
             } else {
+              // Emit TEXT_MESSAGE_START on first text content
+              if (!hasEmittedTextMessageStart) {
+                hasEmittedTextMessageStart = true
+                yield {
+                  type: 'TEXT_MESSAGE_START',
+                  messageId,
+                  model,
+                  timestamp,
+                  role: 'assistant',
+                }
+              }
+
               accumulatedContent += part.text
               yield {
-                type: 'content',
-                id: generateId(this.name),
+                type: 'TEXT_MESSAGE_CONTENT',
+                messageId,
                 model,
                 timestamp,
                 delta: part.text,
                 content: accumulatedContent,
-                role: 'assistant',
               }
             }
           }
@@ -252,6 +288,7 @@ export class GeminiTextAdapter<
                     ? functionArgs
                     : JSON.stringify(functionArgs),
                 index: nextToolIndex++,
+                started: false,
               }
               toolCallMap.set(toolCallId, toolCallData)
             } else {
@@ -271,33 +308,51 @@ export class GeminiTextAdapter<
               }
             }
 
+            // Emit TOOL_CALL_START if not already started
+            if (!toolCallData.started) {
+              toolCallData.started = true
+              yield {
+                type: 'TOOL_CALL_START',
+                toolCallId,
+                toolName: toolCallData.name,
+                model,
+                timestamp,
+                index: toolCallData.index,
+              }
+            }
+
+            // Emit TOOL_CALL_ARGS
             yield {
-              type: 'tool_call',
-              id: generateId(this.name),
+              type: 'TOOL_CALL_ARGS',
+              toolCallId,
               model,
               timestamp,
-              toolCall: {
-                id: toolCallId,
-                type: 'function',
-                function: {
-                  name: toolCallData.name,
-                  arguments: toolCallData.args,
-                },
-              },
-              index: toolCallData.index,
+              delta: toolCallData.args,
+              args: toolCallData.args,
             }
           }
         }
       } else if (chunk.data) {
+        // Emit TEXT_MESSAGE_START on first text content
+        if (!hasEmittedTextMessageStart) {
+          hasEmittedTextMessageStart = true
+          yield {
+            type: 'TEXT_MESSAGE_START',
+            messageId,
+            model,
+            timestamp,
+            role: 'assistant',
+          }
+        }
+
         accumulatedContent += chunk.data
         yield {
-          type: 'content',
-          id: generateId(this.name),
+          type: 'TEXT_MESSAGE_CONTENT',
+          messageId,
           model,
           timestamp,
           delta: chunk.data,
           content: accumulatedContent,
-          role: 'assistant',
         }
       }
 
@@ -314,53 +369,98 @@ export class GeminiTextAdapter<
                   `${functionCall.name}_${Date.now()}_${nextToolIndex}`
                 const functionArgs = functionCall.args || {}
 
+                const argsString =
+                  typeof functionArgs === 'string'
+                    ? functionArgs
+                    : JSON.stringify(functionArgs)
+
                 toolCallMap.set(toolCallId, {
                   name: functionCall.name || '',
-                  args:
-                    typeof functionArgs === 'string'
-                      ? functionArgs
-                      : JSON.stringify(functionArgs),
+                  args: argsString,
                   index: nextToolIndex++,
+                  started: true,
                 })
 
+                // Emit TOOL_CALL_START
                 yield {
-                  type: 'tool_call',
-                  id: generateId(this.name),
+                  type: 'TOOL_CALL_START',
+                  toolCallId,
+                  toolName: functionCall.name || '',
                   model,
                   timestamp,
-                  toolCall: {
-                    id: toolCallId,
-                    type: 'function',
-                    function: {
-                      name: functionCall.name || '',
-                      arguments:
-                        typeof functionArgs === 'string'
-                          ? functionArgs
-                          : JSON.stringify(functionArgs),
-                    },
-                  },
                   index: nextToolIndex - 1,
                 }
+
+                // Emit TOOL_CALL_END with parsed input
+                let parsedInput: unknown = {}
+                try {
+                  parsedInput =
+                    typeof functionArgs === 'string'
+                      ? JSON.parse(functionArgs)
+                      : functionArgs
+                } catch {
+                  parsedInput = {}
+                }
+
+                yield {
+                  type: 'TOOL_CALL_END',
+                  toolCallId,
+                  toolName: functionCall.name || '',
+                  model,
+                  timestamp,
+                  input: parsedInput,
+                }
               }
             }
           }
         }
+
+        // Emit TOOL_CALL_END for all tracked tool calls
+        for (const [toolCallId, toolCallData] of toolCallMap.entries()) {
+          let parsedInput: unknown = {}
+          try {
+            parsedInput = JSON.parse(toolCallData.args)
+          } catch {
+            parsedInput = {}
+          }
+
+          yield {
+            type: 'TOOL_CALL_END',
+            toolCallId,
+            toolName: toolCallData.name,
+            model,
+            timestamp,
+            input: parsedInput,
+          }
+        }
+
         if (finishReason === FinishReason.MAX_TOKENS) {
           yield {
-            type: 'error',
-            id: generateId(this.name),
+            type: 'RUN_ERROR',
+            runId,
             model,
             timestamp,
             error: {
               message:
                 'The response was cut off because the maximum token limit was reached.',
+              code: 'max_tokens',
             },
           }
         }
 
+        // Emit TEXT_MESSAGE_END if we had text content
+        if (hasEmittedTextMessageStart) {
+          yield {
+            type: 'TEXT_MESSAGE_END',
+            messageId,
+            model,
+            timestamp,
+          }
+        }
+
         yield {
-          type: 'done',
-          id: generateId(this.name),
+          type: 'RUN_FINISHED',
+          runId,
           model,
           timestamp,
           finishReason: toolCallMap.size > 0 ? 'tool_calls' : 'stop',
@@ -377,20 +477,6 @@ export class GeminiTextAdapter<
   }
 
   private convertContentPartToGemini(part: ContentPart): Part {
-    const getDefaultFileType = (
-      part: 'image' | 'audio' | 'video' | 'document',
-    ) => {
-      switch (part) {
-        case 'image':
-          return 'image/jpeg'
-        case 'audio':
-          return 'audio/mp3'
-        case 'video':
-          return 'video/mp4'
-        case 'document':
-          return 'application/pdf'
-      }
-    }
     switch (part.type) {
       case 'text':
         return { text: part.content }
@@ -398,24 +484,26 @@ export class GeminiTextAdapter<
       case 'audio':
       case 'video':
       case 'document': {
-        const metadata = part.metadata as
-          | GeminiDocumentMetadata
-          | GeminiImageMetadata
-          | GeminiVideoMetadata
-          | GeminiAudioMetadata
-          | undefined
         if (part.source.type === 'data') {
           return {
             inlineData: {
               data: part.source.value,
-              mimeType: metadata?.mimeType ?? getDefaultFileType(part.type),
+              mimeType: part.source.mimeType,
             },
           }
         } else {
+          // For URL sources, use provided mimeType or fall back to reasonable defaults
+          const defaultMimeType = {
+            image: 'image/jpeg',
+            audio: 'audio/mp3',
+            video: 'video/mp4',
+            document: 'application/pdf',
+          }[part.type]
+
           return {
             fileData: {
               fileUri: part.source.value,
-              mimeType: metadata?.mimeType ?? getDefaultFileType(part.type),
+              mimeType: part.source.mimeType ?? defaultMimeType,
             },
           }
         }

package/src/index.ts

@@ -79,9 +79,5 @@ export type {
   GeminiAudioMetadata,
   GeminiVideoMetadata,
   GeminiDocumentMetadata,
-  GeminiImageMimeType,
-  GeminiAudioMimeType,
-  GeminiVideoMimeType,
-  GeminiDocumentMimeType,
   GeminiMessageMetadataByModality,
 } from './message-types'

package/src/utils/client.ts

@@ -1,6 +1,7 @@
 import { GoogleGenAI } from '@google/genai'
+import type { GoogleGenAIOptions } from '@google/genai'
 
-export interface GeminiClientConfig {
+export interface GeminiClientConfig extends GoogleGenAIOptions {
   apiKey: string
 }
 
@@ -9,6 +10,7 @@ export interface GeminiClientConfig {
  */
 export function createGeminiClient(config: GeminiClientConfig): GoogleGenAI {
   return new GoogleGenAI({
+    ...config,
     apiKey: config.apiKey,
   })
 }