seacloud-sdk 0.9.6 → 0.9.7

package/dist/cli.js

@@ -505,6 +505,39 @@ function createTextMessage(role, text) {
     content: [{ type: "text", text }]
   };
 }
+
+// src/api/app_search.ts
+async function appSearch(params) {
+  const client = getClient();
+  const config = client.getConfig();
+  const url = `${config.baseUrl}/model/v1/template/specs`;
+  const controller = new AbortController();
+  const timeoutId = setTimeout(() => controller.abort(), config.timeout);
+  try {
+    const requestId = `req-${Date.now()}-${Math.random().toString(36).substring(7)}`;
+    const response = await config.fetch(url, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        "Authorization": `Bearer ${config.apiKey}`,
+        "X-Request-Id": requestId,
+        "X-Project": "KIIRA"
+      },
+      body: JSON.stringify(params),
+      signal: controller.signal
+    });
+    clearTimeout(timeoutId);
+    if (!response.ok) {
+      const errorBody = await response.text();
+      throw new Error(`HTTP ${response.status}: ${errorBody}`);
+    }
+    const result = await response.json();
+    return result;
+  } catch (error) {
+    clearTimeout(timeoutId);
+    throw error;
+  }
+}
 var __filename$1 = fileURLToPath(import.meta.url);
 dirname(__filename$1);
 function showHelp() {
@@ -517,8 +550,13 @@ Usage:
 Commands:
   llm <prompt>         Chat with LLM models
   agent <prompt>       Chat with Fast Agent (supports image/video generation)
+  app <subcommand>     App-related operations (search, generation)
   <model>              Test specific model generation
 
+App Subcommands:
+  app search <ids>     Query template specifications by template IDs
+  app generation       Create generation task from template
+
 LLM Options:
   --model <name>       Model name (default: seaart-mix-sonnet-4-5)
   --stream             Enable streaming mode
@@ -531,6 +569,13 @@ Agent Options:
   --stream             Enable streaming mode
   --session-id <id>    Session ID for multi-turn conversation
 
+App Search Options:
+  --type <type>        Template type (default: comfyui)
+
+App Generation Options:
+  --template-id <id>   Template ID (required)
+  --params <json>      Input parameters as JSON string (required)
+
 Model Generation Options:
   --api-key <key>      API key (or set API_SERVICE_TOKEN env var)
   --base-url <url>     Base URL (default: http://proxy.sg.seaverse.dev)
@@ -552,6 +597,13 @@ Examples:
   # Chat with Agent (streaming)
   seacloud agent "Create a cat image" --stream
 
+  # Query template specifications (app search)
+  seacloud app search "template-id-1,template-id-2"
+  seacloud app search "template-abc" --type comfyui
+
+  # Create generation task (app generation)
+  seacloud app generation --template-id "d26trpte878eqsnm3bjg" --params '[{"field":"prompt1","value":"hello"}]'
+
   # Test model generation
   seacloud flux_1_1_pro --params '{"prompt":"a beautiful sunset"}'
 
@@ -591,7 +643,7 @@ function parseArgs(args) {
 }
 async function testModel(model, options) {
   const apiKey = options.apiKey || process.env.API_SERVICE_TOKEN || "";
-  const baseUrl = options.baseUrl || process.env.API_BASE_URL || "http://localhost:8080";
+  const baseUrl = options.baseUrl || process.env.SEACLOUD_BASE_URL || "http://proxy.sg.seaverse.dev";
   if (!options.params) {
     console.error("Error: --params required. Provide JSON parameters for the model");
     process.exit(1);
@@ -761,6 +813,95 @@ async function runAgent(prompt, args) {
     console.log("Message ID:", response.msg_id);
   }
 }
+async function runAppGeneration(args) {
+  const options = {};
+  for (let i = 0; i < args.length; i++) {
+    const arg = args[i];
+    if (arg === "--template-id") options.templateId = args[++i];
+    else if (arg === "--params") {
+      try {
+        options.params = JSON.parse(args[++i]);
+      } catch (e) {
+        console.error("Error: --params must be valid JSON array");
+        console.error(`Example: --params '[{"field":"prompt1","value":"hello"}]'`);
+        process.exit(1);
+      }
+    } else if (arg === "--api-key") options.apiKey = args[++i];
+    else if (arg === "--base-url") options.baseUrl = args[++i];
+  }
+  if (!options.templateId) {
+    console.error("Error: --template-id required");
+    console.error("Usage: seacloud app generation --template-id <id> --params <json>");
+    process.exit(1);
+  }
+  if (!options.params) {
+    console.error("Error: --params required");
+    console.error("Usage: seacloud app generation --template-id <id> --params <json>");
+    console.error(`Example: --params '[{"field":"prompt1","value":"hello"}]'`);
+    process.exit(1);
+  }
+  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 });
+  try {
+    console.log("Creating task...");
+    const task = await client.createTask("/model/v1/generation", {
+      model: "comfyui",
+      input: [{ params: { template_id: options.templateId, inputs: options.params } }]
+    });
+    console.log(`Task created: ${task.id}`);
+    console.log(`Initial status: ${task.status}`);
+    console.log("");
+    if (task.status === "failed") {
+      console.error("Task failed immediately:", task.error);
+      process.exit(1);
+    }
+    console.log("Polling for results...");
+    let attempt = 0;
+    const maxAttempts = 120;
+    const intervalMs = 3e3;
+    while (attempt < maxAttempts) {
+      attempt++;
+      const result = await client.getTaskStatus("/model/v1/generation", task.id);
+      process.stdout.write(`\rAttempt ${attempt}/${maxAttempts} - Status: ${result.status}    `);
+      if (result.status === "completed") {
+        console.log("\n\nTask completed!");
+        console.log("\nResults:");
+        console.log(JSON.stringify(result, null, 2));
+        process.exit(0);
+      }
+      if (result.status === "failed") {
+        console.log("\n\nTask failed!");
+        console.error("Error:", result.error);
+        process.exit(1);
+      }
+      await new Promise((resolve) => setTimeout(resolve, intervalMs));
+    }
+    console.log("\n\nTimeout: Task did not complete within the time limit");
+    process.exit(1);
+  } catch (error) {
+    console.error("\nError:", error.message);
+    process.exit(1);
+  }
+}
+async function runAppSearch(templateIdsStr, args) {
+  const options = { type: "comfyui" };
+  for (let i = 0; i < args.length; i++) {
+    const arg = args[i];
+    if (arg === "--type") options.type = args[++i];
+    else 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";
+  initSeacloud(apiKey, { baseUrl });
+  const templateIds = templateIdsStr.split(",").map((id) => id.trim());
+  const result = await appSearch({
+    type: options.type,
+    template_ids: templateIds
+  });
+  console.log(JSON.stringify(result, null, 2));
+}
 async function main() {
   const args = process.argv.slice(2);
   if (args.length === 0 || args[0] === "--help" || args[0] === "-h") {
@@ -783,6 +924,28 @@ async function main() {
         process.exit(1);
       }
       await runAgent(args[1], args.slice(2));
+    } else if (command === "app") {
+      const subcommand = args[1];
+      if (!subcommand) {
+        console.error("Error: subcommand required for app command");
+        console.log("Usage: seacloud app <subcommand> [options]");
+        console.log("Subcommands: search, generation");
+        process.exit(1);
+      }
+      if (subcommand === "search") {
+        if (args.length < 3) {
+          console.error("Error: template IDs required for app search command");
+          console.log('Usage: seacloud app search "<id1,id2,...>" [options]');
+          process.exit(1);
+        }
+        await runAppSearch(args[2], args.slice(3));
+      } else if (subcommand === "generation") {
+        await runAppGeneration(args.slice(2));
+      } else {
+        console.error(`Error: unknown subcommand '${subcommand}'`);
+        console.log("Available subcommands: search, generation");
+        process.exit(1);
+      }
     } else {
       const model = command;
       if (model.startsWith("--")) {