@tanstack/ai-gemini 0.0.3 → 0.1.0

package/dist/esm/tools/tool-converter.js.map

@@ -1 +1 @@
-{"version":3,"file":"tool-converter.js","sources":["../../../src/tools/tool-converter.ts"],"sourcesContent":["import { convertZodToJsonSchema } from '@tanstack/ai'\nimport { convertCodeExecutionToolToAdapterFormat } from './code-execution-tool'\nimport { convertComputerUseToolToAdapterFormat } from './computer-use-tool'\nimport { convertFileSearchToolToAdapterFormat } from './file-search-tool'\nimport { convertGoogleMapsToolToAdapterFormat } from './google-maps-tool'\nimport { convertGoogleSearchRetrievalToolToAdapterFormat } from './google-search-retriveal-tool'\nimport { convertGoogleSearchToolToAdapterFormat } from './google-search-tool'\nimport { convertUrlContextToolToAdapterFormat } from './url-context-tool'\nimport type { Tool } from '@tanstack/ai'\nimport type { ToolUnion } from '@google/genai'\n\n/**\n * Converts standard Tool format to Gemini-specific tool format\n *\n * @param tools - Array of standard Tool objects\n * @returns Array of Gemini-specific tool definitions\n *\n * @example\n * ```typescript\n * const tools: Tool[] = [{\n *   name: \"get_weather\",\n *   description: \"Get weather for a location\",\n *   inputSchema: z.object({\n *     location: z.string()\n *   })\n * }];\n *\n * const geminiTools = convertToolsToProviderFormat(tools);\n * ```\n */\nexport function convertToolsToProviderFormat<TTool extends Tool>(\n  tools: Array<TTool> | undefined,\n): Array<ToolUnion> {\n  if (!tools || tools.length === 0) {\n    return []\n  }\n  const result: Array<ToolUnion> = []\n  const functionDeclarations: Array<{\n    name: string\n    description?: string\n    parameters?: any\n  }> = []\n\n  // Process each tool and group function declarations together\n  for (const tool of tools) {\n    const name = tool.name\n\n    switch (name) {\n      case 'code_execution':\n        result.push(convertCodeExecutionToolToAdapterFormat(tool))\n        break\n      case 'computer_use':\n        result.push(convertComputerUseToolToAdapterFormat(tool))\n        break\n      case 'file_search':\n        result.push(convertFileSearchToolToAdapterFormat(tool))\n        break\n      case 'google_maps':\n        result.push(convertGoogleMapsToolToAdapterFormat(tool))\n        break\n      case 'google_search_retrieval':\n        result.push(convertGoogleSearchRetrievalToolToAdapterFormat(tool))\n        break\n      case 'google_search':\n        result.push(convertGoogleSearchToolToAdapterFormat(tool))\n        break\n      case 'url_context':\n        result.push(convertUrlContextToolToAdapterFormat(tool))\n        break\n      default:\n        // Collect function declarations to group together\n        // Description is required for Gemini function declarations\n        if (!tool.description) {\n          throw new Error(\n            `Tool ${tool.name} requires a description for Gemini adapter`,\n          )\n        }\n\n        // Convert Zod schema to JSON Schema\n        const jsonSchema = convertZodToJsonSchema(tool.inputSchema)\n\n        functionDeclarations.push({\n          name: tool.name,\n          description: tool.description,\n          parameters: jsonSchema,\n        })\n        break\n    }\n  }\n\n  // If we have function declarations, add them as a single tool\n  if (functionDeclarations.length > 0) {\n    result.push({\n      functionDeclarations: functionDeclarations,\n    })\n  }\n\n  return result\n}\n"],"names":[],"mappings":";;;;;;;;AA8BO,SAAS,6BACd,OACkB;AAClB,MAAI,CAAC,SAAS,MAAM,WAAW,GAAG;AAChC,WAAO,CAAA;AAAA,EACT;AACA,QAAM,SAA2B,CAAA;AACjC,QAAM,uBAID,CAAA;AAGL,aAAW,QAAQ,OAAO;AACxB,UAAM,OAAO,KAAK;AAElB,YAAQ,MAAA;AAAA,MACN,KAAK;AACH,eAAO,KAAK,wCAA4C,CAAC;AACzD;AAAA,MACF,KAAK;AACH,eAAO,KAAK,sCAAsC,IAAI,CAAC;AACvD;AAAA,MACF,KAAK;AACH,eAAO,KAAK,qCAAqC,IAAI,CAAC;AACtD;AAAA,MACF,KAAK;AACH,eAAO,KAAK,qCAAqC,IAAI,CAAC;AACtD;AAAA,MACF,KAAK;AACH,eAAO,KAAK,gDAAgD,IAAI,CAAC;AACjE;AAAA,MACF,KAAK;AACH,eAAO,KAAK,uCAAuC,IAAI,CAAC;AACxD;AAAA,MACF,KAAK;AACH,eAAO,KAAK,qCAAyC,CAAC;AACtD;AAAA,MACF;AAGE,YAAI,CAAC,KAAK,aAAa;AACrB,gBAAM,IAAI;AAAA,YACR,QAAQ,KAAK,IAAI;AAAA,UAAA;AAAA,QAErB;AAGA,cAAM,aAAa,uBAAuB,KAAK,WAAW;AAE1D,6BAAqB,KAAK;AAAA,UACxB,MAAM,KAAK;AAAA,UACX,aAAa,KAAK;AAAA,UAClB,YAAY;AAAA,QAAA,CACb;AACD;AAAA,IAAA;AAAA,EAEN;AAGA,MAAI,qBAAqB,SAAS,GAAG;AACnC,WAAO,KAAK;AAAA,MACV;AAAA,IAAA,CACD;AAAA,EACH;AAEA,SAAO;AACT;"}
+{"version":3,"file":"tool-converter.js","sources":["../../../src/tools/tool-converter.ts"],"sourcesContent":["import { convertCodeExecutionToolToAdapterFormat } from './code-execution-tool'\nimport { convertComputerUseToolToAdapterFormat } from './computer-use-tool'\nimport { convertFileSearchToolToAdapterFormat } from './file-search-tool'\nimport { convertGoogleMapsToolToAdapterFormat } from './google-maps-tool'\nimport { convertGoogleSearchRetrievalToolToAdapterFormat } from './google-search-retriveal-tool'\nimport { convertGoogleSearchToolToAdapterFormat } from './google-search-tool'\nimport { convertUrlContextToolToAdapterFormat } from './url-context-tool'\nimport type { Tool } from '@tanstack/ai'\nimport type { ToolUnion } from '@google/genai'\n\n/**\n * Converts standard Tool format to Gemini-specific tool format\n *\n * @param tools - Array of standard Tool objects\n * @returns Array of Gemini-specific tool definitions\n *\n * @example\n * ```typescript\n * const tools: Tool[] = [{\n *   name: \"get_weather\",\n *   description: \"Get weather for a location\",\n *   inputSchema: z.object({\n *     location: z.string()\n *   })\n * }];\n *\n * const geminiTools = convertToolsToProviderFormat(tools);\n * ```\n */\nexport function convertToolsToProviderFormat<TTool extends Tool>(\n  tools: Array<TTool> | undefined,\n): Array<ToolUnion> {\n  if (!tools || tools.length === 0) {\n    return []\n  }\n  const result: Array<ToolUnion> = []\n  const functionDeclarations: Array<{\n    name: string\n    description?: string\n    parameters?: any\n  }> = []\n\n  // Process each tool and group function declarations together\n  for (const tool of tools) {\n    const name = tool.name\n\n    switch (name) {\n      case 'code_execution':\n        result.push(convertCodeExecutionToolToAdapterFormat(tool))\n        break\n      case 'computer_use':\n        result.push(convertComputerUseToolToAdapterFormat(tool))\n        break\n      case 'file_search':\n        result.push(convertFileSearchToolToAdapterFormat(tool))\n        break\n      case 'google_maps':\n        result.push(convertGoogleMapsToolToAdapterFormat(tool))\n        break\n      case 'google_search_retrieval':\n        result.push(convertGoogleSearchRetrievalToolToAdapterFormat(tool))\n        break\n      case 'google_search':\n        result.push(convertGoogleSearchToolToAdapterFormat(tool))\n        break\n      case 'url_context':\n        result.push(convertUrlContextToolToAdapterFormat(tool))\n        break\n      default:\n        // Collect function declarations to group together\n        // Description is required for Gemini function declarations\n        if (!tool.description) {\n          throw new Error(\n            `Tool ${tool.name} requires a description for Gemini adapter`,\n          )\n        }\n\n        // Tool schemas are already converted to JSON Schema in the ai layer\n        functionDeclarations.push({\n          name: tool.name,\n          description: tool.description,\n          parameters: tool.inputSchema ?? {\n            type: 'object',\n            properties: {},\n            required: [],\n          },\n        })\n        break\n    }\n  }\n\n  // If we have function declarations, add them as a single tool\n  if (functionDeclarations.length > 0) {\n    result.push({\n      functionDeclarations: functionDeclarations,\n    })\n  }\n\n  return result\n}\n"],"names":[],"mappings":";;;;;;;AA6BO,SAAS,6BACd,OACkB;AAClB,MAAI,CAAC,SAAS,MAAM,WAAW,GAAG;AAChC,WAAO,CAAA;AAAA,EACT;AACA,QAAM,SAA2B,CAAA;AACjC,QAAM,uBAID,CAAA;AAGL,aAAW,QAAQ,OAAO;AACxB,UAAM,OAAO,KAAK;AAElB,YAAQ,MAAA;AAAA,MACN,KAAK;AACH,eAAO,KAAK,wCAA4C,CAAC;AACzD;AAAA,MACF,KAAK;AACH,eAAO,KAAK,sCAAsC,IAAI,CAAC;AACvD;AAAA,MACF,KAAK;AACH,eAAO,KAAK,qCAAqC,IAAI,CAAC;AACtD;AAAA,MACF,KAAK;AACH,eAAO,KAAK,qCAAqC,IAAI,CAAC;AACtD;AAAA,MACF,KAAK;AACH,eAAO,KAAK,gDAAgD,IAAI,CAAC;AACjE;AAAA,MACF,KAAK;AACH,eAAO,KAAK,uCAAuC,IAAI,CAAC;AACxD;AAAA,MACF,KAAK;AACH,eAAO,KAAK,qCAAyC,CAAC;AACtD;AAAA,MACF;AAGE,YAAI,CAAC,KAAK,aAAa;AACrB,gBAAM,IAAI;AAAA,YACR,QAAQ,KAAK,IAAI;AAAA,UAAA;AAAA,QAErB;AAGA,6BAAqB,KAAK;AAAA,UACxB,MAAM,KAAK;AAAA,UACX,aAAa,KAAK;AAAA,UAClB,YAAY,KAAK,eAAe;AAAA,YAC9B,MAAM;AAAA,YACN,YAAY,CAAA;AAAA,YACZ,UAAU,CAAA;AAAA,UAAC;AAAA,QACb,CACD;AACD;AAAA,IAAA;AAAA,EAEN;AAGA,MAAI,qBAAqB,SAAS,GAAG;AACnC,WAAO,KAAK;AAAA,MACV;AAAA,IAAA,CACD;AAAA,EACH;AAEA,SAAO;AACT;"}

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

@@ -0,0 +1,17 @@
+import { GoogleGenAI } from '@google/genai';
+export interface GeminiClientConfig {
+    apiKey: string;
+}
+/**
+ * Creates a Google Generative AI client instance
+ */
+export declare function createGeminiClient(config: GeminiClientConfig): GoogleGenAI;
+/**
+ * Gets Google API key from environment variables
+ * @throws Error if GOOGLE_API_KEY or GEMINI_API_KEY is not found
+ */
+export declare function getGeminiApiKeyFromEnv(): string;
+/**
+ * Generates a unique ID with a prefix
+ */
+export declare function generateId(prefix: string): string;

package/dist/esm/utils/client.js

@@ -0,0 +1,25 @@
+import { GoogleGenAI } from "@google/genai";
+function createGeminiClient(config) {
+  return new GoogleGenAI({
+    apiKey: config.apiKey
+  });
+}
+function getGeminiApiKeyFromEnv() {
+  const env = typeof globalThis !== "undefined" && globalThis.window?.env ? globalThis.window.env : typeof process !== "undefined" ? process.env : void 0;
+  const key = env?.GOOGLE_API_KEY || env?.GEMINI_API_KEY;
+  if (!key) {
+    throw new Error(
+      "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."
+    );
+  }
+  return key;
+}
+function generateId(prefix) {
+  return `${prefix}-${Date.now()}-${Math.random().toString(36).substring(7)}`;
+}
+export {
+  createGeminiClient,
+  generateId,
+  getGeminiApiKeyFromEnv
+};
+//# sourceMappingURL=client.js.map

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

@@ -0,0 +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;"}

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

@@ -0,0 +1 @@
+export { createGeminiClient, generateId, getGeminiApiKeyFromEnv, type GeminiClientConfig, } from './client.js';

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@tanstack/ai-gemini",
-  "version": "0.0.3",
+  "version": "0.1.0",
   "description": "Google Gemini adapter for TanStack AI",
   "author": "",
   "license": "MIT",
@@ -31,14 +31,15 @@
   ],
   "dependencies": {
     "@google/genai": "^1.30.0",
-    "@tanstack/ai": "0.0.3"
+    "@tanstack/ai": "0.1.0"
   },
   "devDependencies": {
     "@vitest/coverage-v8": "4.0.14",
-    "vite": "^7.2.4"
+    "vite": "^7.2.7"
   },
   "peerDependencies": {
-    "@tanstack/ai": "0.0.3"
+    "zod": "^4.0.0",
+    "@tanstack/ai": "0.1.0"
   },
   "scripts": {
     "build": "vite build",

package/src/adapters/image.ts

@@ -0,0 +1,188 @@
+import { BaseImageAdapter } from '@tanstack/ai/adapters'
+import {
+  createGeminiClient,
+  generateId,
+  getGeminiApiKeyFromEnv,
+} from '../utils'
+import {
+  sizeToAspectRatio,
+  validateImageSize,
+  validateNumberOfImages,
+  validatePrompt,
+} from '../image/image-provider-options'
+import type { GEMINI_IMAGE_MODELS } from '../model-meta'
+import type {
+  GeminiImageModelProviderOptionsByName,
+  GeminiImageModelSizeByName,
+  GeminiImageProviderOptions,
+} from '../image/image-provider-options'
+import type {
+  GeneratedImage,
+  ImageGenerationOptions,
+  ImageGenerationResult,
+} from '@tanstack/ai'
+import type {
+  GenerateImagesConfig,
+  GenerateImagesResponse,
+  GoogleGenAI,
+} from '@google/genai'
+import type { GeminiClientConfig } from '../utils'
+
+/**
+ * Configuration for Gemini image adapter
+ */
+export interface GeminiImageConfig extends GeminiClientConfig {}
+
+/** Model type for Gemini Image */
+export type GeminiImageModel = (typeof GEMINI_IMAGE_MODELS)[number]
+
+/**
+ * Gemini Image Generation Adapter
+ *
+ * Tree-shakeable adapter for Gemini Imagen image generation functionality.
+ * Supports Imagen 3 and Imagen 4 models.
+ *
+ * Features:
+ * - Aspect ratio-based image sizing
+ * - Person generation controls
+ * - Safety filtering
+ * - Watermark options
+ */
+export class GeminiImageAdapter<
+  TModel extends GeminiImageModel,
+> extends BaseImageAdapter<
+  TModel,
+  GeminiImageProviderOptions,
+  GeminiImageModelProviderOptionsByName,
+  GeminiImageModelSizeByName
+> {
+  readonly kind = 'image' as const
+  readonly name = 'gemini' as const
+
+  // Type-only property - never assigned at runtime
+  declare '~types': {
+    providerOptions: GeminiImageProviderOptions
+    modelProviderOptionsByName: GeminiImageModelProviderOptionsByName
+    modelSizeByName: GeminiImageModelSizeByName
+  }
+
+  private client: GoogleGenAI
+
+  constructor(config: GeminiImageConfig, model: TModel) {
+    super({}, model)
+    this.client = createGeminiClient(config)
+  }
+
+  async generateImages(
+    options: ImageGenerationOptions<GeminiImageProviderOptions>,
+  ): Promise<ImageGenerationResult> {
+    const { model, prompt, numberOfImages, size } = options
+
+    // Validate inputs
+    validatePrompt({ prompt, model })
+    validateImageSize(model, size)
+    validateNumberOfImages(model, numberOfImages)
+
+    // Build request config
+    const config = this.buildConfig(options)
+
+    const response = await this.client.models.generateImages({
+      model,
+      prompt,
+      config,
+    })
+
+    return this.transformResponse(model, response)
+  }
+
+  private buildConfig(
+    options: ImageGenerationOptions<GeminiImageProviderOptions>,
+  ): GenerateImagesConfig {
+    const { size, numberOfImages, modelOptions } = options
+
+    return {
+      numberOfImages: numberOfImages ?? 1,
+      // Map size to aspect ratio if provided (modelOptions.aspectRatio will override)
+      aspectRatio: size ? sizeToAspectRatio(size) : undefined,
+      ...modelOptions,
+    }
+  }
+
+  private transformResponse(
+    model: string,
+    response: GenerateImagesResponse,
+  ): ImageGenerationResult {
+    const images: Array<GeneratedImage> = (response.generatedImages ?? []).map(
+      (item) => ({
+        b64Json: item.image?.imageBytes,
+        revisedPrompt: item.enhancedPrompt,
+      }),
+    )
+
+    return {
+      id: generateId(this.name),
+      model,
+      images,
+      usage: undefined,
+    }
+  }
+}
+
+/**
+ * Creates a Gemini image adapter with explicit API key.
+ * Type resolution happens here at the call site.
+ *
+ * @param model - The model name (e.g., 'imagen-3.0-generate-002')
+ * @param apiKey - Your Google API key
+ * @param config - Optional additional configuration
+ * @returns Configured Gemini image adapter instance with resolved types
+ *
+ * @example
+ * ```typescript
+ * const adapter = createGeminiImage('imagen-3.0-generate-002', "your-api-key");
+ *
+ * const result = await generateImage({
+ *   adapter,
+ *   prompt: 'A cute baby sea otter'
+ * });
+ * ```
+ */
+export function createGeminiImage<TModel extends GeminiImageModel>(
+  model: TModel,
+  apiKey: string,
+  config?: Omit<GeminiImageConfig, 'apiKey'>,
+): GeminiImageAdapter<TModel> {
+  return new GeminiImageAdapter({ apiKey, ...config }, model)
+}
+
+/**
+ * Creates a Gemini image adapter with automatic API key detection from environment variables.
+ * Type resolution happens here at the call site.
+ *
+ * Looks for `GOOGLE_API_KEY` or `GEMINI_API_KEY` in:
+ * - `process.env` (Node.js)
+ * - `window.env` (Browser with injected env)
+ *
+ * @param model - The model name (e.g., 'imagen-4.0-generate-001')
+ * @param config - Optional configuration (excluding apiKey which is auto-detected)
+ * @returns Configured Gemini image adapter instance with resolved types
+ * @throws Error if GOOGLE_API_KEY or GEMINI_API_KEY is not found in environment
+ *
+ * @example
+ * ```typescript
+ * // Automatically uses GOOGLE_API_KEY from environment
+ * const adapter = geminiImage('imagen-4.0-generate-001');
+ *
+ * const result = await generateImage({
+ *   adapter,
+ *   prompt: 'A beautiful sunset over mountains'
+ * });
+ * ```
+ */
+export function geminiImage<TModel extends GeminiImageModel>(
+  model: TModel,
+  config?: Omit<GeminiImageConfig, 'apiKey'>,
+): GeminiImageAdapter<TModel> {
+  const apiKey = getGeminiApiKeyFromEnv()
+  return createGeminiImage(model, apiKey, config)
+}

package/src/adapters/summarize.ts

@@ -0,0 +1,242 @@
+import { FinishReason } from '@google/genai'
+import {
+  createGeminiClient,
+  generateId,
+  getGeminiApiKeyFromEnv,
+} from '../utils'
+
+import type { GoogleGenAI } from '@google/genai'
+import type { SummarizeAdapter } from '@tanstack/ai/adapters'
+import type {
+  StreamChunk,
+  SummarizationOptions,
+  SummarizationResult,
+} from '@tanstack/ai'
+
+/**
+ * Available Gemini models for summarization
+ */
+export const GeminiSummarizeModels = [
+  'gemini-2.0-flash',
+  'gemini-1.5-flash',
+  'gemini-1.5-pro',
+  'gemini-2.0-flash-lite',
+] as const
+
+export type GeminiSummarizeModel = (typeof GeminiSummarizeModels)[number]
+
+/**
+ * Provider-specific options for Gemini summarization
+ */
+export interface GeminiSummarizeProviderOptions {
+  /** Generation configuration */
+  generationConfig?: {
+    temperature?: number
+    topP?: number
+    topK?: number
+    maxOutputTokens?: number
+    stopSequences?: Array<string>
+  }
+  /** Safety settings */
+  safetySettings?: Array<{
+    category: string
+    threshold: string
+  }>
+}
+
+export interface GeminiSummarizeAdapterOptions {
+  // Additional adapter options can be added here
+}
+
+/**
+ * Gemini Summarize Adapter
+ * A tree-shakeable summarization adapter for Google Gemini
+ */
+export class GeminiSummarizeAdapter<
+  TModel extends GeminiSummarizeModel,
+> implements SummarizeAdapter<TModel, GeminiSummarizeProviderOptions> {
+  readonly kind = 'summarize' as const
+  readonly name = 'gemini' as const
+  readonly model: TModel
+
+  // Type-only property - never assigned at runtime
+  declare '~types': {
+    providerOptions: GeminiSummarizeProviderOptions
+  }
+
+  private client: GoogleGenAI
+
+  constructor(
+    apiKeyOrClient: string | GoogleGenAI,
+    model: TModel,
+    _options: GeminiSummarizeAdapterOptions = {},
+  ) {
+    this.client =
+      typeof apiKeyOrClient === 'string'
+        ? createGeminiClient({ apiKey: apiKeyOrClient })
+        : apiKeyOrClient
+    this.model = model
+  }
+
+  async summarize(options: SummarizationOptions): Promise<SummarizationResult> {
+    const model = options.model
+
+    // Build the system prompt based on format
+    const formatInstructions = this.getFormatInstructions(options.style)
+    const lengthInstructions = options.maxLength
+      ? ` Keep the summary under ${options.maxLength} tokens.`
+      : ''
+
+    const systemPrompt = `You are a helpful assistant that summarizes text. ${formatInstructions}${lengthInstructions}`
+
+    const response = await this.client.models.generateContent({
+      model,
+      contents: [
+        {
+          role: 'user',
+          parts: [
+            { text: `Please summarize the following:\n\n${options.text}` },
+          ],
+        },
+      ],
+      config: {
+        systemInstruction: systemPrompt,
+      },
+    })
+
+    const summary = response.text ?? ''
+    const inputTokens = response.usageMetadata?.promptTokenCount ?? 0
+    const outputTokens = response.usageMetadata?.candidatesTokenCount ?? 0
+
+    return {
+      id: generateId('sum'),
+      model,
+      summary,
+      usage: {
+        promptTokens: inputTokens,
+        completionTokens: outputTokens,
+        totalTokens: inputTokens + outputTokens,
+      },
+    }
+  }
+
+  async *summarizeStream(
+    options: SummarizationOptions,
+  ): AsyncIterable<StreamChunk> {
+    const model = options.model
+    const id = generateId('sum')
+    let accumulatedContent = ''
+    let inputTokens = 0
+    let outputTokens = 0
+
+    // Build the system prompt based on format
+    const formatInstructions = this.getFormatInstructions(options.style)
+    const lengthInstructions = options.maxLength
+      ? ` Keep the summary under ${options.maxLength} words.`
+      : ''
+
+    const systemPrompt = `You are a helpful assistant that summarizes text. ${formatInstructions}${lengthInstructions}`
+
+    const result = await this.client.models.generateContentStream({
+      model,
+      contents: [
+        {
+          role: 'user',
+          parts: [
+            { text: `Please summarize the following:\n\n${options.text}` },
+          ],
+        },
+      ],
+      config: {
+        systemInstruction: systemPrompt,
+      },
+    })
+
+    for await (const chunk of result) {
+      // Track usage metadata
+      if (chunk.usageMetadata) {
+        inputTokens = chunk.usageMetadata.promptTokenCount ?? inputTokens
+        outputTokens = chunk.usageMetadata.candidatesTokenCount ?? outputTokens
+      }
+
+      if (chunk.candidates?.[0]?.content?.parts) {
+        for (const part of chunk.candidates[0].content.parts) {
+          if (part.text) {
+            accumulatedContent += part.text
+            yield {
+              type: 'content',
+              id,
+              model,
+              timestamp: Date.now(),
+              delta: part.text,
+              content: accumulatedContent,
+              role: 'assistant',
+            }
+          }
+        }
+      }
+
+      // Check for finish reason
+      const finishReason = chunk.candidates?.[0]?.finishReason
+      if (
+        finishReason === FinishReason.STOP ||
+        finishReason === FinishReason.MAX_TOKENS ||
+        finishReason === FinishReason.SAFETY
+      ) {
+        yield {
+          type: 'done',
+          id,
+          model,
+          timestamp: Date.now(),
+          finishReason:
+            finishReason === FinishReason.STOP
+              ? 'stop'
+              : finishReason === FinishReason.MAX_TOKENS
+                ? 'length'
+                : 'content_filter',
+          usage: {
+            promptTokens: inputTokens,
+            completionTokens: outputTokens,
+            totalTokens: inputTokens + outputTokens,
+          },
+        }
+      }
+    }
+  }
+
+  private getFormatInstructions(
+    style?: 'paragraph' | 'bullet-points' | 'concise',
+  ): string {
+    switch (style) {
+      case 'bullet-points':
+        return 'Provide the summary as bullet points.'
+      case 'concise':
+        return 'Provide a very brief one or two sentence summary.'
+      case 'paragraph':
+      default:
+        return 'Provide the summary in paragraph form.'
+    }
+  }
+}
+
+/**
+ * Creates a Gemini summarize adapter with explicit API key and model
+ */
+export function createGeminiSummarize<TModel extends GeminiSummarizeModel>(
+  apiKey: string,
+  model: TModel,
+  options?: GeminiSummarizeAdapterOptions,
+): GeminiSummarizeAdapter<TModel> {
+  return new GeminiSummarizeAdapter(apiKey, model, options)
+}
+
+/**
+ * Creates a Gemini summarize adapter with API key from environment and required model
+ */
+export function geminiSummarize<TModel extends GeminiSummarizeModel>(
+  model: TModel,
+  options?: GeminiSummarizeAdapterOptions,
+): GeminiSummarizeAdapter<TModel> {
+  const apiKey = getGeminiApiKeyFromEnv()
+  return new GeminiSummarizeAdapter(apiKey, model, options)
+}