@ai-sdk/google 2.0.0-canary.11 → 2.0.0-canary.13

package/dist/index.mjs

@@ -43,37 +43,54 @@ var googleGenerativeAIEmbeddingProviderOptions = z2.object({
    * Optional. Optional reduced dimension for the output embedding.
    * If set, excessive values in the output embedding are truncated from the end.
    */
-  outputDimensionality: z2.number().optional()
+  outputDimensionality: z2.number().optional(),
+  /**
+   * Optional. Specifies the task type for generating embeddings.
+   * Supported task types:
+   * - SEMANTIC_SIMILARITY: Optimized for text similarity.
+   * - CLASSIFICATION: Optimized for text classification.
+   * - CLUSTERING: Optimized for clustering texts based on similarity.
+   * - RETRIEVAL_DOCUMENT: Optimized for document retrieval.
+   * - RETRIEVAL_QUERY: Optimized for query-based retrieval.
+   * - QUESTION_ANSWERING: Optimized for answering questions.
+   * - FACT_VERIFICATION: Optimized for verifying factual information.
+   * - CODE_RETRIEVAL_QUERY: Optimized for retrieving code blocks based on natural language queries.
+   */
+  taskType: z2.enum([
+    "SEMANTIC_SIMILARITY",
+    "CLASSIFICATION",
+    "CLUSTERING",
+    "RETRIEVAL_DOCUMENT",
+    "RETRIEVAL_QUERY",
+    "QUESTION_ANSWERING",
+    "FACT_VERIFICATION",
+    "CODE_RETRIEVAL_QUERY"
+  ]).optional()
 });
 
 // src/google-generative-ai-embedding-model.ts
 var GoogleGenerativeAIEmbeddingModel = class {
   constructor(modelId, config) {
     this.specificationVersion = "v2";
+    this.maxEmbeddingsPerCall = 2048;
+    this.supportsParallelCalls = true;
     this.modelId = modelId;
     this.config = config;
   }
   get provider() {
     return this.config.provider;
   }
-  get maxEmbeddingsPerCall() {
-    return 2048;
-  }
-  get supportsParallelCalls() {
-    return true;
-  }
   async doEmbed({
     values,
     headers,
     abortSignal,
     providerOptions
   }) {
-    var _a;
-    const googleOptions = (_a = parseProviderOptions({
+    const googleOptions = await parseProviderOptions({
       provider: "google",
       providerOptions,
       schema: googleGenerativeAIEmbeddingProviderOptions
-    })) != null ? _a : {};
+    });
     if (values.length > this.maxEmbeddingsPerCall) {
       throw new TooManyEmbeddingValuesForCallError({
         provider: this.provider,
@@ -97,7 +114,8 @@ var GoogleGenerativeAIEmbeddingModel = class {
         requests: values.map((value) => ({
           model: `models/${this.modelId}`,
           content: { role: "user", parts: [{ text: value }] },
-          outputDimensionality: googleOptions.outputDimensionality
+          outputDimensionality: googleOptions == null ? void 0 : googleOptions.outputDimensionality,
+          taskType: googleOptions == null ? void 0 : googleOptions.taskType
         }))
       },
       failedResponseHandler: googleFailedResponseHandler,
@@ -127,7 +145,7 @@ import {
   postJsonToApi as postJsonToApi2,
   resolve as resolve2
 } from "@ai-sdk/provider-utils";
-import { z as z4 } from "zod";
+import { z as z5 } from "zod";
 
 // src/convert-json-schema-to-openapi-schema.ts
 function convertJSONSchemaToOpenAPISchema(jsonSchema) {
@@ -348,6 +366,92 @@ function getModelPath(modelId) {
   return modelId.includes("/") ? modelId : `models/${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()
+});
+var googleGenerativeAIProviderOptions = z4.object({
+  responseModalities: z4.array(z4.enum(["TEXT", "IMAGE"])).optional(),
+  thinkingConfig: z4.object({
+    thinkingBudget: z4.number().optional()
+  }).optional(),
+  /**
+  Optional.
+  The name of the cached content used as context to serve the prediction.
+  Format: cachedContents/{cachedContent}
+     */
+  cachedContent: z4.string().optional(),
+  /**
+   * Optional. Enable structured output. Default is true.
+   *
+   * This is useful when the JSON Schema contains elements that are
+   * not supported by the OpenAPI schema version that
+   * Google Generative AI uses. You can use this to disable
+   * structured outputs if you need to.
+   */
+  structuredOutputs: z4.boolean().optional(),
+  /**
+  Optional. A list of unique safety settings for blocking unsafe content.
+   */
+  safetySettings: z4.array(
+    z4.object({
+      category: z4.enum([
+        "HARM_CATEGORY_UNSPECIFIED",
+        "HARM_CATEGORY_HATE_SPEECH",
+        "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "HARM_CATEGORY_HARASSMENT",
+        "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "HARM_CATEGORY_CIVIC_INTEGRITY"
+      ]),
+      threshold: z4.enum([
+        "HARM_BLOCK_THRESHOLD_UNSPECIFIED",
+        "BLOCK_LOW_AND_ABOVE",
+        "BLOCK_MEDIUM_AND_ABOVE",
+        "BLOCK_ONLY_HIGH",
+        "BLOCK_NONE",
+        "OFF"
+      ])
+    })
+  ).optional(),
+  threshold: z4.enum([
+    "HARM_BLOCK_THRESHOLD_UNSPECIFIED",
+    "BLOCK_LOW_AND_ABOVE",
+    "BLOCK_MEDIUM_AND_ABOVE",
+    "BLOCK_ONLY_HIGH",
+    "BLOCK_NONE",
+    "OFF"
+  ]).optional(),
+  /**
+   * Optional. Enables timestamp understanding for audio-only files.
+   *
+   * 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()
+});
+
 // src/google-prepare-tools.ts
 import {
   UnsupportedFunctionalityError as UnsupportedFunctionalityError2
@@ -356,7 +460,7 @@ function prepareTools({
   tools,
   toolChoice,
   useSearchGrounding,
-  dynamicRetrievalConfig,
+  dynamicRetrievalConfig: dynamicRetrievalConfig2,
   modelId
 }) {
   var _a;
@@ -367,7 +471,7 @@ function prepareTools({
   if (useSearchGrounding) {
     return {
       tools: isGemini2 ? { googleSearch: {} } : {
-        googleSearchRetrieval: !supportsDynamicRetrieval || !dynamicRetrievalConfig ? {} : { dynamicRetrievalConfig }
+        googleSearchRetrieval: !supportsDynamicRetrieval || !dynamicRetrievalConfig2 ? {} : { dynamicRetrievalConfig: dynamicRetrievalConfig2 }
       },
       toolConfig: void 0,
       toolWarnings
@@ -464,10 +568,9 @@ function mapGoogleGenerativeAIFinishReason({
 
 // src/google-generative-ai-language-model.ts
 var GoogleGenerativeAILanguageModel = class {
-  constructor(modelId, settings, config) {
+  constructor(modelId, config) {
     this.specificationVersion = "v2";
     this.modelId = modelId;
-    this.settings = settings;
     this.config = config;
   }
   get provider() {
@@ -494,10 +597,10 @@ var GoogleGenerativeAILanguageModel = class {
   }) {
     var _a, _b;
     const warnings = [];
-    const googleOptions = parseProviderOptions2({
+    const googleOptions = await parseProviderOptions2({
       provider: "google",
       providerOptions,
-      schema: googleGenerativeAIProviderOptionsSchema
+      schema: googleGenerativeAIProviderOptions
     });
     const { contents, systemInstruction } = convertToGoogleGenerativeAIMessages(prompt);
     const {
@@ -507,8 +610,8 @@ var GoogleGenerativeAILanguageModel = class {
     } = prepareTools({
       tools,
       toolChoice,
-      useSearchGrounding: (_a = this.settings.useSearchGrounding) != null ? _a : false,
-      dynamicRetrievalConfig: this.settings.dynamicRetrievalConfig,
+      useSearchGrounding: (_a = googleOptions == null ? void 0 : googleOptions.useSearchGrounding) != null ? _a : false,
+      dynamicRetrievalConfig: googleOptions == null ? void 0 : googleOptions.dynamicRetrievalConfig,
       modelId: this.modelId
     });
     return {
@@ -528,9 +631,9 @@ 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
-          ((_b = this.settings.structuredOutputs) != null ? _b : true) ? convertJSONSchemaToOpenAPISchema(responseFormat.schema) : void 0,
-          ...this.settings.audioTimestamp && {
-            audioTimestamp: this.settings.audioTimestamp
+          ((_b = googleOptions == null ? void 0 : googleOptions.structuredOutputs) != null ? _b : true) ? convertJSONSchemaToOpenAPISchema(responseFormat.schema) : void 0,
+          ...(googleOptions == null ? void 0 : googleOptions.audioTimestamp) && {
+            audioTimestamp: googleOptions.audioTimestamp
           },
           // provider options:
           responseModalities: googleOptions == null ? void 0 : googleOptions.responseModalities,
@@ -538,10 +641,10 @@ var GoogleGenerativeAILanguageModel = class {
         },
         contents,
         systemInstruction,
-        safetySettings: this.settings.safetySettings,
+        safetySettings: googleOptions == null ? void 0 : googleOptions.safetySettings,
         tools: googleTools,
         toolConfig: googleToolConfig,
-        cachedContent: this.settings.cachedContent
+        cachedContent: googleOptions == null ? void 0 : googleOptions.cachedContent
       },
       warnings: [...warnings, ...toolWarnings]
     };
@@ -789,100 +892,94 @@ function extractSources({
     title: chunk.web.title
   }));
 }
-var contentSchema = z4.object({
-  role: z4.string(),
-  parts: z4.array(
-    z4.union([
-      z4.object({
-        text: z4.string()
+var contentSchema = z5.object({
+  role: z5.string(),
+  parts: z5.array(
+    z5.union([
+      z5.object({
+        text: z5.string()
       }),
-      z4.object({
-        functionCall: z4.object({
-          name: z4.string(),
-          args: z4.unknown()
+      z5.object({
+        functionCall: z5.object({
+          name: z5.string(),
+          args: z5.unknown()
         })
       }),
-      z4.object({
-        inlineData: z4.object({
-          mimeType: z4.string(),
-          data: z4.string()
+      z5.object({
+        inlineData: z5.object({
+          mimeType: z5.string(),
+          data: z5.string()
         })
       })
     ])
   ).nullish()
 });
-var groundingChunkSchema = z4.object({
-  web: z4.object({ uri: z4.string(), title: z4.string() }).nullish(),
-  retrievedContext: z4.object({ uri: z4.string(), title: z4.string() }).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 groundingMetadataSchema = z4.object({
-  webSearchQueries: z4.array(z4.string()).nullish(),
-  retrievalQueries: z4.array(z4.string()).nullish(),
-  searchEntryPoint: z4.object({ renderedContent: z4.string() }).nullish(),
-  groundingChunks: z4.array(groundingChunkSchema).nullish(),
-  groundingSupports: z4.array(
-    z4.object({
-      segment: z4.object({
-        startIndex: z4.number().nullish(),
-        endIndex: z4.number().nullish(),
-        text: z4.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: z4.string().nullish(),
-      groundingChunkIndices: z4.array(z4.number()).nullish(),
-      supportChunkIndices: z4.array(z4.number()).nullish(),
-      confidenceScores: z4.array(z4.number()).nullish(),
-      confidenceScore: z4.array(z4.number()).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: z4.union([
-    z4.object({
-      webDynamicRetrievalScore: z4.number()
+  retrievalMetadata: z5.union([
+    z5.object({
+      webDynamicRetrievalScore: z5.number()
     }),
-    z4.object({})
+    z5.object({})
   ]).nullish()
 });
-var safetyRatingSchema = z4.object({
-  category: z4.string(),
-  probability: z4.string(),
-  probabilityScore: z4.number().nullish(),
-  severity: z4.string().nullish(),
-  severityScore: z4.number().nullish(),
-  blocked: z4.boolean().nullish()
+var safetyRatingSchema = z5.object({
+  category: z5.string(),
+  probability: z5.string(),
+  probabilityScore: z5.number().nullish(),
+  severity: z5.string().nullish(),
+  severityScore: z5.number().nullish(),
+  blocked: z5.boolean().nullish()
 });
-var responseSchema = z4.object({
-  candidates: z4.array(
-    z4.object({
-      content: contentSchema.nullish().or(z4.object({}).strict()),
-      finishReason: z4.string().nullish(),
-      safetyRatings: z4.array(safetyRatingSchema).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()
     })
   ),
-  usageMetadata: z4.object({
-    promptTokenCount: z4.number().nullish(),
-    candidatesTokenCount: z4.number().nullish(),
-    totalTokenCount: z4.number().nullish()
+  usageMetadata: z5.object({
+    promptTokenCount: z5.number().nullish(),
+    candidatesTokenCount: z5.number().nullish(),
+    totalTokenCount: z5.number().nullish()
   }).nullish()
 });
-var chunkSchema = z4.object({
-  candidates: z4.array(
-    z4.object({
+var chunkSchema = z5.object({
+  candidates: z5.array(
+    z5.object({
       content: contentSchema.nullish(),
-      finishReason: z4.string().nullish(),
-      safetyRatings: z4.array(safetyRatingSchema).nullish(),
+      finishReason: z5.string().nullish(),
+      safetyRatings: z5.array(safetyRatingSchema).nullish(),
       groundingMetadata: groundingMetadataSchema.nullish()
     })
   ).nullish(),
-  usageMetadata: z4.object({
-    promptTokenCount: z4.number().nullish(),
-    candidatesTokenCount: z4.number().nullish(),
-    totalTokenCount: z4.number().nullish()
-  }).nullish()
-});
-var googleGenerativeAIProviderOptionsSchema = z4.object({
-  responseModalities: z4.array(z4.enum(["TEXT", "IMAGE"])).nullish(),
-  thinkingConfig: z4.object({
-    thinkingBudget: z4.number().nullish()
+  usageMetadata: z5.object({
+    promptTokenCount: z5.number().nullish(),
+    candidatesTokenCount: z5.number().nullish(),
+    totalTokenCount: z5.number().nullish()
   }).nullish()
 });
 
@@ -898,9 +995,9 @@ function createGoogleGenerativeAI(options = {}) {
     }),
     ...options.headers
   });
-  const createChatModel = (modelId, settings = {}) => {
+  const createChatModel = (modelId) => {
     var _a2;
-    return new GoogleGenerativeAILanguageModel(modelId, settings, {
+    return new GoogleGenerativeAILanguageModel(modelId, {
       provider: "google.generative-ai",
       baseURL,
       headers: getHeaders,
@@ -920,13 +1017,13 @@ function createGoogleGenerativeAI(options = {}) {
     headers: getHeaders,
     fetch: options.fetch
   });
-  const provider = function(modelId, settings) {
+  const provider = function(modelId) {
     if (new.target) {
       throw new Error(
         "The Google Generative AI model function cannot be called with the new keyword."
       );
     }
-    return createChatModel(modelId, settings);
+    return createChatModel(modelId);
   };
   provider.languageModel = createChatModel;
   provider.chat = createChatModel;