@ai-sdk/google 4.0.50 → 4.0.51

package/dist/internal/index.d.ts

@@ -1,12 +1,75 @@
 import * as _ai_sdk_provider_utils from '@ai-sdk/provider-utils';
 import { WORKFLOW_SERIALIZE, WORKFLOW_DESERIALIZE, Resolvable, FetchFunction, InferSchema } from '@ai-sdk/provider-utils';
 import * as _ai_sdk_provider from '@ai-sdk/provider';
-import { LanguageModelV4, JSONObject, LanguageModelV4CallOptions, LanguageModelV4GenerateResult, LanguageModelV4StreamResult, SpeechModelV4 } from '@ai-sdk/provider';
+import { LanguageModelV4, JSONObject, LanguageModelV4CallOptions, SharedV4Warning, LanguageModelV4GenerateResult, LanguageModelV4StreamResult, SpeechModelV4 } from '@ai-sdk/provider';
 import { z } from 'zod/v4';
 
+type GoogleSystemInstruction = {
+    parts: Array<{
+        text: string;
+    }>;
+};
+type GoogleContent = {
+    role: 'user' | 'model';
+    parts: Array<GoogleContentPart>;
+};
+type GoogleContentPart = {
+    text: string;
+    thought?: boolean;
+    thoughtSignature?: string;
+} | {
+    inlineData: {
+        mimeType: string;
+        data: string;
+    };
+    thought?: boolean;
+    thoughtSignature?: string;
+} | {
+    functionCall: {
+        id?: string;
+        name: string;
+        args: unknown;
+    };
+    thoughtSignature?: string;
+} | {
+    functionResponse: {
+        id?: string;
+        name: string;
+        response: unknown;
+        parts?: Array<GoogleFunctionResponsePart>;
+    };
+} | {
+    fileData: {
+        mimeType: string;
+        fileUri: string;
+    };
+    thought?: boolean;
+    thoughtSignature?: string;
+} | {
+    toolCall: {
+        toolType: string;
+        args?: unknown;
+        id: string;
+    };
+    thoughtSignature?: string;
+} | {
+    toolResponse: {
+        toolType: string;
+        response?: unknown;
+        id: string;
+    };
+    thoughtSignature?: string;
+};
+type GoogleFunctionResponsePart = {
+    inlineData: {
+        mimeType: string;
+        data: string;
+    };
+};
+
 type GoogleModelId = 'gemini-2.0-flash' | 'gemini-2.0-flash-001' | 'gemini-2.0-flash-lite' | 'gemini-2.0-flash-lite-001' | 'gemini-2.5-pro' | 'gemini-2.5-flash' | 'gemini-2.5-flash-image' | 'gemini-2.5-flash-lite' | 'gemini-2.5-flash-preview-tts' | 'gemini-2.5-pro-preview-tts' | 'gemini-2.5-flash-native-audio-latest' | 'gemini-2.5-flash-native-audio-preview-09-2025' | 'gemini-2.5-flash-native-audio-preview-12-2025' | 'gemini-2.5-computer-use-preview-10-2025' | 'gemini-3-pro-preview' | 'gemini-3-pro-image-preview' | 'gemini-3-flash-preview' | 'gemini-3.1-pro-preview' | 'gemini-3.1-pro-preview-customtools' | 'gemini-3.1-flash-image-preview' | 'gemini-3.1-flash-lite-preview' | 'gemini-3.1-flash-tts-preview' | 'gemini-3.5-flash' | 'gemini-3.5-flash-lite' | 'gemini-3.6-flash' | 'gemini-3.7-flash' | 'gemini-pro-latest' | 'gemini-flash-latest' | 'gemini-flash-lite-latest' | 'deep-research-pro-preview-12-2025' | 'deep-research-max-preview-04-2026' | 'deep-research-preview-04-2026' | 'nano-banana-pro-preview' | 'aqa' | 'gemini-robotics-er-1.5-preview' | 'gemma-3-1b-it' | 'gemma-3-4b-it' | 'gemma-3n-e4b-it' | 'gemma-3n-e2b-it' | 'gemma-3-12b-it' | 'gemma-3-27b-it' | (string & {});
 
-type GoogleConfig = {
+type GoogleLanguageModelConfig = {
     provider: string;
     baseURL: string;
     headers?: Resolvable<Record<string, string | undefined>>;
@@ -28,12 +91,89 @@ declare class GoogleLanguageModel implements LanguageModelV4 {
     };
     static [WORKFLOW_DESERIALIZE](options: {
         modelId: string;
-        config: GoogleConfig;
+        config: GoogleLanguageModelConfig;
     }): GoogleLanguageModel;
-    constructor(modelId: GoogleModelId, config: GoogleConfig);
+    constructor(modelId: GoogleModelId, config: GoogleLanguageModelConfig);
     get provider(): string;
     get supportedUrls(): Record<string, RegExp[]> | PromiseLike<Record<string, RegExp[]>>;
-    private getArgs;
+    protected getArgs({ prompt, maxOutputTokens, temperature, topP, topK, frequencyPenalty, presencePenalty, stopSequences, responseFormat, seed, tools, toolChoice, reasoning, providerOptions, }: LanguageModelV4CallOptions, { isStreaming }?: {
+        isStreaming?: boolean;
+    }): Promise<{
+        args: {
+            generationConfig: {
+                imageConfig?: {
+                    aspectRatio?: "1:1" | "2:3" | "3:2" | "3:4" | "4:3" | "4:5" | "5:4" | "9:16" | "16:9" | "21:9" | "1:8" | "8:1" | "1:4" | "4:1" | undefined;
+                    imageSize?: "1K" | "2K" | "4K" | "512" | undefined;
+                    personGeneration?: "PERSON_GENERATION_UNSPECIFIED" | "ALLOW_ALL" | "ALLOW_ADULT" | "ALLOW_NONE" | undefined;
+                    prominentPeople?: "PROMINENT_PEOPLE_UNSPECIFIED" | "ALLOW_PROMINENT_PEOPLE" | "BLOCK_PROMINENT_PEOPLE" | undefined;
+                    imageOutputOptions?: {
+                        mimeType?: "image/jpeg" | "image/png" | undefined;
+                        compressionQuality?: number | undefined;
+                    } | undefined;
+                } | undefined;
+                mediaResolution?: "MEDIA_RESOLUTION_UNSPECIFIED" | "MEDIA_RESOLUTION_LOW" | "MEDIA_RESOLUTION_MEDIUM" | "MEDIA_RESOLUTION_HIGH" | undefined;
+                responseModalities: ("TEXT" | "IMAGE")[] | undefined;
+                thinkingConfig: {
+                    thinkingBudget?: number | undefined;
+                    includeThoughts?: boolean | undefined;
+                    thinkingLevel?: "minimal" | "low" | "medium" | "high" | undefined;
+                } | undefined;
+                audioTimestamp?: true | undefined;
+                maxOutputTokens: number | undefined;
+                temperature: number | undefined;
+                topK: number | undefined;
+                topP: number | undefined;
+                frequencyPenalty: number | undefined;
+                presencePenalty: number | undefined;
+                stopSequences: string[] | undefined;
+                seed: number | undefined;
+                responseMimeType: string | undefined;
+                responseSchema: unknown;
+            };
+            contents: GoogleContent[];
+            systemInstruction: GoogleSystemInstruction | undefined;
+            safetySettings: {
+                category: "HARM_CATEGORY_UNSPECIFIED" | "HARM_CATEGORY_HATE_SPEECH" | "HARM_CATEGORY_DANGEROUS_CONTENT" | "HARM_CATEGORY_HARASSMENT" | "HARM_CATEGORY_SEXUALLY_EXPLICIT" | "HARM_CATEGORY_CIVIC_INTEGRITY";
+                threshold: "HARM_BLOCK_THRESHOLD_UNSPECIFIED" | "BLOCK_LOW_AND_ABOVE" | "BLOCK_MEDIUM_AND_ABOVE" | "BLOCK_ONLY_HIGH" | "BLOCK_NONE" | "OFF";
+            }[] | undefined;
+            tools: ({
+                functionDeclarations: Array<{
+                    name: string;
+                    description: string;
+                    parameters: unknown;
+                }>;
+            } | Record<string, any>)[] | undefined;
+            toolConfig: {
+                retrievalConfig?: {
+                    latLng?: {
+                        latitude: number;
+                        longitude: number;
+                    } | undefined;
+                } | undefined;
+                functionCallingConfig?: {
+                    mode: "AUTO" | "NONE" | "ANY" | "VALIDATED";
+                    allowedFunctionNames?: string[];
+                    streamFunctionCallArguments?: boolean;
+                } | {
+                    streamFunctionCallArguments: true;
+                    mode?: "AUTO" | "NONE" | "ANY" | "VALIDATED" | undefined;
+                    allowedFunctionNames?: string[];
+                } | undefined;
+                includeServerSideToolInvocations?: boolean;
+            } | undefined;
+            cachedContent: string | undefined;
+            labels: Record<string, string> | undefined;
+            serviceTier: "standard" | "flex" | "priority" | undefined;
+        };
+        warnings: SharedV4Warning[];
+        providerOptionsNames: readonly string[];
+        extraHeaders: Record<string, string> | undefined;
+    }>;
+    protected convertGenerateContentResponse({ response, warnings, providerOptionsNames, }: {
+        response: InferSchema<typeof responseSchema>;
+        warnings: SharedV4Warning[];
+        providerOptionsNames: readonly string[];
+    }): LanguageModelV4GenerateResult;
     doGenerate(options: LanguageModelV4CallOptions): Promise<LanguageModelV4GenerateResult>;
     doStream(options: LanguageModelV4CallOptions): Promise<LanguageModelV4StreamResult>;
 }
@@ -419,4 +559,4 @@ declare class GoogleInteractionsLanguageModel implements LanguageModelV4 {
  */
 type GoogleInteractionsAgentName = 'deep-research-pro-preview-12-2025' | 'deep-research-preview-04-2026' | 'deep-research-max-preview-04-2026' | 'antigravity-preview-05-2026';
 
-export { type GoogleInteractionsAgentName, GoogleInteractionsLanguageModel, type GoogleInteractionsModelId, type GoogleInteractionsModelInput, GoogleLanguageModel, type GoogleModelId, GoogleSpeechModel, type GroundingMetadataSchema, type PromptFeedbackSchema, type SafetyRatingSchema, type UrlContextMetadataSchema, type UsageMetadataSchema, getGroundingMetadataSchema, getUrlContextMetadataSchema, googleTools };
+export { type GoogleInteractionsAgentName, GoogleInteractionsLanguageModel, type GoogleInteractionsModelId, type GoogleInteractionsModelInput, GoogleLanguageModel, type GoogleLanguageModelConfig, type GoogleModelId, GoogleSpeechModel, type GroundingMetadataSchema, type PromptFeedbackSchema, type SafetyRatingSchema, type UrlContextMetadataSchema, type UsageMetadataSchema, getGroundingMetadataSchema, getUrlContextMetadataSchema, googleTools, responseSchema };

package/dist/internal/index.js

@@ -1740,32 +1740,15 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
       extraHeaders: vertexPaygoHeaders
     };
   }
-  async doGenerate(options) {
+  convertGenerateContentResponse({
+    response,
+    warnings,
+    providerOptionsNames
+  }) {
     var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m, _n, _o, _p;
-    const { args, warnings, providerOptionsNames, extraHeaders } = await this.getArgs(options);
     const wrapProviderMetadata = (payload) => Object.fromEntries(
       providerOptionsNames.map((name) => [name, payload])
     );
-    const mergedHeaders = combineHeaders(
-      this.config.headers ? await resolve(this.config.headers) : void 0,
-      options.headers,
-      extraHeaders
-    );
-    const {
-      responseHeaders,
-      value: response,
-      rawValue: rawResponse
-    } = await postJsonToApi({
-      url: `${this.config.baseURL}/${getModelPath(
-        this.modelId
-      )}:generateContent`,
-      headers: mergedHeaders,
-      body: args,
-      failedResponseHandler: googleFailedResponseHandler,
-      successfulResponseHandler: createJsonResponseHandler(responseSchema),
-      abortSignal: options.abortSignal,
-      fetch: this.config.fetch
-    });
     const candidate = response.candidates[0];
     const content = [];
     const parts = (_b = (_a = candidate.content) == null ? void 0 : _a.parts) != null ? _b : [];
@@ -1899,10 +1882,44 @@ var GoogleLanguageModel = class _GoogleLanguageModel {
         finishMessage: (_n = candidate.finishMessage) != null ? _n : null,
         serviceTier: (_o = usageMetadata == null ? void 0 : usageMetadata.serviceTier) != null ? _o : null
       }),
-      request: { body: args },
       response: {
         // TODO timestamp, model id
-        id: (_p = response.responseId) != null ? _p : void 0,
+        id: (_p = response.responseId) != null ? _p : void 0
+      }
+    };
+  }
+  async doGenerate(options) {
+    const { args, warnings, providerOptionsNames, extraHeaders } = await this.getArgs(options);
+    const mergedHeaders = combineHeaders(
+      this.config.headers ? await resolve(this.config.headers) : void 0,
+      options.headers,
+      extraHeaders
+    );
+    const {
+      responseHeaders,
+      value: response,
+      rawValue: rawResponse
+    } = await postJsonToApi({
+      url: `${this.config.baseURL}/${getModelPath(
+        this.modelId
+      )}:generateContent`,
+      headers: mergedHeaders,
+      body: args,
+      failedResponseHandler: googleFailedResponseHandler,
+      successfulResponseHandler: createJsonResponseHandler(responseSchema),
+      abortSignal: options.abortSignal,
+      fetch: this.config.fetch
+    });
+    const result = this.convertGenerateContentResponse({
+      response,
+      warnings,
+      providerOptionsNames
+    });
+    return {
+      ...result,
+      request: { body: args },
+      response: {
+        ...result.response,
         headers: responseHeaders,
         body: rawResponse
       }
@@ -6075,6 +6092,7 @@ export {
   GoogleSpeechModel,
   getGroundingMetadataSchema,
   getUrlContextMetadataSchema,
-  googleTools
+  googleTools,
+  responseSchema
 };
 //# sourceMappingURL=index.js.map