@ai-sdk/xai 4.0.56 → 4.0.57

package/CHANGELOG.md

@@ -1,5 +1,14 @@
 # @ai-sdk/xai
 
+## 4.0.57
+
+### Patch Changes
+
+- 4b8c4fa: feat(xai): add batch cancellation and listing
+- Updated dependencies [9942196]
+  - @ai-sdk/provider@4.0.13
+  - @ai-sdk/provider-utils@5.0.39
+
 ## 4.0.56
 
 ### Patch Changes

package/dist/index.js

@@ -3531,24 +3531,23 @@ var xaiBatchProviderOptionsSchema = lazySchema7(
     })
   )
 );
+var xaiBatchResponseZodSchema = () => z15.object({
+  batch_id: z15.string(),
+  name: z15.string().nullish(),
+  create_time: z15.string().nullish(),
+  expire_time: z15.string().nullish(),
+  cancel_time: z15.string().nullish(),
+  cancel_by_xai_message: z15.string().nullish(),
+  state: z15.object({
+    num_requests: z15.number().nullish(),
+    num_pending: z15.number().nullish(),
+    num_success: z15.number().nullish(),
+    num_error: z15.number().nullish(),
+    num_cancelled: z15.number().nullish()
+  }).nullish()
+});
 var xaiBatchResponseSchema = lazySchema7(
-  () => zodSchema7(
-    z15.object({
-      batch_id: z15.string(),
-      name: z15.string().nullish(),
-      create_time: z15.string().nullish(),
-      expire_time: z15.string().nullish(),
-      cancel_time: z15.string().nullish(),
-      cancel_by_xai_message: z15.string().nullish(),
-      state: z15.object({
-        num_requests: z15.number().nullish(),
-        num_pending: z15.number().nullish(),
-        num_success: z15.number().nullish(),
-        num_error: z15.number().nullish(),
-        num_cancelled: z15.number().nullish()
-      }).nullish()
-    })
-  )
+  () => zodSchema7(xaiBatchResponseZodSchema())
 );
 var xaiBatchErrorSchema = z15.object({
   code: z15.union([z15.string(), z15.number()]).nullish(),
@@ -3572,6 +3571,14 @@ var xaiBatchResultsPageSchema = lazySchema7(
     })
   )
 );
+var xaiBatchListResponseSchema = lazySchema7(
+  () => zodSchema7(
+    z15.object({
+      batches: z15.array(xaiBatchResponseZodSchema()),
+      pagination_token: z15.string().nullish()
+    })
+  )
+);
 var XaiBatch = class {
   constructor(options) {
     this.options = options;
@@ -3670,6 +3677,51 @@ var XaiBatch = class {
   async doGetBatchStatus(options) {
     return convertXaiBatchStatus(await this.retrieveBatch(options));
   }
+  async doCancelBatch(options) {
+    var _a, _b;
+    await postJsonToApi4({
+      url: this.getUrl(
+        `/batches/${encodeURIComponent(options.batchId)}:cancel`
+      ),
+      headers: combineHeaders4((_b = (_a = this.options.config).headers) == null ? void 0 : _b.call(_a), options.headers),
+      body: {},
+      failedResponseHandler: xaiFailedResponseHandler,
+      successfulResponseHandler: createJsonResponseHandler4(
+        xaiBatchResponseSchema
+      ),
+      abortSignal: options.abortSignal,
+      fetch: this.options.config.fetch
+    });
+    return {};
+  }
+  async doListBatches(options) {
+    var _a, _b;
+    const url = new URL(this.getUrl("/batches"));
+    if (options.limit != null) {
+      url.searchParams.set("limit", String(options.limit));
+    }
+    if (options.cursor != null) {
+      url.searchParams.set("pagination_token", options.cursor);
+    }
+    const { value: page } = await getFromApi2({
+      url: url.toString(),
+      headers: combineHeaders4((_b = (_a = this.options.config).headers) == null ? void 0 : _b.call(_a), options.headers),
+      failedResponseHandler: xaiFailedResponseHandler,
+      successfulResponseHandler: createJsonResponseHandler4(
+        xaiBatchListResponseSchema
+      ),
+      abortSignal: options.abortSignal,
+      fetch: this.options.config.fetch,
+      validateUrl: false
+    });
+    return {
+      batches: page.batches.map((batch) => ({
+        batchId: batch.batch_id,
+        ...convertXaiBatchStatus(batch)
+      })),
+      ...page.pagination_token != null ? { nextCursor: page.pagination_token } : {}
+    };
+  }
   async doGetBatchResults(options) {
     const batch = await this.retrieveBatch(options);
     if (convertXaiBatchStatus(batch).status === "pending") {
@@ -4358,7 +4410,7 @@ var xaiTools = {
 };
 
 // src/version.ts
-var VERSION = true ? "4.0.56" : "0.0.0-test";
+var VERSION = true ? "4.0.57" : "0.0.0-test";
 
 // src/files/xai-files.ts
 import {