@ai-sdk/google 2.0.0-beta.1 → 2.0.0-beta.11

package/dist/index.mjs

@@ -1,9 +1,6 @@
 // src/google-provider.ts
 import {
-  NoSuchModelError
-} from "@ai-sdk/provider";
-import {
-  generateId,
+  generateId as generateId2,
   loadApiKey,
   withoutTrailingSlash
 } from "@ai-sdk/provider-utils";
@@ -19,11 +16,11 @@ import {
   postJsonToApi,
   resolve
 } from "@ai-sdk/provider-utils";
-import { z as z3 } from "zod";
+import { z as z3 } from "zod/v4";
 
 // src/google-error.ts
 import { createJsonErrorResponseHandler } from "@ai-sdk/provider-utils";
-import { z } from "zod";
+import { z } from "zod/v4";
 var googleErrorDataSchema = z.object({
   error: z.object({
     code: z.number().nullable(),
@@ -37,7 +34,7 @@ var googleFailedResponseHandler = createJsonErrorResponseHandler({
 });
 
 // src/google-generative-ai-embedding-options.ts
-import { z as z2 } from "zod";
+import { z as z2 } from "zod/v4";
 var googleGenerativeAIEmbeddingProviderOptions = z2.object({
   /**
    * Optional. Optional reduced dimension for the output embedding.
@@ -103,6 +100,35 @@ var GoogleGenerativeAIEmbeddingModel = class {
       await resolve(this.config.headers),
       headers
     );
+    if (values.length === 1) {
+      const {
+        responseHeaders: responseHeaders2,
+        value: response2,
+        rawValue: rawValue2
+      } = await postJsonToApi({
+        url: `${this.config.baseURL}/models/${this.modelId}:embedContent`,
+        headers: mergedHeaders,
+        body: {
+          model: `models/${this.modelId}`,
+          content: {
+            parts: [{ text: values[0] }]
+          },
+          outputDimensionality: googleOptions == null ? void 0 : googleOptions.outputDimensionality,
+          taskType: googleOptions == null ? void 0 : googleOptions.taskType
+        },
+        failedResponseHandler: googleFailedResponseHandler,
+        successfulResponseHandler: createJsonResponseHandler(
+          googleGenerativeAISingleEmbeddingResponseSchema
+        ),
+        abortSignal,
+        fetch: this.config.fetch
+      });
+      return {
+        embeddings: [response2.embedding.values],
+        usage: void 0,
+        response: { headers: responseHeaders2, body: rawValue2 }
+      };
+    }
     const {
       responseHeaders,
       value: response,
@@ -135,17 +161,21 @@ var GoogleGenerativeAIEmbeddingModel = class {
 var googleGenerativeAITextEmbeddingResponseSchema = z3.object({
   embeddings: z3.array(z3.object({ values: z3.array(z3.number()) }))
 });
+var googleGenerativeAISingleEmbeddingResponseSchema = z3.object({
+  embedding: z3.object({ values: z3.array(z3.number()) })
+});
 
 // src/google-generative-ai-language-model.ts
 import {
   combineHeaders as combineHeaders2,
   createEventSourceResponseHandler,
   createJsonResponseHandler as createJsonResponseHandler2,
+  generateId,
   parseProviderOptions as parseProviderOptions2,
   postJsonToApi as postJsonToApi2,
   resolve as resolve2
 } from "@ai-sdk/provider-utils";
-import { z as z5 } from "zod";
+import { z as z7 } from "zod/v4";
 
 // src/convert-json-schema-to-openapi-schema.ts
 function convertJSONSchemaToOpenAPISchema(jsonSchema) {
@@ -241,7 +271,7 @@ function convertJSONSchemaToOpenAPISchema(jsonSchema) {
   return result;
 }
 function isEmptyObjectSchema(jsonSchema) {
-  return jsonSchema != null && typeof jsonSchema === "object" && jsonSchema.type === "object" && (jsonSchema.properties == null || Object.keys(jsonSchema.properties).length === 0);
+  return jsonSchema != null && typeof jsonSchema === "object" && jsonSchema.type === "object" && (jsonSchema.properties == null || Object.keys(jsonSchema.properties).length === 0) && !jsonSchema.additionalProperties;
 }
 
 // src/convert-to-google-generative-ai-messages.ts
@@ -249,10 +279,12 @@ import {
   UnsupportedFunctionalityError
 } from "@ai-sdk/provider";
 import { convertToBase64 } from "@ai-sdk/provider-utils";
-function convertToGoogleGenerativeAIMessages(prompt) {
+function convertToGoogleGenerativeAIMessages(prompt, options) {
+  var _a;
   const systemInstructionParts = [];
   const contents = [];
   let systemMessagesAllowed = true;
+  const isGemmaModel = (_a = options == null ? void 0 : options.isGemmaModel) != null ? _a : false;
   for (const { role, content } of prompt) {
     switch (role) {
       case "system": {
@@ -353,8 +385,12 @@ function convertToGoogleGenerativeAIMessages(prompt) {
       }
     }
   }
+  if (isGemmaModel && systemInstructionParts.length > 0 && contents.length > 0 && contents[0].role === "user") {
+    const systemText = systemInstructionParts.map((part) => part.text).join("\n\n");
+    contents[0].parts.unshift({ text: systemText + "\n\n" });
+  }
   return {
-    systemInstruction: systemInstructionParts.length > 0 ? { parts: systemInstructionParts } : void 0,
+    systemInstruction: systemInstructionParts.length > 0 && !isGemmaModel ? { parts: systemInstructionParts } : void 0,
     contents
   };
 }
@@ -365,18 +401,7 @@ function getModelPath(modelId) {
 }
 
 // src/google-generative-ai-options.ts
-import { z as z4 } from "zod";
-var dynamicRetrievalConfig = z4.object({
-  /**
-   * The mode of the predictor to be used in dynamic retrieval.
-   */
-  mode: z4.enum(["MODE_UNSPECIFIED", "MODE_DYNAMIC"]).optional(),
-  /**
-   * The threshold to be used in dynamic retrieval. If not set, a system default
-   * value is used.
-   */
-  dynamicThreshold: z4.number().optional()
-});
+import { z as z4 } from "zod/v4";
 var googleGenerativeAIProviderOptions = z4.object({
   responseModalities: z4.array(z4.enum(["TEXT", "IMAGE"])).optional(),
   thinkingConfig: z4.object({
@@ -434,21 +459,7 @@ var googleGenerativeAIProviderOptions = z4.object({
    *
    * https://cloud.google.com/vertex-ai/generative-ai/docs/multimodal/audio-understanding
    */
-  audioTimestamp: z4.boolean().optional(),
-  /**
-  Optional. When enabled, the model will use Google search to ground the response.
-  
-  @see https://cloud.google.com/vertex-ai/generative-ai/docs/grounding/overview
-   */
-  useSearchGrounding: z4.boolean().optional(),
-  /**
-  Optional. Specifies the dynamic retrieval configuration.
-  
-  @note Dynamic retrieval is only compatible with Gemini 1.5 Flash.
-  
-  @see https://cloud.google.com/vertex-ai/generative-ai/docs/multimodal/ground-with-google-search#dynamic-retrieval
-   */
-  dynamicRetrievalConfig: dynamicRetrievalConfig.optional()
+  audioTimestamp: z4.boolean().optional()
 });
 
 // src/google-prepare-tools.ts
@@ -458,8 +469,6 @@ import {
 function prepareTools({
   tools,
   toolChoice,
-  useSearchGrounding,
-  dynamicRetrievalConfig: dynamicRetrievalConfig2,
   modelId
 }) {
   var _a;
@@ -467,28 +476,76 @@ function prepareTools({
   const toolWarnings = [];
   const isGemini2 = modelId.includes("gemini-2");
   const supportsDynamicRetrieval = modelId.includes("gemini-1.5-flash") && !modelId.includes("-8b");
-  if (useSearchGrounding) {
+  if (tools == null) {
+    return { tools: void 0, toolConfig: void 0, toolWarnings };
+  }
+  const hasFunctionTools = tools.some((tool) => tool.type === "function");
+  const hasProviderDefinedTools = tools.some(
+    (tool) => tool.type === "provider-defined"
+  );
+  if (hasFunctionTools && hasProviderDefinedTools) {
+    toolWarnings.push({
+      type: "unsupported-tool",
+      tool: tools.find((tool) => tool.type === "function"),
+      details: "Cannot mix function tools with provider-defined tools in the same request. Please use either function tools or provider-defined tools, but not both."
+    });
+  }
+  if (hasProviderDefinedTools) {
+    const googleTools2 = {};
+    const providerDefinedTools = tools.filter(
+      (tool) => tool.type === "provider-defined"
+    );
+    providerDefinedTools.forEach((tool) => {
+      switch (tool.id) {
+        case "google.google_search":
+          if (isGemini2) {
+            googleTools2.googleSearch = {};
+          } else if (supportsDynamicRetrieval) {
+            googleTools2.googleSearchRetrieval = {
+              dynamicRetrievalConfig: {
+                mode: tool.args.mode,
+                dynamicThreshold: tool.args.dynamicThreshold
+              }
+            };
+          } else {
+            googleTools2.googleSearchRetrieval = {};
+          }
+          break;
+        case "google.url_context":
+          if (isGemini2) {
+            googleTools2.urlContext = {};
+          } else {
+            toolWarnings.push({
+              type: "unsupported-tool",
+              tool,
+              details: "The URL context tool is not supported with other Gemini models than Gemini 2."
+            });
+          }
+          break;
+        default:
+          toolWarnings.push({ type: "unsupported-tool", tool });
+          break;
+      }
+    });
     return {
-      tools: isGemini2 ? { googleSearch: {} } : {
-        googleSearchRetrieval: !supportsDynamicRetrieval || !dynamicRetrievalConfig2 ? {} : { dynamicRetrievalConfig: dynamicRetrievalConfig2 }
-      },
+      tools: Object.keys(googleTools2).length > 0 ? googleTools2 : void 0,
       toolConfig: void 0,
       toolWarnings
     };
   }
-  if (tools == null) {
-    return { tools: void 0, toolConfig: void 0, toolWarnings };
-  }
   const functionDeclarations = [];
   for (const tool of tools) {
-    if (tool.type === "provider-defined") {
-      toolWarnings.push({ type: "unsupported-tool", tool });
-    } else {
-      functionDeclarations.push({
-        name: tool.name,
-        description: (_a = tool.description) != null ? _a : "",
-        parameters: convertJSONSchemaToOpenAPISchema(tool.inputSchema)
-      });
+    switch (tool.type) {
+      case "function":
+        functionDeclarations.push({
+          name: tool.name,
+          description: (_a = tool.description) != null ? _a : "",
+          parameters: convertJSONSchemaToOpenAPISchema(tool.inputSchema)
+        });
+        break;
+      default:
+        toolWarnings.push({ type: "unsupported-tool", tool });
+        break;
     }
   }
   if (toolChoice == null) {
@@ -565,12 +622,72 @@ function mapGoogleGenerativeAIFinishReason({
   }
 }
 
+// src/tool/google-search.ts
+import { createProviderDefinedToolFactory } from "@ai-sdk/provider-utils";
+import { z as z5 } from "zod/v4";
+var groundingChunkSchema = z5.object({
+  web: z5.object({ uri: z5.string(), title: z5.string() }).nullish(),
+  retrievedContext: z5.object({ uri: z5.string(), title: z5.string() }).nullish()
+});
+var groundingMetadataSchema = z5.object({
+  webSearchQueries: z5.array(z5.string()).nullish(),
+  retrievalQueries: z5.array(z5.string()).nullish(),
+  searchEntryPoint: z5.object({ renderedContent: z5.string() }).nullish(),
+  groundingChunks: z5.array(groundingChunkSchema).nullish(),
+  groundingSupports: z5.array(
+    z5.object({
+      segment: z5.object({
+        startIndex: z5.number().nullish(),
+        endIndex: z5.number().nullish(),
+        text: z5.string().nullish()
+      }),
+      segment_text: z5.string().nullish(),
+      groundingChunkIndices: z5.array(z5.number()).nullish(),
+      supportChunkIndices: z5.array(z5.number()).nullish(),
+      confidenceScores: z5.array(z5.number()).nullish(),
+      confidenceScore: z5.array(z5.number()).nullish()
+    })
+  ).nullish(),
+  retrievalMetadata: z5.union([
+    z5.object({
+      webDynamicRetrievalScore: z5.number()
+    }),
+    z5.object({})
+  ]).nullish()
+});
+var googleSearch = createProviderDefinedToolFactory({
+  id: "google.google_search",
+  name: "google_search",
+  inputSchema: z5.object({
+    mode: z5.enum(["MODE_DYNAMIC", "MODE_UNSPECIFIED"]).default("MODE_UNSPECIFIED"),
+    dynamicThreshold: z5.number().default(1)
+  })
+});
+
+// src/tool/url-context.ts
+import { createProviderDefinedToolFactory as createProviderDefinedToolFactory2 } from "@ai-sdk/provider-utils";
+import { z as z6 } from "zod/v4";
+var urlMetadataSchema = z6.object({
+  retrievedUrl: z6.string(),
+  urlRetrievalStatus: z6.string()
+});
+var urlContextMetadataSchema = z6.object({
+  urlMetadata: z6.array(urlMetadataSchema)
+});
+var urlContext = createProviderDefinedToolFactory2({
+  id: "google.url_context",
+  name: "url_context",
+  inputSchema: z6.object({})
+});
+
 // src/google-generative-ai-language-model.ts
 var GoogleGenerativeAILanguageModel = class {
   constructor(modelId, config) {
     this.specificationVersion = "v2";
+    var _a;
     this.modelId = modelId;
     this.config = config;
+    this.generateId = (_a = config.generateId) != null ? _a : generateId;
   }
   get provider() {
     return this.config.provider;
@@ -594,7 +711,7 @@ var GoogleGenerativeAILanguageModel = class {
     toolChoice,
     providerOptions
   }) {
-    var _a, _b, _c;
+    var _a, _b;
     const warnings = [];
     const googleOptions = await parseProviderOptions2({
       provider: "google",
@@ -607,16 +724,18 @@ var GoogleGenerativeAILanguageModel = class {
         message: `The 'includeThoughts' option is only supported with the Google Vertex provider and might not be supported or could behave unexpectedly with the current Google provider (${this.config.provider}).`
       });
     }
-    const { contents, systemInstruction } = convertToGoogleGenerativeAIMessages(prompt);
+    const isGemmaModel = this.modelId.toLowerCase().startsWith("gemma-");
+    const { contents, systemInstruction } = convertToGoogleGenerativeAIMessages(
+      prompt,
+      { isGemmaModel }
+    );
     const {
-      tools: googleTools,
+      tools: googleTools2,
       toolConfig: googleToolConfig,
       toolWarnings
     } = prepareTools({
       tools,
       toolChoice,
-      useSearchGrounding: (_b = googleOptions == null ? void 0 : googleOptions.useSearchGrounding) != null ? _b : false,
-      dynamicRetrievalConfig: googleOptions == null ? void 0 : googleOptions.dynamicRetrievalConfig,
       modelId: this.modelId
     });
     return {
@@ -636,7 +755,7 @@ var GoogleGenerativeAILanguageModel = class {
           responseSchema: (responseFormat == null ? void 0 : responseFormat.type) === "json" && responseFormat.schema != null && // Google GenAI does not support all OpenAPI Schema features,
           // so this is needed as an escape hatch:
           // TODO convert into provider option
-          ((_c = googleOptions == null ? void 0 : googleOptions.structuredOutputs) != null ? _c : true) ? convertJSONSchemaToOpenAPISchema(responseFormat.schema) : void 0,
+          ((_b = googleOptions == null ? void 0 : googleOptions.structuredOutputs) != null ? _b : true) ? convertJSONSchemaToOpenAPISchema(responseFormat.schema) : void 0,
           ...(googleOptions == null ? void 0 : googleOptions.audioTimestamp) && {
             audioTimestamp: googleOptions.audioTimestamp
           },
@@ -645,9 +764,9 @@ var GoogleGenerativeAILanguageModel = class {
           thinkingConfig: googleOptions == null ? void 0 : googleOptions.thinkingConfig
         },
         contents,
-        systemInstruction,
+        systemInstruction: isGemmaModel ? void 0 : systemInstruction,
         safetySettings: googleOptions == null ? void 0 : googleOptions.safetySettings,
-        tools: googleTools,
+        tools: googleTools2,
         toolConfig: googleToolConfig,
         cachedContent: googleOptions == null ? void 0 : googleOptions.cachedContent
       },
@@ -655,7 +774,7 @@ var GoogleGenerativeAILanguageModel = class {
     };
   }
   async doGenerate(options) {
-    var _a, _b, _c, _d, _e, _f, _g, _h, _i;
+    var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j;
     const { args, warnings } = await this.getArgs(options);
     const body = JSON.stringify(args);
     const mergedHeaders = combineHeaders2(
@@ -727,7 +846,9 @@ var GoogleGenerativeAILanguageModel = class {
       providerMetadata: {
         google: {
           groundingMetadata: (_h = candidate.groundingMetadata) != null ? _h : null,
-          safetyRatings: (_i = candidate.safetyRatings) != null ? _i : null
+          urlContextMetadata: (_i = candidate.urlContextMetadata) != null ? _i : null,
+          safetyRatings: (_j = candidate.safetyRatings) != null ? _j : null,
+          usageMetadata: usageMetadata != null ? usageMetadata : null
         }
       },
       request: { body },
@@ -763,11 +884,12 @@ var GoogleGenerativeAILanguageModel = class {
       totalTokens: void 0
     };
     let providerMetadata = void 0;
-    const generateId2 = this.config.generateId;
+    const generateId3 = this.config.generateId;
     let hasToolCalls = false;
     let currentTextBlockId = null;
     let currentReasoningBlockId = null;
     let blockCounter = 0;
+    const emittedSourceUrls = /* @__PURE__ */ new Set();
     return {
       stream: response.pipeThrough(
         new TransformStream({
@@ -797,6 +919,18 @@ var GoogleGenerativeAILanguageModel = class {
               return;
             }
             const content = candidate.content;
+            const sources = extractSources({
+              groundingMetadata: candidate.groundingMetadata,
+              generateId: generateId3
+            });
+            if (sources != null) {
+              for (const source of sources) {
+                if (source.sourceType === "url" && !emittedSourceUrls.has(source.url)) {
+                  emittedSourceUrls.add(source.url);
+                  controller.enqueue(source);
+                }
+              }
+            }
             if (content != null) {
               const parts = (_g = content.parts) != null ? _g : [];
               for (const part of parts) {
@@ -856,7 +990,7 @@ var GoogleGenerativeAILanguageModel = class {
               }
               const toolCallDeltas = getToolCallsFromParts({
                 parts: content.parts,
-                generateId: generateId2
+                generateId: generateId3
               });
               if (toolCallDeltas != null) {
                 for (const toolCall of toolCallDeltas) {
@@ -889,19 +1023,16 @@ var GoogleGenerativeAILanguageModel = class {
                 finishReason: candidate.finishReason,
                 hasToolCalls
               });
-              const sources = (_h = extractSources({
-                groundingMetadata: candidate.groundingMetadata,
-                generateId: generateId2
-              })) != null ? _h : [];
-              for (const source of sources) {
-                controller.enqueue(source);
-              }
               providerMetadata = {
                 google: {
-                  groundingMetadata: (_i = candidate.groundingMetadata) != null ? _i : null,
+                  groundingMetadata: (_h = candidate.groundingMetadata) != null ? _h : null,
+                  urlContextMetadata: (_i = candidate.urlContextMetadata) != null ? _i : null,
                   safetyRatings: (_j = candidate.safetyRatings) != null ? _j : null
                 }
               };
+              if (usageMetadata != null) {
+                providerMetadata.google.usageMetadata = usageMetadata;
+              }
             }
           },
           flush(controller) {
@@ -933,14 +1064,14 @@ var GoogleGenerativeAILanguageModel = class {
 };
 function getToolCallsFromParts({
   parts,
-  generateId: generateId2
+  generateId: generateId3
 }) {
   const functionCallParts = parts == null ? void 0 : parts.filter(
     (part) => "functionCall" in part
   );
   return functionCallParts == null || functionCallParts.length === 0 ? void 0 : functionCallParts.map((part) => ({
     type: "tool-call",
-    toolCallId: generateId2(),
+    toolCallId: generateId3(),
     toolName: part.functionCall.name,
     args: JSON.stringify(part.functionCall.args)
   }));
@@ -952,7 +1083,7 @@ function getInlineDataParts(parts) {
 }
 function extractSources({
   groundingMetadata,
-  generateId: generateId2
+  generateId: generateId3
 }) {
   var _a;
   return (_a = groundingMetadata == null ? void 0 : groundingMetadata.groundingChunks) == null ? void 0 : _a.filter(
@@ -960,102 +1091,196 @@ function extractSources({
   ).map((chunk) => ({
     type: "source",
     sourceType: "url",
-    id: generateId2(),
+    id: generateId3(),
     url: chunk.web.uri,
     title: chunk.web.title
   }));
 }
-var contentSchema = z5.object({
-  parts: z5.array(
-    z5.union([
+var contentSchema = z7.object({
+  parts: z7.array(
+    z7.union([
       // note: order matters since text can be fully empty
-      z5.object({
-        functionCall: z5.object({
-          name: z5.string(),
-          args: z5.unknown()
+      z7.object({
+        functionCall: z7.object({
+          name: z7.string(),
+          args: z7.unknown()
         })
       }),
-      z5.object({
-        inlineData: z5.object({
-          mimeType: z5.string(),
-          data: z5.string()
+      z7.object({
+        inlineData: z7.object({
+          mimeType: z7.string(),
+          data: z7.string()
         })
       }),
-      z5.object({
-        text: z5.string().nullish(),
-        thought: z5.boolean().nullish()
+      z7.object({
+        text: z7.string().nullish(),
+        thought: z7.boolean().nullish()
       })
     ])
   ).nullish()
 });
-var groundingChunkSchema = z5.object({
-  web: z5.object({ uri: z5.string(), title: z5.string() }).nullish(),
-  retrievedContext: z5.object({ uri: z5.string(), title: z5.string() }).nullish()
+var safetyRatingSchema = z7.object({
+  category: z7.string().nullish(),
+  probability: z7.string().nullish(),
+  probabilityScore: z7.number().nullish(),
+  severity: z7.string().nullish(),
+  severityScore: z7.number().nullish(),
+  blocked: z7.boolean().nullish()
 });
-var groundingMetadataSchema = z5.object({
-  webSearchQueries: z5.array(z5.string()).nullish(),
-  retrievalQueries: z5.array(z5.string()).nullish(),
-  searchEntryPoint: z5.object({ renderedContent: z5.string() }).nullish(),
-  groundingChunks: z5.array(groundingChunkSchema).nullish(),
-  groundingSupports: z5.array(
-    z5.object({
-      segment: z5.object({
-        startIndex: z5.number().nullish(),
-        endIndex: z5.number().nullish(),
-        text: z5.string().nullish()
-      }),
-      segment_text: z5.string().nullish(),
-      groundingChunkIndices: z5.array(z5.number()).nullish(),
-      supportChunkIndices: z5.array(z5.number()).nullish(),
-      confidenceScores: z5.array(z5.number()).nullish(),
-      confidenceScore: z5.array(z5.number()).nullish()
-    })
-  ).nullish(),
-  retrievalMetadata: z5.union([
-    z5.object({
-      webDynamicRetrievalScore: z5.number()
-    }),
-    z5.object({})
-  ]).nullish()
-});
-var safetyRatingSchema = z5.object({
-  category: z5.string().nullish(),
-  probability: z5.string().nullish(),
-  probabilityScore: z5.number().nullish(),
-  severity: z5.string().nullish(),
-  severityScore: z5.number().nullish(),
-  blocked: z5.boolean().nullish()
-});
-var usageSchema = z5.object({
-  cachedContentTokenCount: z5.number().nullish(),
-  thoughtsTokenCount: z5.number().nullish(),
-  promptTokenCount: z5.number().nullish(),
-  candidatesTokenCount: z5.number().nullish(),
-  totalTokenCount: z5.number().nullish()
+var usageSchema = z7.object({
+  cachedContentTokenCount: z7.number().nullish(),
+  thoughtsTokenCount: z7.number().nullish(),
+  promptTokenCount: z7.number().nullish(),
+  candidatesTokenCount: z7.number().nullish(),
+  totalTokenCount: z7.number().nullish()
 });
-var responseSchema = z5.object({
-  candidates: z5.array(
-    z5.object({
-      content: contentSchema.nullish().or(z5.object({}).strict()),
-      finishReason: z5.string().nullish(),
-      safetyRatings: z5.array(safetyRatingSchema).nullish(),
-      groundingMetadata: groundingMetadataSchema.nullish()
+var responseSchema = z7.object({
+  candidates: z7.array(
+    z7.object({
+      content: contentSchema.nullish().or(z7.object({}).strict()),
+      finishReason: z7.string().nullish(),
+      safetyRatings: z7.array(safetyRatingSchema).nullish(),
+      groundingMetadata: groundingMetadataSchema.nullish(),
+      urlContextMetadata: urlContextMetadataSchema.nullish()
     })
   ),
   usageMetadata: usageSchema.nullish()
 });
-var chunkSchema = z5.object({
-  candidates: z5.array(
-    z5.object({
+var chunkSchema = z7.object({
+  candidates: z7.array(
+    z7.object({
       content: contentSchema.nullish(),
-      finishReason: z5.string().nullish(),
-      safetyRatings: z5.array(safetyRatingSchema).nullish(),
-      groundingMetadata: groundingMetadataSchema.nullish()
+      finishReason: z7.string().nullish(),
+      safetyRatings: z7.array(safetyRatingSchema).nullish(),
+      groundingMetadata: groundingMetadataSchema.nullish(),
+      urlContextMetadata: urlContextMetadataSchema.nullish()
     })
   ).nullish(),
   usageMetadata: usageSchema.nullish()
 });
 
+// src/google-tools.ts
+var googleTools = {
+  /**
+   * Creates a Google search tool that gives Google direct access to real-time web content.
+   * Must have name "google_search".
+   */
+  googleSearch,
+  /**
+   * Creates a URL context tool that gives Google direct access to real-time web content.
+   * Must have name "url_context".
+   */
+  urlContext
+};
+
+// src/google-generative-ai-image-model.ts
+import {
+  combineHeaders as combineHeaders3,
+  createJsonResponseHandler as createJsonResponseHandler3,
+  parseProviderOptions as parseProviderOptions3,
+  postJsonToApi as postJsonToApi3,
+  resolve as resolve3
+} from "@ai-sdk/provider-utils";
+import { z as z8 } from "zod/v4";
+var GoogleGenerativeAIImageModel = class {
+  constructor(modelId, settings, config) {
+    this.modelId = modelId;
+    this.settings = settings;
+    this.config = config;
+    this.specificationVersion = "v2";
+  }
+  get maxImagesPerCall() {
+    var _a;
+    return (_a = this.settings.maxImagesPerCall) != null ? _a : 4;
+  }
+  get provider() {
+    return this.config.provider;
+  }
+  async doGenerate(options) {
+    var _a, _b, _c;
+    const {
+      prompt,
+      n = 1,
+      size = "1024x1024",
+      aspectRatio = "1:1",
+      seed,
+      providerOptions,
+      headers,
+      abortSignal
+    } = options;
+    const warnings = [];
+    if (size != null) {
+      warnings.push({
+        type: "unsupported-setting",
+        setting: "size",
+        details: "This model does not support the `size` option. Use `aspectRatio` instead."
+      });
+    }
+    if (seed != null) {
+      warnings.push({
+        type: "unsupported-setting",
+        setting: "seed",
+        details: "This model does not support the `seed` option through this provider."
+      });
+    }
+    const googleOptions = await parseProviderOptions3({
+      provider: "google",
+      providerOptions,
+      schema: googleImageProviderOptionsSchema
+    });
+    const currentDate = (_c = (_b = (_a = this.config._internal) == null ? void 0 : _a.currentDate) == null ? void 0 : _b.call(_a)) != null ? _c : /* @__PURE__ */ new Date();
+    const parameters = {
+      sampleCount: n
+    };
+    if (aspectRatio != null) {
+      parameters.aspectRatio = aspectRatio;
+    }
+    if (googleOptions) {
+      Object.assign(parameters, googleOptions);
+    }
+    const body = {
+      instances: [{ prompt }],
+      parameters
+    };
+    const { responseHeaders, value: response } = await postJsonToApi3({
+      url: `${this.config.baseURL}/models/${this.modelId}:predict`,
+      headers: combineHeaders3(await resolve3(this.config.headers), headers),
+      body,
+      failedResponseHandler: googleFailedResponseHandler,
+      successfulResponseHandler: createJsonResponseHandler3(
+        googleImageResponseSchema
+      ),
+      abortSignal,
+      fetch: this.config.fetch
+    });
+    return {
+      images: response.predictions.map(
+        (p) => p.bytesBase64Encoded
+      ),
+      warnings: warnings != null ? warnings : [],
+      providerMetadata: {
+        google: {
+          images: response.predictions.map((prediction) => ({
+            // Add any prediction-specific metadata here
+          }))
+        }
+      },
+      response: {
+        timestamp: currentDate,
+        modelId: this.modelId,
+        headers: responseHeaders
+      }
+    };
+  }
+};
+var googleImageResponseSchema = z8.object({
+  predictions: z8.array(z8.object({ bytesBase64Encoded: z8.string() })).default([])
+});
+var googleImageProviderOptionsSchema = z8.object({
+  personGeneration: z8.enum(["dont_allow", "allow_adult", "allow_all"]).nullish(),
+  aspectRatio: z8.enum(["1:1", "3:4", "4:3", "9:16", "16:9"]).nullish()
+});
+
 // src/google-provider.ts
 function createGoogleGenerativeAI(options = {}) {
   var _a;
@@ -1074,7 +1299,7 @@ function createGoogleGenerativeAI(options = {}) {
       provider: "google.generative-ai",
       baseURL,
       headers: getHeaders,
-      generateId: (_a2 = options.generateId) != null ? _a2 : generateId,
+      generateId: (_a2 = options.generateId) != null ? _a2 : generateId2,
       supportedUrls: () => ({
         "*": [
           // Only allow requests to the Google Generative Language "files" endpoint
@@ -1091,6 +1316,12 @@ function createGoogleGenerativeAI(options = {}) {
     headers: getHeaders,
     fetch: options.fetch
   });
+  const createImageModel = (modelId, settings = {}) => new GoogleGenerativeAIImageModel(modelId, settings, {
+    provider: "google.generative-ai",
+    baseURL,
+    headers: getHeaders,
+    fetch: options.fetch
+  });
   const provider = function(modelId) {
     if (new.target) {
       throw new Error(
@@ -1105,9 +1336,9 @@ function createGoogleGenerativeAI(options = {}) {
   provider.embedding = createEmbeddingModel;
   provider.textEmbedding = createEmbeddingModel;
   provider.textEmbeddingModel = createEmbeddingModel;
-  provider.imageModel = (modelId) => {
-    throw new NoSuchModelError({ modelId, modelType: "imageModel" });
-  };
+  provider.image = createImageModel;
+  provider.imageModel = createImageModel;
+  provider.tools = googleTools;
   return provider;
 }
 var google = createGoogleGenerativeAI();