feat: Add experimental generate_video support

PiperOrigin-RevId: 741253145

Author: IvanLH

api-report/genai-node.api.md

@@ -734,6 +734,10 @@ export class GoogleGenAI {
     readonly live: Live;
     // (undocumented)
     readonly models: Models;
+    // Warning: (ae-forgotten-export) The symbol "Operations" needs to be exported by the entry point index.d.ts
+    //
+    // (undocumented)
+    readonly operations: Operations;
     // (undocumented)
     readonly vertexai: boolean;
 }
@@ -1185,6 +1189,12 @@ export class Models extends BaseModule {
     generateVideos(params: types.GenerateVideosParameters): Promise<types.GenerateVideosOperation>;
 }
 
+// @public
+export interface OperationGetParameters {
+    config?: GetOperationConfig;
+    operation: GenerateVideosOperation;
+}
+
 // @public
 export enum Outcome {
     // (undocumented)

api-report/genai-web.api.md

@@ -734,6 +734,10 @@ export class GoogleGenAI {
     readonly live: Live;
     // (undocumented)
     readonly models: Models;
+    // Warning: (ae-forgotten-export) The symbol "Operations" needs to be exported by the entry point index.d.ts
+    //
+    // (undocumented)
+    readonly operations: Operations;
     // (undocumented)
     readonly vertexai: boolean;
 }
@@ -1185,6 +1189,12 @@ export class Models extends BaseModule {
     generateVideos(params: types.GenerateVideosParameters): Promise<types.GenerateVideosOperation>;
 }
 
+// @public
+export interface OperationGetParameters {
+    config?: GetOperationConfig;
+    operation: GenerateVideosOperation;
+}
+
 // @public
 export enum Outcome {
     // (undocumented)

api-report/genai.api.md

@@ -734,6 +734,10 @@ export class GoogleGenAI {
     readonly live: Live;
     // (undocumented)
     readonly models: Models;
+    // Warning: (ae-forgotten-export) The symbol "Operations" needs to be exported by the entry point index.d.ts
+    //
+    // (undocumented)
+    readonly operations: Operations;
     // (undocumented)
     readonly vertexai: boolean;
 }
@@ -1185,6 +1189,12 @@ export class Models extends BaseModule {
     generateVideos(params: types.GenerateVideosParameters): Promise<types.GenerateVideosOperation>;
 }
 
+// @public
+export interface OperationGetParameters {
+    config?: GetOperationConfig;
+    operation: GenerateVideosOperation;
+}
+
 // @public
 export enum Outcome {
     // (undocumented)

src/client.ts

@@ -15,6 +15,7 @@ import {CrossWebSocketFactory} from './cross/_cross_websocket';
 import {Files} from './files';
 import {Live} from './live';
 import {Models} from './models';
+import {Operations} from './operations';
 import {HttpOptions} from './types';
 import {WebAuth} from './web/_web_auth';
 
@@ -122,6 +123,7 @@ export class GoogleGenAI {
   readonly chats: Chats;
   readonly caches: Caches;
   readonly files: Files;
+  readonly operations: Operations;
 
   constructor(options: GoogleGenAIOptions) {
     if (options.apiKey == null) {
@@ -147,5 +149,6 @@ export class GoogleGenAI {
     this.chats = new Chats(this.models, this.apiClient);
     this.caches = new Caches(this.apiClient);
     this.files = new Files(this.apiClient);
+    this.operations = new Operations(this.apiClient);
   }
 }

src/node/node_client.ts

@@ -15,6 +15,7 @@ import {Live} from '../live';
 import {Models} from '../models';
 import {NodeAuth} from '../node/_node_auth';
 import {NodeWebSocketFactory} from '../node/_node_websocket';
+import {Operations} from '../operations';
 
 import {NodeUploader} from './_node_uploader';
 
@@ -73,6 +74,7 @@ export class GoogleGenAI {
   readonly chats: Chats;
   readonly caches: Caches;
   readonly files: Files;
+  readonly operations: Operations;
 
   constructor(options: GoogleGenAIOptions) {
     // Validate explicitly set initializer values.
@@ -141,6 +143,7 @@ export class GoogleGenAI {
     this.chats = new Chats(this.models, this.apiClient);
     this.caches = new Caches(this.apiClient);
     this.files = new Files(this.apiClient);
+    this.operations = new Operations(this.apiClient);
   }
 }
 

src/operations.ts

@@ -17,6 +17,43 @@ export class Operations extends BaseModule {
     super();
   }
 
+  /**
+   * Gets the status of a long-running operation.
+   *
+   * @param operation The Operation object returned by a previous API call.
+   * @return The updated Operation object, with the latest status or result.
+   */
+  async get(
+    parameters: types.OperationGetParameters,
+  ): Promise<types.GenerateVideosOperation> {
+    const operation = parameters.operation;
+    const config = parameters.config;
+
+    if (operation.name === undefined || operation.name === '') {
+      throw new Error('Operation name is required.');
+    }
+
+    if (this.apiClient.isVertexAI()) {
+      const resourceName = operation.name.split('/operations/')[0];
+      var httpOptions: types.HttpOptions | undefined = undefined;
+
+      if (config && 'httpOptions' in config) {
+        httpOptions = config.httpOptions;
+      }
+
+      return this.fetchPredictVideosOperationInternal({
+        operationName: operation.name,
+        resourceName: resourceName,
+        config: {httpOptions: httpOptions},
+      });
+    } else {
+      return this.getVideosOperationInternal({
+        operationName: operation.name,
+        config: config,
+      });
+    }
+  }
+
   private async getVideosOperationInternal(
     params: types.GetOperationParameters,
   ): Promise<types.GenerateVideosOperation> {

src/types.ts

@@ -2458,6 +2458,14 @@ export class LiveSendToolResponseParameters {
   functionResponses: FunctionResponse[] | FunctionResponse = [];
 }
 
+/** Parameters for the get method of the operations module. */
+export declare interface OperationGetParameters {
+  /** The operation to be retrieved. */
+  operation: GenerateVideosOperation;
+  /** Used to override the default configuration. */
+  config?: GetOperationConfig;
+}
+
 export type PartUnion = Part | string;
 
 export type PartListUnion = PartUnion[] | PartUnion;

src/web/web_client.ts

@@ -11,6 +11,7 @@ import {GoogleGenAIOptions} from '../client';
 import {Files} from '../files';
 import {Live} from '../live';
 import {Models} from '../models';
+import {Operations} from '../operations';
 
 import {BrowserUploader} from './_browser_uploader';
 import {BrowserWebSocketFactory} from './_browser_websocket';
@@ -64,6 +65,7 @@ export class GoogleGenAI {
   readonly chats: Chats;
   readonly caches: Caches;
   readonly files: Files;
+  readonly operations: Operations;
 
   constructor(options: GoogleGenAIOptions) {
     if (options.apiKey == null) {
@@ -94,5 +96,6 @@ export class GoogleGenAI {
     this.chats = new Chats(this.models, this.apiClient);
     this.caches = new Caches(this.apiClient);
     this.files = new Files(this.apiClient);
+    this.operations = new Operations(this.apiClient);
   }
 }