@google/genai 1.15.0 → 1.17.0

package/dist/node/index.mjs

@@ -805,6 +805,10 @@ var FunctionCallingConfigMode;
      * Model will not predict any function calls. Model behavior is same as when not passing any function declarations.
      */
     FunctionCallingConfigMode["NONE"] = "NONE";
+    /**
+     * Model decides to predict either a function call or a natural language response, but will validate function calls with constrained decoding. If "allowed_function_names" are set, the predicted function call will be limited to any one of "allowed_function_names", else the predicted function call will be any one of the provided "function_declarations".
+     */
+    FunctionCallingConfigMode["VALIDATED"] = "VALIDATED";
 })(FunctionCallingConfigMode || (FunctionCallingConfigMode = {}));
 /** Enum that controls the safety filter level for objectionable content. */
 var SafetyFilterLevel;
@@ -891,7 +895,7 @@ var SubjectReferenceType;
     SubjectReferenceType["SUBJECT_TYPE_ANIMAL"] = "SUBJECT_TYPE_ANIMAL";
     SubjectReferenceType["SUBJECT_TYPE_PRODUCT"] = "SUBJECT_TYPE_PRODUCT";
 })(SubjectReferenceType || (SubjectReferenceType = {}));
-/** Enum representing the Imagen 3 Edit mode. */
+/** Enum representing the editing mode. */
 var EditMode;
 (function (EditMode) {
     EditMode["EDIT_MODE_DEFAULT"] = "EDIT_MODE_DEFAULT";
@@ -912,6 +916,21 @@ var SegmentMode;
     SegmentMode["SEMANTIC"] = "SEMANTIC";
     SegmentMode["INTERACTIVE"] = "INTERACTIVE";
 })(SegmentMode || (SegmentMode = {}));
+/** Enum for the reference type of a video generation reference image. */
+var VideoGenerationReferenceType;
+(function (VideoGenerationReferenceType) {
+    /**
+     * A reference image that provides assets to the generated video,
+        such as the scene, an object, a character, etc.
+     */
+    VideoGenerationReferenceType["ASSET"] = "ASSET";
+    /**
+     * A reference image that provides aesthetics including colors,
+        lighting, texture, etc., to be used as the style of the generated video,
+        such as 'anime', 'photography', 'origami', etc.
+     */
+    VideoGenerationReferenceType["STYLE"] = "STYLE";
+})(VideoGenerationReferenceType || (VideoGenerationReferenceType = {}));
 /** Enum that controls the compression quality of the generated videos. */
 var VideoCompressionQuality;
 (function (VideoCompressionQuality) {
@@ -1574,6 +1593,65 @@ class ComputeTokensResponse {
 /** Response with generated videos. */
 class GenerateVideosResponse {
 }
+/** A video generation operation. */
+class GenerateVideosOperation {
+    /**
+     * Instantiates an Operation of the same type as the one being called with the fields set from the API response.
+     * @internal
+     */
+    _fromAPIResponse({ apiResponse, isVertexAI, }) {
+        const operation = new GenerateVideosOperation();
+        operation.name = apiResponse['name'];
+        operation.metadata = apiResponse['metadata'];
+        operation.done = apiResponse['done'];
+        operation.error = apiResponse['error'];
+        if (isVertexAI) {
+            const response = apiResponse['response'];
+            if (response) {
+                const operationResponse = new GenerateVideosResponse();
+                const responseVideos = response['videos'];
+                operationResponse.generatedVideos = responseVideos === null || responseVideos === void 0 ? void 0 : responseVideos.map((generatedVideo) => {
+                    return {
+                        video: {
+                            uri: generatedVideo['gcsUri'],
+                            videoBytes: generatedVideo['bytesBase64Encoded']
+                                ? tBytes$1(generatedVideo['bytesBase64Encoded'])
+                                : undefined,
+                            mimeType: generatedVideo['mimeType'],
+                        },
+                    };
+                });
+                operationResponse.raiMediaFilteredCount = response['raiMediaFilteredCount'];
+                operationResponse.raiMediaFilteredReasons = response['raiMediaFilteredReasons'];
+                operation.response = operationResponse;
+            }
+        }
+        else {
+            const response = apiResponse['response'];
+            if (response) {
+                const operationResponse = new GenerateVideosResponse();
+                const generatedVideoResponse = response['generateVideoResponse'];
+                const responseVideos = generatedVideoResponse === null || generatedVideoResponse === void 0 ? void 0 : generatedVideoResponse['generatedSamples'];
+                operationResponse.generatedVideos = responseVideos === null || responseVideos === void 0 ? void 0 : responseVideos.map((generatedVideo) => {
+                    const video = generatedVideo['video'];
+                    return {
+                        video: {
+                            uri: video === null || video === void 0 ? void 0 : video['uri'],
+                            videoBytes: (video === null || video === void 0 ? void 0 : video['encodedVideo'])
+                                ? tBytes$1(video === null || video === void 0 ? void 0 : video['encodedVideo'])
+                                : undefined,
+                            mimeType: generatedVideo['encoding'],
+                        },
+                    };
+                });
+                operationResponse.raiMediaFilteredCount = generatedVideoResponse === null || generatedVideoResponse === void 0 ? void 0 : generatedVideoResponse['raiMediaFilteredCount'];
+                operationResponse.raiMediaFilteredReasons = generatedVideoResponse === null || generatedVideoResponse === void 0 ? void 0 : generatedVideoResponse['raiMediaFilteredReasons'];
+                operation.response = operationResponse;
+            }
+        }
+        return operation;
+    }
+}
 /** Response for the list tuning jobs method. */
 class ListTuningJobsResponse {
 }
@@ -1765,65 +1843,6 @@ class LiveServerMessage {
         return data.length > 0 ? btoa(data) : undefined;
     }
 }
-/** A video generation long-running operation. */
-class GenerateVideosOperation {
-    /**
-     * Instantiates an Operation of the same type as the one being called with the fields set from the API response.
-     * @internal
-     */
-    _fromAPIResponse({ apiResponse, isVertexAI, }) {
-        const operation = new GenerateVideosOperation();
-        operation.name = apiResponse['name'];
-        operation.metadata = apiResponse['metadata'];
-        operation.done = apiResponse['done'];
-        operation.error = apiResponse['error'];
-        if (isVertexAI) {
-            const response = apiResponse['response'];
-            if (response) {
-                const operationResponse = new GenerateVideosResponse();
-                const responseVideos = response['videos'];
-                operationResponse.generatedVideos = responseVideos === null || responseVideos === void 0 ? void 0 : responseVideos.map((generatedVideo) => {
-                    return {
-                        video: {
-                            uri: generatedVideo['gcsUri'],
-                            videoBytes: generatedVideo['bytesBase64Encoded']
-                                ? tBytes$1(generatedVideo['bytesBase64Encoded'])
-                                : undefined,
-                            mimeType: generatedVideo['mimeType'],
-                        },
-                    };
-                });
-                operationResponse.raiMediaFilteredCount = response['raiMediaFilteredCount'];
-                operationResponse.raiMediaFilteredReasons = response['raiMediaFilteredReasons'];
-                operation.response = operationResponse;
-            }
-        }
-        else {
-            const response = apiResponse['response'];
-            if (response) {
-                const operationResponse = new GenerateVideosResponse();
-                const generatedVideoResponse = response['generateVideoResponse'];
-                const responseVideos = generatedVideoResponse === null || generatedVideoResponse === void 0 ? void 0 : generatedVideoResponse['generatedSamples'];
-                operationResponse.generatedVideos = responseVideos === null || responseVideos === void 0 ? void 0 : responseVideos.map((generatedVideo) => {
-                    const video = generatedVideo['video'];
-                    return {
-                        video: {
-                            uri: video === null || video === void 0 ? void 0 : video['uri'],
-                            videoBytes: (video === null || video === void 0 ? void 0 : video['encodedVideo'])
-                                ? tBytes$1(video === null || video === void 0 ? void 0 : video['encodedVideo'])
-                                : undefined,
-                            mimeType: generatedVideo['encoding'],
-                        },
-                    };
-                });
-                operationResponse.raiMediaFilteredCount = generatedVideoResponse === null || generatedVideoResponse === void 0 ? void 0 : generatedVideoResponse['raiMediaFilteredCount'];
-                operationResponse.raiMediaFilteredReasons = generatedVideoResponse === null || generatedVideoResponse === void 0 ? void 0 : generatedVideoResponse['raiMediaFilteredReasons'];
-                operation.response = operationResponse;
-            }
-        }
-        return operation;
-    }
-}
 /** Client generated response to a `ToolCall` received from the server.
 
   Individual `FunctionResponse` objects are matched to the respective
@@ -2470,6 +2489,9 @@ function mcpToGeminiTool(mcpTool, config = {}) {
         description: mcpToolSchema['description'],
         parametersJsonSchema: mcpToolSchema['inputSchema'],
     };
+    if (mcpToolSchema['outputSchema']) {
+        functionDeclaration['responseJsonSchema'] = mcpToolSchema['outputSchema'];
+    }
     if (config.behavior) {
         functionDeclaration['behavior'] = config.behavior;
     }
@@ -5717,8 +5739,14 @@ function cachedContentFromMldev(fromObject) {
     }
     return toObject;
 }
-function deleteCachedContentResponseFromMldev() {
+function deleteCachedContentResponseFromMldev(fromObject) {
     const toObject = {};
+    const fromSdkHttpResponse = getValueByPath(fromObject, [
+        'sdkHttpResponse',
+    ]);
+    if (fromSdkHttpResponse != null) {
+        setValueByPath(toObject, ['sdkHttpResponse'], fromSdkHttpResponse);
+    }
     return toObject;
 }
 function listCachedContentsResponseFromMldev(fromObject) {
@@ -5783,8 +5811,14 @@ function cachedContentFromVertex(fromObject) {
     }
     return toObject;
 }
-function deleteCachedContentResponseFromVertex() {
+function deleteCachedContentResponseFromVertex(fromObject) {
     const toObject = {};
+    const fromSdkHttpResponse = getValueByPath(fromObject, [
+        'sdkHttpResponse',
+    ]);
+    if (fromSdkHttpResponse != null) {
+        setValueByPath(toObject, ['sdkHttpResponse'], fromSdkHttpResponse);
+    }
     return toObject;
 }
 function listCachedContentsResponseFromVertex(fromObject) {
@@ -6021,10 +6055,16 @@ class Caches extends BaseModule {
                 abortSignal: (_b = params.config) === null || _b === void 0 ? void 0 : _b.abortSignal,
             })
                 .then((httpResponse) => {
-                return httpResponse.json();
+                return httpResponse.json().then((jsonResponse) => {
+                    const response = jsonResponse;
+                    response.sdkHttpResponse = {
+                        headers: httpResponse.headers,
+                    };
+                    return response;
+                });
             });
-            return response.then(() => {
-                const resp = deleteCachedContentResponseFromVertex();
+            return response.then((apiResponse) => {
+                const resp = deleteCachedContentResponseFromVertex(apiResponse);
                 const typedResp = new DeleteCachedContentResponse();
                 Object.assign(typedResp, resp);
                 return typedResp;
@@ -6047,10 +6087,16 @@ class Caches extends BaseModule {
                 abortSignal: (_d = params.config) === null || _d === void 0 ? void 0 : _d.abortSignal,
             })
                 .then((httpResponse) => {
-                return httpResponse.json();
+                return httpResponse.json().then((jsonResponse) => {
+                    const response = jsonResponse;
+                    response.sdkHttpResponse = {
+                        headers: httpResponse.headers,
+                    };
+                    return response;
+                });
             });
-            return response.then(() => {
-                const resp = deleteCachedContentResponseFromMldev();
+            return response.then((apiResponse) => {
+                const resp = deleteCachedContentResponseFromMldev(apiResponse);
                 const typedResp = new DeleteCachedContentResponse();
                 Object.assign(typedResp, resp);
                 return typedResp;
@@ -6854,8 +6900,14 @@ function createFileResponseFromMldev(fromObject) {
     }
     return toObject;
 }
-function deleteFileResponseFromMldev() {
+function deleteFileResponseFromMldev(fromObject) {
     const toObject = {};
+    const fromSdkHttpResponse = getValueByPath(fromObject, [
+        'sdkHttpResponse',
+    ]);
+    if (fromSdkHttpResponse != null) {
+        setValueByPath(toObject, ['sdkHttpResponse'], fromSdkHttpResponse);
+    }
     return toObject;
 }
 
@@ -7124,10 +7176,16 @@ class Files extends BaseModule {
                 abortSignal: (_b = params.config) === null || _b === void 0 ? void 0 : _b.abortSignal,
             })
                 .then((httpResponse) => {
-                return httpResponse.json();
+                return httpResponse.json().then((jsonResponse) => {
+                    const response = jsonResponse;
+                    response.sdkHttpResponse = {
+                        headers: httpResponse.headers,
+                    };
+                    return response;
+                });
             });
-            return response.then(() => {
-                const resp = deleteFileResponseFromMldev();
+            return response.then((apiResponse) => {
+                const resp = deleteFileResponseFromMldev(apiResponse);
                 const typedResp = new DeleteFileResponse();
                 Object.assign(typedResp, resp);
                 return typedResp;
@@ -10430,6 +10488,21 @@ function imageToMldev(fromObject) {
     }
     return toObject;
 }
+function generateVideosSourceToMldev(fromObject, parentObject) {
+    const toObject = {};
+    const fromPrompt = getValueByPath(fromObject, ['prompt']);
+    if (parentObject !== undefined && fromPrompt != null) {
+        setValueByPath(parentObject, ['instances[0]', 'prompt'], fromPrompt);
+    }
+    const fromImage = getValueByPath(fromObject, ['image']);
+    if (parentObject !== undefined && fromImage != null) {
+        setValueByPath(parentObject, ['instances[0]', 'image'], imageToMldev(fromImage));
+    }
+    if (getValueByPath(fromObject, ['video']) !== undefined) {
+        throw new Error('video parameter is not supported in Gemini API.');
+    }
+    return toObject;
+}
 function generateVideosConfigToMldev(fromObject, parentObject) {
     const toObject = {};
     const fromNumberOfVideos = getValueByPath(fromObject, [
@@ -10457,8 +10530,9 @@ function generateVideosConfigToMldev(fromObject, parentObject) {
     if (parentObject !== undefined && fromAspectRatio != null) {
         setValueByPath(parentObject, ['parameters', 'aspectRatio'], fromAspectRatio);
     }
-    if (getValueByPath(fromObject, ['resolution']) !== undefined) {
-        throw new Error('resolution parameter is not supported in Gemini API.');
+    const fromResolution = getValueByPath(fromObject, ['resolution']);
+    if (parentObject !== undefined && fromResolution != null) {
+        setValueByPath(parentObject, ['parameters', 'resolution'], fromResolution);
     }
     const fromPersonGeneration = getValueByPath(fromObject, [
         'personGeneration',
@@ -10512,6 +10586,10 @@ function generateVideosParametersToMldev(apiClient, fromObject) {
     if (getValueByPath(fromObject, ['video']) !== undefined) {
         throw new Error('video parameter is not supported in Gemini API.');
     }
+    const fromSource = getValueByPath(fromObject, ['source']);
+    if (fromSource != null) {
+        setValueByPath(toObject, ['config'], generateVideosSourceToMldev(fromSource, toObject));
+    }
     const fromConfig = getValueByPath(fromObject, ['config']);
     if (fromConfig != null) {
         setValueByPath(toObject, ['config'], generateVideosConfigToMldev(fromConfig, toObject));
@@ -11636,6 +11714,10 @@ function editImageParametersInternalToVertex(apiClient, fromObject) {
 }
 function upscaleImageAPIConfigInternalToVertex(fromObject, parentObject) {
     const toObject = {};
+    const fromOutputGcsUri = getValueByPath(fromObject, ['outputGcsUri']);
+    if (parentObject !== undefined && fromOutputGcsUri != null) {
+        setValueByPath(parentObject, ['parameters', 'storageUri'], fromOutputGcsUri);
+    }
     const fromIncludeRaiReason = getValueByPath(fromObject, [
         'includeRaiReason',
     ]);
@@ -11764,6 +11846,10 @@ function recontextImageConfigToVertex(fromObject, parentObject) {
     if (parentObject !== undefined && fromPersonGeneration != null) {
         setValueByPath(parentObject, ['parameters', 'personGeneration'], fromPersonGeneration);
     }
+    const fromAddWatermark = getValueByPath(fromObject, ['addWatermark']);
+    if (parentObject !== undefined && fromAddWatermark != null) {
+        setValueByPath(parentObject, ['parameters', 'addWatermark'], fromAddWatermark);
+    }
     const fromOutputMimeType = getValueByPath(fromObject, [
         'outputMimeType',
     ]);
@@ -12040,6 +12126,22 @@ function videoToVertex(fromObject) {
     }
     return toObject;
 }
+function generateVideosSourceToVertex(fromObject, parentObject) {
+    const toObject = {};
+    const fromPrompt = getValueByPath(fromObject, ['prompt']);
+    if (parentObject !== undefined && fromPrompt != null) {
+        setValueByPath(parentObject, ['instances[0]', 'prompt'], fromPrompt);
+    }
+    const fromImage = getValueByPath(fromObject, ['image']);
+    if (parentObject !== undefined && fromImage != null) {
+        setValueByPath(parentObject, ['instances[0]', 'image'], imageToVertex(fromImage));
+    }
+    const fromVideo = getValueByPath(fromObject, ['video']);
+    if (parentObject !== undefined && fromVideo != null) {
+        setValueByPath(parentObject, ['instances[0]', 'video'], videoToVertex(fromVideo));
+    }
+    return toObject;
+}
 function videoGenerationReferenceImageToVertex(fromObject) {
     const toObject = {};
     const fromImage = getValueByPath(fromObject, ['image']);
@@ -12158,6 +12260,10 @@ function generateVideosParametersToVertex(apiClient, fromObject) {
     if (fromVideo != null) {
         setValueByPath(toObject, ['instances[0]', 'video'], videoToVertex(fromVideo));
     }
+    const fromSource = getValueByPath(fromObject, ['source']);
+    if (fromSource != null) {
+        setValueByPath(toObject, ['config'], generateVideosSourceToVertex(fromSource, toObject));
+    }
     const fromConfig = getValueByPath(fromObject, ['config']);
     if (fromConfig != null) {
         setValueByPath(toObject, ['config'], generateVideosConfigToVertex(fromConfig, toObject));
@@ -12608,8 +12714,14 @@ function listModelsResponseFromMldev(fromObject) {
     }
     return toObject;
 }
-function deleteModelResponseFromMldev() {
+function deleteModelResponseFromMldev(fromObject) {
     const toObject = {};
+    const fromSdkHttpResponse = getValueByPath(fromObject, [
+        'sdkHttpResponse',
+    ]);
+    if (fromSdkHttpResponse != null) {
+        setValueByPath(toObject, ['sdkHttpResponse'], fromSdkHttpResponse);
+    }
     return toObject;
 }
 function countTokensResponseFromMldev(fromObject) {
@@ -13354,8 +13466,14 @@ function listModelsResponseFromVertex(fromObject) {
     }
     return toObject;
 }
-function deleteModelResponseFromVertex() {
+function deleteModelResponseFromVertex(fromObject) {
     const toObject = {};
+    const fromSdkHttpResponse = getValueByPath(fromObject, [
+        'sdkHttpResponse',
+    ]);
+    if (fromSdkHttpResponse != null) {
+        setValueByPath(toObject, ['sdkHttpResponse'], fromSdkHttpResponse);
+    }
     return toObject;
 }
 function countTokensResponseFromVertex(fromObject) {
@@ -13472,7 +13590,7 @@ const CONTENT_TYPE_HEADER = 'Content-Type';
 const SERVER_TIMEOUT_HEADER = 'X-Server-Timeout';
 const USER_AGENT_HEADER = 'User-Agent';
 const GOOGLE_API_CLIENT_HEADER = 'x-goog-api-client';
-const SDK_VERSION = '1.15.0'; // x-release-please-version
+const SDK_VERSION = '1.17.0'; // x-release-please-version
 const LIBRARY_LABEL = `google-genai-sdk/${SDK_VERSION}`;
 const VERTEX_AI_API_DEFAULT_VERSION = 'v1beta1';
 const GOOGLE_AI_API_DEFAULT_VERSION = 'v1beta';
@@ -15244,7 +15362,9 @@ class Models extends BaseModule {
          * ```ts
          * const operation = await ai.models.generateVideos({
          *  model: 'veo-2.0-generate-001',
-         *  prompt: 'A neon hologram of a cat driving at top speed',
+         *  source: {
+         *    prompt: 'A neon hologram of a cat driving at top speed',
+         *  },
          *  config: {
          *    numberOfVideos: 1
          * });
@@ -15258,6 +15378,9 @@ class Models extends BaseModule {
          * ```
          */
         this.generateVideos = async (params) => {
+            if ((params.prompt || params.image || params.video) && params.source) {
+                throw new Error('Source and prompt/image/video are mutually exclusive. Please only use source.');
+            }
             return await this.generateVideosInternal(params);
         };
     }
@@ -15672,23 +15795,7 @@ class Models extends BaseModule {
         }
     }
     /**
-     * Generates an image based on a text description and configuration.
-     *
-     * @param params - The parameters for generating images.
-     * @return The response from the API.
-     *
-     * @example
-     * ```ts
-     * const response = await ai.models.generateImages({
-     *  model: 'imagen-3.0-generate-002',
-     *  prompt: 'Robot holding a red skateboard',
-     *  config: {
-     *    numberOfImages: 1,
-     *    includeRaiReason: true,
-     *  },
-     * });
-     * console.log(response?.generatedImages?.[0]?.image?.imageBytes);
-     * ```
+     * Private method for generating images.
      */
     async generateImagesInternal(params) {
         var _a, _b, _c, _d;
@@ -15760,6 +15867,9 @@ class Models extends BaseModule {
             });
         }
     }
+    /**
+     * Private method for editing an image.
+     */
     async editImageInternal(params) {
         var _a, _b;
         let response;
@@ -15801,6 +15911,9 @@ class Models extends BaseModule {
             throw new Error('This method is only supported by the Vertex AI.');
         }
     }
+    /**
+     * Private method for upscaling an image.
+     */
     async upscaleImageInternal(params) {
         var _a, _b;
         let response;
@@ -16206,10 +16319,16 @@ class Models extends BaseModule {
                 abortSignal: (_b = params.config) === null || _b === void 0 ? void 0 : _b.abortSignal,
             })
                 .then((httpResponse) => {
-                return httpResponse.json();
+                return httpResponse.json().then((jsonResponse) => {
+                    const response = jsonResponse;
+                    response.sdkHttpResponse = {
+                        headers: httpResponse.headers,
+                    };
+                    return response;
+                });
             });
-            return response.then(() => {
-                const resp = deleteModelResponseFromVertex();
+            return response.then((apiResponse) => {
+                const resp = deleteModelResponseFromVertex(apiResponse);
                 const typedResp = new DeleteModelResponse();
                 Object.assign(typedResp, resp);
                 return typedResp;
@@ -16232,10 +16351,16 @@ class Models extends BaseModule {
                 abortSignal: (_d = params.config) === null || _d === void 0 ? void 0 : _d.abortSignal,
             })
                 .then((httpResponse) => {
-                return httpResponse.json();
+                return httpResponse.json().then((jsonResponse) => {
+                    const response = jsonResponse;
+                    response.sdkHttpResponse = {
+                        headers: httpResponse.headers,
+                    };
+                    return response;
+                });
             });
-            return response.then(() => {
-                const resp = deleteModelResponseFromMldev();
+            return response.then((apiResponse) => {
+                const resp = deleteModelResponseFromMldev(apiResponse);
                 const typedResp = new DeleteModelResponse();
                 Object.assign(typedResp, resp);
                 return typedResp;
@@ -16388,27 +16513,7 @@ class Models extends BaseModule {
         }
     }
     /**
-     *  Generates videos based on a text description and configuration.
-     *
-     * @param params - The parameters for generating videos.
-     * @return A Promise<GenerateVideosOperation> which allows you to track the progress and eventually retrieve the generated videos using the operations.get method.
-     *
-     * @example
-     * ```ts
-     * const operation = await ai.models.generateVideos({
-     *  model: 'veo-2.0-generate-001',
-     *  prompt: 'A neon hologram of a cat driving at top speed',
-     *  config: {
-     *    numberOfVideos: 1
-     * });
-     *
-     * while (!operation.done) {
-     *   await new Promise(resolve => setTimeout(resolve, 10000));
-     *   operation = await ai.operations.getVideosOperation({operation: operation});
-     * }
-     *
-     * console.log(operation.response?.generatedVideos?.[0]?.video?.uri);
-     * ```
+     * Private method for generating videos.
      */
     async generateVideosInternal(params) {
         var _a, _b, _c, _d;
@@ -16490,13 +16595,17 @@ function getOperationParametersToMldev(fromObject) {
     }
     return toObject;
 }
-function getOperationParametersToVertex(fromObject) {
+function fetchPredictOperationParametersToVertex(fromObject) {
     const toObject = {};
     const fromOperationName = getValueByPath(fromObject, [
         'operationName',
     ]);
     if (fromOperationName != null) {
-        setValueByPath(toObject, ['_url', 'operationName'], fromOperationName);
+        setValueByPath(toObject, ['operationName'], fromOperationName);
+    }
+    const fromResourceName = getValueByPath(fromObject, ['resourceName']);
+    if (fromResourceName != null) {
+        setValueByPath(toObject, ['_url', 'resourceName'], fromResourceName);
     }
     const fromConfig = getValueByPath(fromObject, ['config']);
     if (fromConfig != null) {
@@ -16504,17 +16613,13 @@ function getOperationParametersToVertex(fromObject) {
     }
     return toObject;
 }
-function fetchPredictOperationParametersToVertex(fromObject) {
+function getOperationParametersToVertex(fromObject) {
     const toObject = {};
     const fromOperationName = getValueByPath(fromObject, [
         'operationName',
     ]);
     if (fromOperationName != null) {
-        setValueByPath(toObject, ['operationName'], fromOperationName);
-    }
-    const fromResourceName = getValueByPath(fromObject, ['resourceName']);
-    if (fromResourceName != null) {
-        setValueByPath(toObject, ['_url', 'resourceName'], fromResourceName);
+        setValueByPath(toObject, ['_url', 'operationName'], fromOperationName);
     }
     const fromConfig = getValueByPath(fromObject, ['config']);
     if (fromConfig != null) {
@@ -17775,6 +17880,18 @@ function listTuningJobsParametersToMldev(fromObject) {
     }
     return toObject;
 }
+function cancelTuningJobParametersToMldev(fromObject) {
+    const toObject = {};
+    const fromName = getValueByPath(fromObject, ['name']);
+    if (fromName != null) {
+        setValueByPath(toObject, ['_url', 'name'], fromName);
+    }
+    const fromConfig = getValueByPath(fromObject, ['config']);
+    if (fromConfig != null) {
+        setValueByPath(toObject, ['config'], fromConfig);
+    }
+    return toObject;
+}
 function tuningExampleToMldev(fromObject) {
     const toObject = {};
     const fromTextInput = getValueByPath(fromObject, ['textInput']);
@@ -17912,6 +18029,18 @@ function listTuningJobsParametersToVertex(fromObject) {
     }
     return toObject;
 }
+function cancelTuningJobParametersToVertex(fromObject) {
+    const toObject = {};
+    const fromName = getValueByPath(fromObject, ['name']);
+    if (fromName != null) {
+        setValueByPath(toObject, ['_url', 'name'], fromName);
+    }
+    const fromConfig = getValueByPath(fromObject, ['config']);
+    if (fromConfig != null) {
+        setValueByPath(toObject, ['config'], fromConfig);
+    }
+    return toObject;
+}
 function tuningDatasetToVertex(fromObject, parentObject) {
     const toObject = {};
     const fromGcsUri = getValueByPath(fromObject, ['gcsUri']);
@@ -18565,6 +18694,54 @@ class Tunings extends BaseModule {
             });
         }
     }
+    /**
+     * Cancels a tuning job.
+     *
+     * @param params - The parameters for the cancel request.
+     * @return The empty response returned by the API.
+     *
+     * @example
+     * ```ts
+     * await ai.tunings.cancel({name: '...'}); // The server-generated resource name.
+     * ```
+     */
+    async cancel(params) {
+        var _a, _b, _c, _d;
+        let path = '';
+        let queryParams = {};
+        if (this.apiClient.isVertexAI()) {
+            const body = cancelTuningJobParametersToVertex(params);
+            path = formatMap('{name}:cancel', body['_url']);
+            queryParams = body['_query'];
+            delete body['config'];
+            delete body['_url'];
+            delete body['_query'];
+            await this.apiClient.request({
+                path: path,
+                queryParams: queryParams,
+                body: JSON.stringify(body),
+                httpMethod: 'POST',
+                httpOptions: (_a = params.config) === null || _a === void 0 ? void 0 : _a.httpOptions,
+                abortSignal: (_b = params.config) === null || _b === void 0 ? void 0 : _b.abortSignal,
+            });
+        }
+        else {
+            const body = cancelTuningJobParametersToMldev(params);
+            path = formatMap('{name}:cancel', body['_url']);
+            queryParams = body['_query'];
+            delete body['config'];
+            delete body['_url'];
+            delete body['_query'];
+            await this.apiClient.request({
+                path: path,
+                queryParams: queryParams,
+                body: JSON.stringify(body),
+                httpMethod: 'POST',
+                httpOptions: (_c = params.config) === null || _c === void 0 ? void 0 : _c.httpOptions,
+                abortSignal: (_d = params.config) === null || _d === void 0 ? void 0 : _d.abortSignal,
+            });
+        }
+    }
     async tuneInternal(params) {
         var _a, _b;
         let response;
@@ -19048,8 +19225,8 @@ function getApiKeyFromEnv() {
     if (envGoogleApiKey && envGeminiApiKey) {
         console.warn('Both GOOGLE_API_KEY and GEMINI_API_KEY are set. Using GOOGLE_API_KEY.');
     }
-    return envGoogleApiKey || envGeminiApiKey;
+    return envGoogleApiKey || envGeminiApiKey || undefined;
 }
 
-export { ActivityHandling, AdapterSize, ApiError, ApiSpec, AuthType, Batches, Behavior, BlockedReason, Caches, Chat, Chats, ComputeTokensResponse, ControlReferenceImage, ControlReferenceType, CountTokensResponse, CreateFileResponse, DeleteCachedContentResponse, DeleteFileResponse, DeleteModelResponse, DynamicRetrievalConfigMode, EditImageResponse, EditMode, EmbedContentResponse, EndSensitivity, Environment, FeatureSelectionPreference, FileSource, FileState, Files, FinishReason, FunctionCallingConfigMode, FunctionResponse, FunctionResponseScheduling, GenerateContentResponse, GenerateContentResponsePromptFeedback, GenerateContentResponseUsageMetadata, GenerateImagesResponse, GenerateVideosOperation, GenerateVideosResponse, GoogleGenAI, HarmBlockMethod, HarmBlockThreshold, HarmCategory, HarmProbability, HarmSeverity, HttpResponse, ImagePromptLanguage, InlinedResponse, JobState, Language, ListBatchJobsResponse, ListCachedContentsResponse, ListFilesResponse, ListModelsResponse, ListTuningJobsResponse, Live, LiveClientToolResponse, LiveMusicPlaybackControl, LiveMusicServerMessage, LiveSendToolResponseParameters, LiveServerMessage, MaskReferenceImage, MaskReferenceMode, MediaModality, MediaResolution, Modality, Mode, Models, MusicGenerationMode, Operations, Outcome, PagedItem, Pager, PersonGeneration, RawReferenceImage, RecontextImageResponse, ReplayResponse, SafetyFilterLevel, Scale, SegmentImageResponse, SegmentMode, Session, StartSensitivity, StyleReferenceImage, SubjectReferenceImage, SubjectReferenceType, Tokens, TrafficType, TuningMode, TurnCoverage, Type, UpscaleImageResponse, UrlRetrievalStatus, VideoCompressionQuality, createModelContent, createPartFromBase64, createPartFromCodeExecutionResult, createPartFromExecutableCode, createPartFromFunctionCall, createPartFromFunctionResponse, createPartFromText, createPartFromUri, createUserContent, mcpToTool, setDefaultBaseUrls };
+export { ActivityHandling, AdapterSize, ApiError, ApiSpec, AuthType, Batches, Behavior, BlockedReason, Caches, Chat, Chats, ComputeTokensResponse, ControlReferenceImage, ControlReferenceType, CountTokensResponse, CreateFileResponse, DeleteCachedContentResponse, DeleteFileResponse, DeleteModelResponse, DynamicRetrievalConfigMode, EditImageResponse, EditMode, EmbedContentResponse, EndSensitivity, Environment, FeatureSelectionPreference, FileSource, FileState, Files, FinishReason, FunctionCallingConfigMode, FunctionResponse, FunctionResponseScheduling, GenerateContentResponse, GenerateContentResponsePromptFeedback, GenerateContentResponseUsageMetadata, GenerateImagesResponse, GenerateVideosOperation, GenerateVideosResponse, GoogleGenAI, HarmBlockMethod, HarmBlockThreshold, HarmCategory, HarmProbability, HarmSeverity, HttpResponse, ImagePromptLanguage, InlinedResponse, JobState, Language, ListBatchJobsResponse, ListCachedContentsResponse, ListFilesResponse, ListModelsResponse, ListTuningJobsResponse, Live, LiveClientToolResponse, LiveMusicPlaybackControl, LiveMusicServerMessage, LiveSendToolResponseParameters, LiveServerMessage, MaskReferenceImage, MaskReferenceMode, MediaModality, MediaResolution, Modality, Mode, Models, MusicGenerationMode, Operations, Outcome, PagedItem, Pager, PersonGeneration, RawReferenceImage, RecontextImageResponse, ReplayResponse, SafetyFilterLevel, Scale, SegmentImageResponse, SegmentMode, Session, StartSensitivity, StyleReferenceImage, SubjectReferenceImage, SubjectReferenceType, Tokens, TrafficType, TuningMode, TurnCoverage, Type, UpscaleImageResponse, UrlRetrievalStatus, VideoCompressionQuality, VideoGenerationReferenceType, createModelContent, createPartFromBase64, createPartFromCodeExecutionResult, createPartFromExecutableCode, createPartFromFunctionCall, createPartFromFunctionResponse, createPartFromText, createPartFromUri, createUserContent, mcpToTool, setDefaultBaseUrls };
 //# sourceMappingURL=index.mjs.map