@ai-sdk/google 4.0.66 → 4.0.67

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ai-sdk/google",
-  "version": "4.0.66",
+  "version": "4.0.67",
   "type": "module",
   "license": "Apache-2.0",
   "sideEffects": false,
@@ -35,8 +35,8 @@
     }
   },
   "dependencies": {
-    "@ai-sdk/provider": "4.0.12",
-    "@ai-sdk/provider-utils": "5.0.38"
+    "@ai-sdk/provider": "4.0.13",
+    "@ai-sdk/provider-utils": "5.0.39"
   },
   "devDependencies": {
     "@ai-sdk/test-server": "2.0.1",

package/src/google-batch.ts

@@ -2,8 +2,11 @@ import {
   InvalidArgumentError,
   InvalidResponseDataError,
   type Experimental_BatchV4 as BatchV4,
+  type Experimental_BatchV4CancelResult as BatchV4CancelResult,
   type Experimental_BatchV4Error as BatchV4Error,
   type Experimental_BatchV4ItemResult as BatchV4ItemResult,
+  type Experimental_BatchV4ListOptions as BatchV4ListOptions,
+  type Experimental_BatchV4ListResult as BatchV4ListResult,
   type Experimental_BatchV4OperationOptions as BatchV4OperationOptions,
   type Experimental_BatchV4StartResult as BatchV4StartResult,
   type Experimental_BatchV4Status as BatchV4Status,
@@ -76,26 +79,40 @@ const googleBatchOutputSchema = z.object({
     .nullish(),
 });
 
+const googleBatchOperationZodSchema = () =>
+  z.object({
+    name: z.string(),
+    metadata: z
+      .object({
+        state: z.string().nullish(),
+        createTime: z.string().nullish(),
+        batchStats: googleBatchStatsSchema.nullish(),
+        output: googleBatchOutputSchema.nullish(),
+      })
+      .nullish(),
+    done: z.boolean().nullish(),
+    error: googleRpcStatusSchema.nullish(),
+    response: googleBatchOutputSchema.nullish(),
+  });
+
 const googleBatchOperationSchema = lazySchema(() =>
+  zodSchema(googleBatchOperationZodSchema()),
+);
+
+type GoogleBatchOperation = InferSchema<typeof googleBatchOperationSchema>;
+
+const googleBatchListResponseSchema = lazySchema(() =>
   zodSchema(
     z.object({
-      name: z.string(),
-      metadata: z
-        .object({
-          state: z.string().nullish(),
-          createTime: z.string().nullish(),
-          batchStats: googleBatchStatsSchema.nullish(),
-          output: googleBatchOutputSchema.nullish(),
-        })
-        .nullish(),
-      done: z.boolean().nullish(),
-      error: googleRpcStatusSchema.nullish(),
-      response: googleBatchOutputSchema.nullish(),
+      operations: z.array(googleBatchOperationZodSchema()).nullish(),
+      nextPageToken: z.string().nullish(),
     }),
   ),
 );
 
-type GoogleBatchOperation = InferSchema<typeof googleBatchOperationSchema>;
+const googleBatchCancelResponseSchema = lazySchema(() =>
+  zodSchema(z.object({})),
+);
 
 const googleFileUploadResponseSchema = lazySchema(() =>
   zodSchema(
@@ -355,6 +372,54 @@ export class GoogleBatch implements BatchV4<{ readonly text: GoogleModelId }> {
     return convertGoogleBatchStatus(await this.retrieveBatch(options));
   }
 
+  async doCancelBatch(
+    options: BatchV4OperationOptions,
+  ): Promise<BatchV4CancelResult> {
+    await postJsonToApi({
+      url: `${this.batchConfig.baseURL}/${options.batchId}:cancel`,
+      headers: await this.getHeaders(options.headers),
+      body: {},
+      failedResponseHandler: googleFailedResponseHandler,
+      successfulResponseHandler: createJsonResponseHandler(
+        googleBatchCancelResponseSchema,
+      ),
+      abortSignal: options.abortSignal,
+      fetch: this.batchConfig.fetch,
+    });
+
+    return {};
+  }
+
+  async doListBatches(options: BatchV4ListOptions): Promise<BatchV4ListResult> {
+    const url = new URL(`${this.batchConfig.baseURL}/batches`);
+    if (options.limit != null) {
+      url.searchParams.set('pageSize', String(options.limit));
+    }
+    if (options.cursor != null) {
+      url.searchParams.set('pageToken', options.cursor);
+    }
+
+    const { value: page } = await getFromApi({
+      url: url.toString(),
+      headers: await this.getHeaders(options.headers),
+      failedResponseHandler: googleFailedResponseHandler,
+      successfulResponseHandler: createJsonResponseHandler(
+        googleBatchListResponseSchema,
+      ),
+      abortSignal: options.abortSignal,
+      fetch: this.batchConfig.fetch,
+      validateUrl: false,
+    });
+
+    return {
+      batches: (page.operations ?? []).map(operation => ({
+        batchId: operation.name,
+        ...convertGoogleBatchStatus(operation),
+      })),
+      ...(page.nextPageToken != null ? { nextCursor: page.nextPageToken } : {}),
+    };
+  }
+
   async doGetBatchResults(
     options: BatchV4OperationOptions,
   ): Promise<ReadableStream<BatchV4ItemResult>> {