koishi-plugin-aka-ai-generator 0.7.3 → 0.7.4

package/lib/index.js

@@ -984,41 +984,18 @@ var YunwuVideoProvider = class {
   async queryTaskStatus(taskId) {
     const { logger, ctx } = this.config;
     try {
-      const possibleEndpoints = [
-        `/v1/video/${taskId}`,
-        `/v1/video/status/${taskId}`,
-        `/v1/videos/generations/${taskId}`
-      ];
-      let response = null;
-      let lastError = null;
-      for (const endpoint of possibleEndpoints) {
-        try {
-          response = await ctx.http.get(
-            `${this.config.apiBase}${endpoint}`,
-            {
-              headers: {
-                "Authorization": `Bearer ${this.config.apiKey}`,
-                "Accept": "application/json"
-              },
-              timeout: this.config.apiTimeout * 1e3
-            }
-          );
-          if (response && !response.error) {
-            logger?.debug("查询任务状态成功", { taskId, endpoint });
-            break;
-          }
-        } catch (err) {
-          lastError = err;
-          if (err.response?.status === 404 && possibleEndpoints.indexOf(endpoint) < possibleEndpoints.length - 1) {
-            logger?.debug("尝试下一个查询端点", { taskId, endpoint, nextEndpoint: possibleEndpoints[possibleEndpoints.indexOf(endpoint) + 1] });
-            continue;
-          }
-          throw err;
+      const endpoint = `/v1/video/${taskId}`;
+      logger?.debug("查询任务状态", { taskId, endpoint });
+      const response = await ctx.http.get(
+        `${this.config.apiBase}${endpoint}`,
+        {
+          headers: {
+            "Authorization": `Bearer ${this.config.apiKey}`,
+            "Accept": "application/json"
+          },
+          timeout: this.config.apiTimeout * 1e3
         }
-      }
-      if (!response) {
-        throw lastError || new Error("所有查询端点都失败");
-      }
+      );
       const status = response.status || response.data?.status || "pending";
       const videoUrl = response.video_url || response.url || response.data?.video_url || response.data?.url;
       return {
@@ -1073,6 +1050,8 @@ var YunwuVideoProvider = class {
     try {
       logger?.info("开始生成视频", { prompt, imageUrl, options });
       const taskId = await this.createVideoTask(prompt, imageUrl, options);
+      logger?.debug("任务已创建，等待 3 秒后开始查询", { taskId });
+      await new Promise((resolve) => setTimeout(resolve, 3e3));
       const videoUrl = await this.pollTaskCompletion(taskId, maxWaitTime);
       logger?.info("视频生成完成", { taskId, videoUrl });
       return videoUrl;

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-aka-ai-generator",
   "description": "自用AI生成插件(GPTGod & Yunwu)",
-  "version": "0.7.3",
+  "version": "0.7.4",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [