@ai-sdk/google-vertex 5.0.38 → 5.0.39

package/dist/index.js

@@ -33,7 +33,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "5.0.38" : "0.0.0-test";
+var VERSION = true ? "5.0.39" : "0.0.0-test";
 
 // src/google-vertex-embedding-model.ts
 import {
@@ -855,7 +855,6 @@ import {
   combineHeaders as combineHeaders4,
   convertUint8ArrayToBase64 as convertUint8ArrayToBase643,
   createJsonResponseHandler as createJsonResponseHandler4,
-  delay,
   parseProviderOptions as parseProviderOptions4,
   postJsonToApi as postJsonToApi4,
   resolve as resolve4
@@ -940,15 +939,14 @@ var GoogleVertexVideoModel = class {
   get maxVideosPerCall() {
     return 4;
   }
-  async doGenerate(options) {
-    var _a, _b, _c, _d, _e, _f, _g, _h;
-    const currentDate = (_c = (_b = (_a = this.config._internal) == null ? void 0 : _a.currentDate) == null ? void 0 : _b.call(_a)) != null ? _c : /* @__PURE__ */ new Date();
+  async buildRequest(options) {
+    var _a, _b;
     const warnings = [];
-    const googleVertexOptions = (_d = await parseProviderOptions4({
+    const googleVertexOptions = (_a = await parseProviderOptions4({
       provider: "googleVertex",
       providerOptions: options.providerOptions,
       schema: googleVertexVideoModelOptionsSchema
-    })) != null ? _d : await parseProviderOptions4({
+    })) != null ? _a : await parseProviderOptions4({
       provider: "vertex",
       providerOptions: options.providerOptions,
       schema: googleVertexVideoModelOptionsSchema
@@ -1001,7 +999,7 @@ var GoogleVertexVideoModel = class {
     if (options.seed) {
       parameters.seed = options.seed;
     }
-    const generateAudio = (_e = options.generateAudio) != null ? _e : googleVertexOptions == null ? void 0 : googleVertexOptions.generateAudio;
+    const generateAudio = (_b = options.generateAudio) != null ? _b : googleVertexOptions == null ? void 0 : googleVertexOptions.generateAudio;
     if (generateAudio != null) {
       parameters.generateAudio = generateAudio;
     }
@@ -1030,74 +1028,14 @@ var GoogleVertexVideoModel = class {
         }
       }
     }
-    const { value: operation } = await postJsonToApi4({
-      url: `${this.config.baseURL}/models/${this.modelId}:predictLongRunning`,
-      headers: combineHeaders4(
-        await resolve4(this.config.headers),
-        options.headers
-      ),
-      body: {
-        instances,
-        parameters
-      },
-      successfulResponseHandler: createJsonResponseHandler4(
-        googleVertexOperationSchema
-      ),
-      failedResponseHandler: googleVertexFailedResponseHandler,
-      abortSignal: options.abortSignal,
-      fetch: this.config.fetch
-    });
-    const operationName = operation.name;
-    if (!operationName) {
-      throw new AISDKError({
-        name: "VERTEX_VIDEO_GENERATION_ERROR",
-        message: "No operation name returned from API"
-      });
-    }
-    const pollIntervalMs = (_f = googleVertexOptions == null ? void 0 : googleVertexOptions.pollIntervalMs) != null ? _f : 1e4;
-    const pollTimeoutMs = (_g = googleVertexOptions == null ? void 0 : googleVertexOptions.pollTimeoutMs) != null ? _g : 6e5;
-    const startTime = Date.now();
-    let finalOperation = operation;
-    let responseHeaders;
-    while (!finalOperation.done) {
-      if (Date.now() - startTime > pollTimeoutMs) {
-        throw new AISDKError({
-          name: "VERTEX_VIDEO_GENERATION_TIMEOUT",
-          message: `Video generation timed out after ${pollTimeoutMs}ms`
-        });
-      }
-      await delay(pollIntervalMs);
-      if ((_h = options.abortSignal) == null ? void 0 : _h.aborted) {
-        throw new AISDKError({
-          name: "VERTEX_VIDEO_GENERATION_ABORTED",
-          message: "Video generation request was aborted"
-        });
-      }
-      const { value: statusOperation, responseHeaders: pollHeaders } = await postJsonToApi4({
-        url: `${this.config.baseURL}/models/${this.modelId}:fetchPredictOperation`,
-        headers: combineHeaders4(
-          await resolve4(this.config.headers),
-          options.headers
-        ),
-        body: {
-          operationName
-        },
-        successfulResponseHandler: createJsonResponseHandler4(
-          googleVertexOperationSchema
-        ),
-        failedResponseHandler: googleVertexFailedResponseHandler,
-        abortSignal: options.abortSignal,
-        fetch: this.config.fetch
-      });
-      finalOperation = statusOperation;
-      responseHeaders = pollHeaders;
-    }
-    if (finalOperation.error) {
-      throw new AISDKError({
-        name: "VERTEX_VIDEO_GENERATION_FAILED",
-        message: `Video generation failed: ${finalOperation.error.message}`
-      });
-    }
+    return { instances, parameters, warnings, googleVertexOptions };
+  }
+  buildCompletedResult({
+    finalOperation,
+    responseHeaders,
+    warnings,
+    currentDate
+  }) {
     const response = finalOperation.response;
     if (!(response == null ? void 0 : response.videos) || response.videos.length === 0) {
       throw new AISDKError({
@@ -1136,6 +1074,7 @@ var GoogleVertexVideoModel = class {
       });
     }
     return {
+      status: "completed",
       videos,
       warnings,
       response: {
@@ -1154,6 +1093,92 @@ var GoogleVertexVideoModel = class {
       })()
     };
   }
+  async doStart(options) {
+    var _a, _b, _c;
+    const currentDate = (_c = (_b = (_a = this.config._internal) == null ? void 0 : _a.currentDate) == null ? void 0 : _b.call(_a)) != null ? _c : /* @__PURE__ */ new Date();
+    const { instances, parameters, warnings } = await this.buildRequest(options);
+    const { value: operation, responseHeaders } = await postJsonToApi4({
+      url: `${this.config.baseURL}/models/${this.modelId}:predictLongRunning`,
+      headers: combineHeaders4(
+        await resolve4(this.config.headers),
+        options.headers
+      ),
+      body: {
+        instances,
+        parameters
+      },
+      successfulResponseHandler: createJsonResponseHandler4(
+        googleVertexOperationSchema
+      ),
+      failedResponseHandler: googleVertexFailedResponseHandler,
+      abortSignal: options.abortSignal,
+      fetch: this.config.fetch
+    });
+    const operationName = operation.name;
+    if (!operationName) {
+      throw new AISDKError({
+        name: "VERTEX_VIDEO_GENERATION_ERROR",
+        message: "No operation name returned from API"
+      });
+    }
+    return {
+      operation: { operationName },
+      warnings,
+      response: {
+        timestamp: currentDate,
+        modelId: this.modelId,
+        headers: responseHeaders
+      }
+    };
+  }
+  async doStatus(options) {
+    var _a, _b, _c;
+    const currentDate = (_c = (_b = (_a = this.config._internal) == null ? void 0 : _a.currentDate) == null ? void 0 : _b.call(_a)) != null ? _c : /* @__PURE__ */ new Date();
+    const { operationName } = options.operation;
+    const { value: statusOperation, responseHeaders } = await postJsonToApi4({
+      url: `${this.config.baseURL}/models/${this.modelId}:fetchPredictOperation`,
+      headers: combineHeaders4(
+        await resolve4(this.config.headers),
+        options.headers
+      ),
+      body: {
+        operationName
+      },
+      successfulResponseHandler: createJsonResponseHandler4(
+        googleVertexOperationSchema
+      ),
+      failedResponseHandler: googleVertexFailedResponseHandler,
+      abortSignal: options.abortSignal,
+      fetch: this.config.fetch
+    });
+    if (!statusOperation.done) {
+      return {
+        status: "pending",
+        response: {
+          timestamp: currentDate,
+          modelId: this.modelId,
+          headers: responseHeaders
+        }
+      };
+    }
+    if (statusOperation.error) {
+      return {
+        status: "error",
+        error: `Video generation failed: ${statusOperation.error.message}`,
+        response: {
+          timestamp: currentDate,
+          modelId: this.modelId,
+          headers: responseHeaders
+        }
+      };
+    }
+    return this.buildCompletedResult({
+      finalOperation: statusOperation,
+      responseHeaders,
+      warnings: [],
+      currentDate
+    });
+  }
 };
 var googleVertexOperationSchema = z9.object({
   name: z9.string().nullish(),