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

package/dist/index.js

@@ -56,37 +56,54 @@ var googleGenerativeAIEmbeddingProviderOptions = import_zod2.z.object({
    * Optional. Optional reduced dimension for the output embedding.
    * If set, excessive values in the output embedding are truncated from the end.
    */
-  outputDimensionality: import_zod2.z.number().optional()
+  outputDimensionality: import_zod2.z.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: import_zod2.z.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 = (0, import_provider_utils2.parseProviderOptions)({
+    const googleOptions = await (0, import_provider_utils2.parseProviderOptions)({
       provider: "google",
       providerOptions,
       schema: googleGenerativeAIEmbeddingProviderOptions
-    })) != null ? _a : {};
+    });
     if (values.length > this.maxEmbeddingsPerCall) {
       throw new import_provider.TooManyEmbeddingValuesForCallError({
         provider: this.provider,
@@ -110,7 +127,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,
@@ -133,7 +151,7 @@ var googleGenerativeAITextEmbeddingResponseSchema = import_zod3.z.object({
 
 // src/google-generative-ai-language-model.ts
 var import_provider_utils4 = require("@ai-sdk/provider-utils");
-var import_zod4 = require("zod");
+var import_zod5 = require("zod");
 
 // src/convert-json-schema-to-openapi-schema.ts
 function convertJSONSchemaToOpenAPISchema(jsonSchema) {
@@ -350,13 +368,99 @@ function getModelPath(modelId) {
   return modelId.includes("/") ? modelId : `models/${modelId}`;
 }
 
+// src/google-generative-ai-options.ts
+var import_zod4 = require("zod");
+var dynamicRetrievalConfig = import_zod4.z.object({
+  /**
+   * The mode of the predictor to be used in dynamic retrieval.
+   */
+  mode: import_zod4.z.enum(["MODE_UNSPECIFIED", "MODE_DYNAMIC"]).optional(),
+  /**
+   * The threshold to be used in dynamic retrieval. If not set, a system default
+   * value is used.
+   */
+  dynamicThreshold: import_zod4.z.number().optional()
+});
+var googleGenerativeAIProviderOptions = import_zod4.z.object({
+  responseModalities: import_zod4.z.array(import_zod4.z.enum(["TEXT", "IMAGE"])).optional(),
+  thinkingConfig: import_zod4.z.object({
+    thinkingBudget: import_zod4.z.number().optional()
+  }).optional(),
+  /**
+  Optional.
+  The name of the cached content used as context to serve the prediction.
+  Format: cachedContents/{cachedContent}
+     */
+  cachedContent: import_zod4.z.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: import_zod4.z.boolean().optional(),
+  /**
+  Optional. A list of unique safety settings for blocking unsafe content.
+   */
+  safetySettings: import_zod4.z.array(
+    import_zod4.z.object({
+      category: import_zod4.z.enum([
+        "HARM_CATEGORY_UNSPECIFIED",
+        "HARM_CATEGORY_HATE_SPEECH",
+        "HARM_CATEGORY_DANGEROUS_CONTENT",
+        "HARM_CATEGORY_HARASSMENT",
+        "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+        "HARM_CATEGORY_CIVIC_INTEGRITY"
+      ]),
+      threshold: import_zod4.z.enum([
+        "HARM_BLOCK_THRESHOLD_UNSPECIFIED",
+        "BLOCK_LOW_AND_ABOVE",
+        "BLOCK_MEDIUM_AND_ABOVE",
+        "BLOCK_ONLY_HIGH",
+        "BLOCK_NONE",
+        "OFF"
+      ])
+    })
+  ).optional(),
+  threshold: import_zod4.z.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: import_zod4.z.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: import_zod4.z.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
 var import_provider3 = require("@ai-sdk/provider");
 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 = (0, import_provider_utils4.parseProviderOptions)({
+    const googleOptions = await (0, import_provider_utils4.parseProviderOptions)({
       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 = import_zod4.z.object({
-  role: import_zod4.z.string(),
-  parts: import_zod4.z.array(
-    import_zod4.z.union([
-      import_zod4.z.object({
-        text: import_zod4.z.string()
+var contentSchema = import_zod5.z.object({
+  role: import_zod5.z.string(),
+  parts: import_zod5.z.array(
+    import_zod5.z.union([
+      import_zod5.z.object({
+        text: import_zod5.z.string()
       }),
-      import_zod4.z.object({
-        functionCall: import_zod4.z.object({
-          name: import_zod4.z.string(),
-          args: import_zod4.z.unknown()
+      import_zod5.z.object({
+        functionCall: import_zod5.z.object({
+          name: import_zod5.z.string(),
+          args: import_zod5.z.unknown()
         })
       }),
-      import_zod4.z.object({
-        inlineData: import_zod4.z.object({
-          mimeType: import_zod4.z.string(),
-          data: import_zod4.z.string()
+      import_zod5.z.object({
+        inlineData: import_zod5.z.object({
+          mimeType: import_zod5.z.string(),
+          data: import_zod5.z.string()
         })
       })
     ])
   ).nullish()
 });
-var groundingChunkSchema = import_zod4.z.object({
-  web: import_zod4.z.object({ uri: import_zod4.z.string(), title: import_zod4.z.string() }).nullish(),
-  retrievedContext: import_zod4.z.object({ uri: import_zod4.z.string(), title: import_zod4.z.string() }).nullish()
+var groundingChunkSchema = import_zod5.z.object({
+  web: import_zod5.z.object({ uri: import_zod5.z.string(), title: import_zod5.z.string() }).nullish(),
+  retrievedContext: import_zod5.z.object({ uri: import_zod5.z.string(), title: import_zod5.z.string() }).nullish()
 });
-var groundingMetadataSchema = import_zod4.z.object({
-  webSearchQueries: import_zod4.z.array(import_zod4.z.string()).nullish(),
-  retrievalQueries: import_zod4.z.array(import_zod4.z.string()).nullish(),
-  searchEntryPoint: import_zod4.z.object({ renderedContent: import_zod4.z.string() }).nullish(),
-  groundingChunks: import_zod4.z.array(groundingChunkSchema).nullish(),
-  groundingSupports: import_zod4.z.array(
-    import_zod4.z.object({
-      segment: import_zod4.z.object({
-        startIndex: import_zod4.z.number().nullish(),
-        endIndex: import_zod4.z.number().nullish(),
-        text: import_zod4.z.string().nullish()
+var groundingMetadataSchema = import_zod5.z.object({
+  webSearchQueries: import_zod5.z.array(import_zod5.z.string()).nullish(),
+  retrievalQueries: import_zod5.z.array(import_zod5.z.string()).nullish(),
+  searchEntryPoint: import_zod5.z.object({ renderedContent: import_zod5.z.string() }).nullish(),
+  groundingChunks: import_zod5.z.array(groundingChunkSchema).nullish(),
+  groundingSupports: import_zod5.z.array(
+    import_zod5.z.object({
+      segment: import_zod5.z.object({
+        startIndex: import_zod5.z.number().nullish(),
+        endIndex: import_zod5.z.number().nullish(),
+        text: import_zod5.z.string().nullish()
       }),
-      segment_text: import_zod4.z.string().nullish(),
-      groundingChunkIndices: import_zod4.z.array(import_zod4.z.number()).nullish(),
-      supportChunkIndices: import_zod4.z.array(import_zod4.z.number()).nullish(),
-      confidenceScores: import_zod4.z.array(import_zod4.z.number()).nullish(),
-      confidenceScore: import_zod4.z.array(import_zod4.z.number()).nullish()
+      segment_text: import_zod5.z.string().nullish(),
+      groundingChunkIndices: import_zod5.z.array(import_zod5.z.number()).nullish(),
+      supportChunkIndices: import_zod5.z.array(import_zod5.z.number()).nullish(),
+      confidenceScores: import_zod5.z.array(import_zod5.z.number()).nullish(),
+      confidenceScore: import_zod5.z.array(import_zod5.z.number()).nullish()
     })
   ).nullish(),
-  retrievalMetadata: import_zod4.z.union([
-    import_zod4.z.object({
-      webDynamicRetrievalScore: import_zod4.z.number()
+  retrievalMetadata: import_zod5.z.union([
+    import_zod5.z.object({
+      webDynamicRetrievalScore: import_zod5.z.number()
     }),
-    import_zod4.z.object({})
+    import_zod5.z.object({})
   ]).nullish()
 });
-var safetyRatingSchema = import_zod4.z.object({
-  category: import_zod4.z.string(),
-  probability: import_zod4.z.string(),
-  probabilityScore: import_zod4.z.number().nullish(),
-  severity: import_zod4.z.string().nullish(),
-  severityScore: import_zod4.z.number().nullish(),
-  blocked: import_zod4.z.boolean().nullish()
+var safetyRatingSchema = import_zod5.z.object({
+  category: import_zod5.z.string(),
+  probability: import_zod5.z.string(),
+  probabilityScore: import_zod5.z.number().nullish(),
+  severity: import_zod5.z.string().nullish(),
+  severityScore: import_zod5.z.number().nullish(),
+  blocked: import_zod5.z.boolean().nullish()
 });
-var responseSchema = import_zod4.z.object({
-  candidates: import_zod4.z.array(
-    import_zod4.z.object({
-      content: contentSchema.nullish().or(import_zod4.z.object({}).strict()),
-      finishReason: import_zod4.z.string().nullish(),
-      safetyRatings: import_zod4.z.array(safetyRatingSchema).nullish(),
+var responseSchema = import_zod5.z.object({
+  candidates: import_zod5.z.array(
+    import_zod5.z.object({
+      content: contentSchema.nullish().or(import_zod5.z.object({}).strict()),
+      finishReason: import_zod5.z.string().nullish(),
+      safetyRatings: import_zod5.z.array(safetyRatingSchema).nullish(),
       groundingMetadata: groundingMetadataSchema.nullish()
     })
   ),
-  usageMetadata: import_zod4.z.object({
-    promptTokenCount: import_zod4.z.number().nullish(),
-    candidatesTokenCount: import_zod4.z.number().nullish(),
-    totalTokenCount: import_zod4.z.number().nullish()
+  usageMetadata: import_zod5.z.object({
+    promptTokenCount: import_zod5.z.number().nullish(),
+    candidatesTokenCount: import_zod5.z.number().nullish(),
+    totalTokenCount: import_zod5.z.number().nullish()
   }).nullish()
 });
-var chunkSchema = import_zod4.z.object({
-  candidates: import_zod4.z.array(
-    import_zod4.z.object({
+var chunkSchema = import_zod5.z.object({
+  candidates: import_zod5.z.array(
+    import_zod5.z.object({
       content: contentSchema.nullish(),
-      finishReason: import_zod4.z.string().nullish(),
-      safetyRatings: import_zod4.z.array(safetyRatingSchema).nullish(),
+      finishReason: import_zod5.z.string().nullish(),
+      safetyRatings: import_zod5.z.array(safetyRatingSchema).nullish(),
       groundingMetadata: groundingMetadataSchema.nullish()
     })
   ).nullish(),
-  usageMetadata: import_zod4.z.object({
-    promptTokenCount: import_zod4.z.number().nullish(),
-    candidatesTokenCount: import_zod4.z.number().nullish(),
-    totalTokenCount: import_zod4.z.number().nullish()
-  }).nullish()
-});
-var googleGenerativeAIProviderOptionsSchema = import_zod4.z.object({
-  responseModalities: import_zod4.z.array(import_zod4.z.enum(["TEXT", "IMAGE"])).nullish(),
-  thinkingConfig: import_zod4.z.object({
-    thinkingBudget: import_zod4.z.number().nullish()
+  usageMetadata: import_zod5.z.object({
+    promptTokenCount: import_zod5.z.number().nullish(),
+    candidatesTokenCount: import_zod5.z.number().nullish(),
+    totalTokenCount: import_zod5.z.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;