feat: Add Veo 2 generate_videos support in Go SDK

google/genai/operations.py

@@ -319,7 +319,7 @@ def _GenerateVideosOperation_from_vertex(
 
 class Operations(_api_module.BaseModule):
 
-  def _get_operation(
+  def _get_videos_operation(
       self,
       *,
       operation_name: str,
@@ -385,7 +385,7 @@ def _get_operation(
     self._api_client._verify_response(return_value)
     return return_value
 
-  def _fetch_predict_operation(
+  def _fetch_predict_videos_operation(
       self,
       *,
       operation_name: str,
@@ -481,21 +481,21 @@ def get(
       fetch_operation_config = types.FetchPredictOperationConfig(
           http_options=http_options
       )
-      return self._fetch_predict_operation(
+      return self._fetch_predict_videos_operation(
           operation_name=operation_name,
           resource_name=resource_name,
           config=fetch_operation_config,
       )
     else:
-      return self._get_operation(
+      return self._get_videos_operation(
           operation_name=operation_name,
           config=config,
       )
 
 
 class AsyncOperations(_api_module.BaseModule):
 
-  async def _get_operation(
+  async def _get_videos_operation(
       self,
       *,
       operation_name: str,
@@ -561,7 +561,7 @@ async def _get_operation(
     self._api_client._verify_response(return_value)
     return return_value
 
-  async def _fetch_predict_operation(
+  async def _fetch_predict_videos_operation(
       self,
       *,
       operation_name: str,
@@ -655,13 +655,13 @@ async def get(
       fetch_operation_config = types.FetchPredictOperationConfig(
           http_options=http_options
       )
-      return await self._fetch_predict_operation(
+      return await self._fetch_predict_videos_operation(
           operation_name=operation_name,
           resource_name=resource_name,
           config=fetch_operation_config,
       )
     else:
-      return await self._get_operation(
+      return await self._get_videos_operation(
           operation_name=operation_name,
           config=config,
       )

google/genai/tests/models/test_generate_videos.py

@@ -68,7 +68,7 @@
             ),
         ),
         exception_if_mldev=(
-            "output_gcs_uri parameter is not supported in Gemini API"
+            "not supported in Gemini API"
         ),
     ),
     pytest_helper.TestTableItem(

google/genai/types.py

@@ -5433,14 +5433,7 @@ class GenerateVideosResponseDict(TypedDict, total=False):
 
 
 class GenerateVideosOperation(_common.BaseModel):
-  """A video generation operation.
-
-  Use the following code to refresh the operation:
-
-  ```
-  operation = client.operations.get(operation)
-  ```
-  """
+  """A video generation operation."""
 
   name: Optional[str] = Field(
       default=None,
@@ -5468,14 +5461,7 @@ class GenerateVideosOperation(_common.BaseModel):
 
 
 class GenerateVideosOperationDict(TypedDict, total=False):
-  """A video generation operation.
-
-  Use the following code to refresh the operation:
-
-  ```
-  operation = client.operations.get(operation)
-  ```
-  """
+  """A video generation operation."""
 
   name: Optional[str]
   """The server-assigned name, which is only unique within the same service that originally returns it. If you use the default HTTP mapping, the `name` should be a resource name ending with `operations/{unique_id}`."""