seacloud-sdk 0.9.8 → 0.9.9

package/dist/cli.js

@@ -601,6 +601,7 @@ Commands:
   llm <prompt>         Chat with LLM models
   agent <prompt>       Chat with Fast Agent (supports image/video generation)
   app <subcommand>     App-related operations (search, generation)
+  status <task-id>     Query task status by task ID
   <model>              Test specific model generation
 
 App Subcommands:
@@ -631,6 +632,10 @@ Model Generation Options:
   --base-url <url>     Base URL (default: http://proxy.sg.seaverse.dev)
   --params <json>      JSON parameters for the model
 
+Status Query Options:
+  --api-key <key>      API key (or set API_SERVICE_TOKEN env var)
+  --base-url <url>     Base URL (default: http://proxy.sg.seaverse.dev)
+
 Examples:
   # Chat with LLM (non-streaming)
   seacloud llm "What is the capital of France?"
@@ -654,6 +659,9 @@ Examples:
   # Create generation task (app generation)
   seacloud app generation --template-id "d26trpte878eqsnm3bjg" --params '[{"field":"prompt1","value":"hello"}]'
 
+  # Query task status
+  seacloud status d123456789abcdef
+
   # Test model generation
   seacloud flux_1_1_pro --params '{"prompt":"a beautiful sunset"}'
 
@@ -952,6 +960,68 @@ async function runAppSearch(templateIdsStr, args) {
   });
   console.log(JSON.stringify(result, null, 2));
 }
+async function runTaskStatus(taskId, args) {
+  const options = {};
+  for (let i = 0; i < args.length; i++) {
+    const arg = args[i];
+    if (arg === "--api-key") options.apiKey = args[++i];
+    else if (arg === "--base-url") options.baseUrl = args[++i];
+  }
+  const apiKey = options.apiKey || process.env.API_SERVICE_TOKEN || "";
+  const baseUrl = options.baseUrl || process.env.SEACLOUD_BASE_URL || "http://proxy.sg.seaverse.dev";
+  const client = new SeacloudClient({ apiKey, baseUrl });
+  console.log(`Querying task status...`);
+  console.log(`Task ID: ${taskId}`);
+  console.log(`Base URL: ${baseUrl}
+`);
+  try {
+    const result = await client.getTaskStatus("/model/v1/generation", taskId);
+    console.log(`Status: ${result.status}`);
+    console.log(`Task ID: ${result.id}
+`);
+    if (result.status === "completed") {
+      console.log("\u2705 Task completed successfully!\n");
+      console.log("Output:");
+      console.log(JSON.stringify(result.output, null, 2));
+      if (result.output) {
+        const urls = [];
+        for (const item of result.output) {
+          if (item.content) {
+            for (const resource of item.content) {
+              if (resource.url) {
+                urls.push(resource.url);
+              }
+            }
+          }
+        }
+        if (urls.length > 0) {
+          console.log("\nGenerated URLs:");
+          urls.forEach((url, i) => {
+            console.log(`  ${i + 1}. ${url}`);
+          });
+        }
+      }
+    } else if (result.status === "failed") {
+      console.log("\u274C Task failed!\n");
+      console.log("Error:", JSON.stringify(result.error, null, 2));
+    } else if (result.status === "processing") {
+      console.log("\u23F3 Task is still processing...");
+      console.log("Please check again later.");
+    } else if (result.status === "pending") {
+      console.log("\u23F3 Task is pending...");
+      console.log("Please check again later.");
+    } else {
+      console.log("Full result:");
+      console.log(JSON.stringify(result, null, 2));
+    }
+  } catch (error) {
+    console.error("\nError querying task status:", error.message);
+    if (error.statusCode) {
+      console.error("HTTP Status Code:", error.statusCode);
+    }
+    process.exit(1);
+  }
+}
 async function main() {
   const args = process.argv.slice(2);
   if (args.length === 0 || args[0] === "--help" || args[0] === "-h") {
@@ -974,6 +1044,13 @@ async function main() {
         process.exit(1);
       }
       await runAgent(args[1], args.slice(2));
+    } else if (command === "status") {
+      if (args.length < 2) {
+        console.error("Error: task ID required for status command");
+        console.log("Usage: seacloud status <task-id> [options]");
+        process.exit(1);
+      }
+      await runTaskStatus(args[1], args.slice(2));
     } else if (command === "app") {
       const subcommand = args[1];
       if (!subcommand) {