@ai-sdk/google 4.0.63 → 4.0.65

package/src/google-provider.ts

@@ -1,6 +1,6 @@
 import type {
   EmbeddingModelV4,
-  Experimental_BatchLanguageModelV4 as BatchLanguageModelV4,
+  Experimental_BatchV4 as BatchV4,
   Experimental_VideoModelV4,
   FilesV4,
   ImageModelV4,
@@ -23,7 +23,8 @@ import {
 import { VERSION } from './version';
 import { GoogleEmbeddingModel } from './google-embedding-model';
 import type { GoogleEmbeddingModelId } from './google-embedding-model-options';
-import { GoogleBatchLanguageModel } from './google-batch';
+import { GoogleBatch } from './google-batch';
+import { GoogleLanguageModel } from './google-language-model';
 import type { GoogleModelId } from './google-language-model-options';
 import { googleTools } from './google-tools';
 
@@ -54,11 +55,13 @@ const googleFilesUrlPattern =
   /^https:\/\/generativelanguage\.googleapis\.com\/v1beta\/files\/.*$/;
 
 export interface GoogleProvider extends ProviderV4 {
-  (modelId: GoogleModelId): BatchLanguageModelV4;
+  (modelId: GoogleModelId): LanguageModelV4;
 
-  languageModel(modelId: GoogleModelId): BatchLanguageModelV4;
+  languageModel(modelId: GoogleModelId): LanguageModelV4;
 
-  chat(modelId: GoogleModelId): BatchLanguageModelV4;
+  chat(modelId: GoogleModelId): LanguageModelV4;
+
+  experimental_batch(): BatchV4<{ text: GoogleModelId }>;
 
   /**
    * Creates a model for image generation.
@@ -71,7 +74,7 @@ export interface GoogleProvider extends ProviderV4 {
   /**
    * @deprecated Use `chat()` instead.
    */
-  generativeAI(modelId: GoogleModelId): BatchLanguageModelV4;
+  generativeAI(modelId: GoogleModelId): LanguageModelV4;
 
   /**
    * Creates a model for text embeddings.
@@ -258,35 +261,49 @@ export function createGoogle(
       `ai-sdk/google/${VERSION}`,
     );
 
+  const getSupportedUrls = (
+    modelId?: GoogleModelId,
+    includeExternalUrls = modelId == null || supportsExternalFileUrls(modelId),
+  ) => ({
+    '*': [
+      googleFilesUrlPattern,
+      new RegExp(`^${baseURL}/files/.*$`),
+      new RegExp(
+        `^https://(?:www\\.)?youtube\\.com/watch\\?v=[\\w-]+(?:&[\\w=&.-]*)?$`,
+      ),
+      new RegExp(`^https://youtu\\.be/[\\w-]+(?:\\?[\\w=&.-]*)?$`),
+    ],
+    ...(includeExternalUrls
+      ? Object.fromEntries(
+          supportedExternalUrlMediaTypes.map(mediaType => [
+            mediaType,
+            [externalHttpsUrlPattern],
+          ]),
+        )
+      : {}),
+  });
+
+  const languageModelConfig = {
+    provider: providerName,
+    baseURL,
+    headers: getHeaders,
+    generateId: options.generateId ?? generateId,
+    fetch: options.fetch,
+  };
+
   const createChatModel = (modelId: GoogleModelId) =>
-    new GoogleBatchLanguageModel(modelId, {
-      provider: providerName,
-      baseURL,
-      headers: getHeaders,
-      generateId: options.generateId ?? generateId,
-      supportedUrls: () => ({
-        '*': [
-          // Default Google Generative Language "files" endpoint
-          // e.g. https://generativelanguage.googleapis.com/v1beta/files/...
-          googleFilesUrlPattern,
-          // Configured Google Generative Language "files" endpoint
-          new RegExp(`^${baseURL}/files/.*$`),
-          // YouTube URLs (public or unlisted videos)
-          new RegExp(
-            `^https://(?:www\\.)?youtube\\.com/watch\\?v=[\\w-]+(?:&[\\w=&.-]*)?$`,
-          ),
-          new RegExp(`^https://youtu\\.be/[\\w-]+(?:\\?[\\w=&.-]*)?$`),
-        ],
-        ...(supportsExternalFileUrls(modelId)
-          ? Object.fromEntries(
-              supportedExternalUrlMediaTypes.map(mediaType => [
-                mediaType,
-                [externalHttpsUrlPattern],
-              ]),
-            )
-          : {}),
-      }),
-      fetch: options.fetch,
+    new GoogleLanguageModel(modelId, {
+      ...languageModelConfig,
+      supportedUrls: () => getSupportedUrls(modelId),
+    });
+
+  const createBatch = () =>
+    new GoogleBatch({
+      provider: `${providerName.replace(/\.generative-ai$/, '')}.batch`,
+      config: languageModelConfig,
+      // Batch prompt conversion happens before the model is available to the
+      // provider. Only advertise URL support shared by every batch model.
+      supportedUrls: getSupportedUrls(undefined, false),
     });
 
   const createEmbeddingModel = (modelId: GoogleEmbeddingModelId) =>
@@ -410,6 +427,7 @@ export function createGoogle(
   provider.languageModel = createChatModel;
   provider.chat = createChatModel;
   provider.generativeAI = createChatModel;
+  provider.experimental_batch = createBatch;
   provider.embedding = createEmbeddingModel;
   provider.embeddingModel = createEmbeddingModel;
   provider.textEmbedding = createEmbeddingModel;