@acosmi/sdk-ts 2.1.0 → 2.2.0

package/dist/index.mjs

@@ -2728,6 +2728,60 @@ var Client = class _Client {
       ctl.dispose();
     }
   }
+  // ===========================================================================
+  // 媒体生成 (v1.3+) — 图片 / 视频生成托管模型 (与 chat 同网关)
+  //
+  // 仅对 capabilities.supports_image_generation / supports_video_generation 的模型有效。
+  // 网关不算钱; 用量由网关上报营销系统结算。
+  // ===========================================================================
+  /** 解析 nexus-v4 {code,message,data} 信封, code!=0 抛 BusinessError, 返回 data。 */
+  unwrapAPIResponse(result) {
+    const env = JSON.parse(new TextDecoder().decode(result));
+    const bizErr = apiResponseBusinessError(env);
+    if (bizErr) throw bizErr;
+    return env.data;
+  }
+  /**
+   * 图片生成 (同步)。POST /managed-models/:id/images/generations
+   *
+   * @param modelID 图片生成托管模型 ID (capabilities.supports_image_generation=true)
+   */
+  async generateImage(modelID, req, signal) {
+    const endpoint = `/managed-models/${encodeURIComponent(modelID)}/images/generations`;
+    const { result } = await this.doJSONFullRaw(
+      "POST",
+      endpoint,
+      req,
+      signal,
+      CHAT_REQUEST_TIMEOUT_MS
+    );
+    return this.unwrapAPIResponse(result);
+  }
+  /**
+   * 创建视频生成任务 (异步)。POST /managed-models/:id/videos/generations
+   * 返回 taskId; 用 pollVideoTask() 轮询直到 status=completed。
+   * 上报真物理量 (视频秒数) 需在 pollVideoTask 时回传 req.duration。
+   *
+   * @param modelID 视频生成托管模型 ID (capabilities.supports_video_generation=true)
+   */
+  async generateVideo(modelID, req, signal) {
+    const endpoint = `/managed-models/${encodeURIComponent(modelID)}/videos/generations`;
+    const { result } = await this.doJSONFullRaw("POST", endpoint, req, signal);
+    return this.unwrapAPIResponse(result);
+  }
+  /**
+   * 轮询视频任务状态。GET /managed-models/:id/videos/tasks/:taskId
+   *
+   * @param durationSeconds 创建时的时长 (秒), 透传给网关在 completed 时上报用量。
+   */
+  async pollVideoTask(modelID, taskID, durationSeconds, signal) {
+    let endpoint = `/managed-models/${encodeURIComponent(modelID)}/videos/tasks/${encodeURIComponent(taskID)}`;
+    if (durationSeconds != null && durationSeconds > 0) {
+      endpoint += `?duration=${encodeURIComponent(String(durationSeconds))}`;
+    }
+    const { result } = await this.doJSONFullRaw("GET", endpoint, null, signal);
+    return this.unwrapAPIResponse(result);
+  }
   /**
    * Anthropic 原生格式同步聊天
    * v0.5.0: 根据 provider 自动路由
@@ -3072,11 +3126,11 @@ var Client = class _Client {
     }
   }
   /** doJSONFull 的 raw bytes 变体 (chat 用, 不立即 JSON.parse) */
-  async doJSONFullRaw(method, path, body, signal) {
-    return this.doJSONFullRawInternal(method, path, body, signal, false);
+  async doJSONFullRaw(method, path, body, signal, timeoutMs = 3e4) {
+    return this.doJSONFullRawInternal(method, path, body, signal, false, timeoutMs);
   }
-  async doJSONFullRawInternal(method, path, body, signal, retried) {
-    const ctl = withRequestTimeout(3e4, signal);
+  async doJSONFullRawInternal(method, path, body, signal, retried, timeoutMs = 3e4) {
+    const ctl = withRequestTimeout(timeoutMs, signal);
     try {
       const token = await this.ensureToken(ctl.signal);
       let bodyStr = null;
@@ -3104,7 +3158,7 @@ var Client = class _Client {
             `unauthorized and refresh failed: ${refreshErr instanceof Error ? refreshErr.message : String(refreshErr)}`
           );
         }
-        return this.doJSONFullRawInternal(method, path, body, signal, true);
+        return this.doJSONFullRawInternal(method, path, body, signal, true, timeoutMs);
       }
       if (resp.status < 200 || resp.status >= 300) {
         const bodyBytes = await readLimited(resp.body, maxErrorBodySize);