seacloud-sdk 0.9.6 → 0.9.8

package/dist/cli.js

@@ -13,7 +13,51 @@ var SeacloudError = class extends Error {
 };
 
 // src/core/config.ts
-function getApiToken(providedApiKey) {
+function isInIframe() {
+  try {
+    return typeof globalThis.window !== "undefined" && globalThis.window.self !== globalThis.window.top;
+  } catch (e) {
+    return true;
+  }
+}
+async function getTokenFromParent(timeout = 5e3) {
+  if (!isInIframe()) {
+    return null;
+  }
+  return new Promise((resolve) => {
+    const messageHandler = (event) => {
+      if (event.data && event.data.type === "seaverse:token") {
+        cleanup();
+        const token = event.data.payload?.accessToken;
+        resolve(token || null);
+      } else if (event.data && event.data.type === "seaverse:error") {
+        cleanup();
+        console.warn("[SeaCloud SDK] Error getting token from parent:", event.data.error);
+        resolve(null);
+      }
+    };
+    const timeoutId = setTimeout(() => {
+      cleanup();
+      resolve(null);
+    }, timeout);
+    const cleanup = () => {
+      clearTimeout(timeoutId);
+      globalThis.window.removeEventListener("message", messageHandler);
+    };
+    globalThis.window.addEventListener("message", messageHandler);
+    try {
+      globalThis.window.parent.postMessage(
+        { type: "seaverse:get_token" },
+        "*"
+        // 允许任何源，支持跨域场景
+      );
+    } catch (e) {
+      cleanup();
+      resolve(null);
+    }
+  });
+}
+async function getApiToken(providedApiKey) {
   if (providedApiKey) {
     return providedApiKey;
   }
@@ -29,19 +73,24 @@ function getApiToken(providedApiKey) {
   if (typeof process !== "undefined" && process.env?.API_SERVICE_TOKEN) {
     return process.env.API_SERVICE_TOKEN;
   }
-  throw new Error(
-    'SeaCloud SDK: No API token found. Please ensure token is available in localStorage.getItem("auth_token") (browser) or process.env.API_SERVICE_TOKEN (Node.js), or initialize with initSeacloud({ apiKey: "your-token" }).'
-  );
+  if (typeof globalThis.window !== "undefined") {
+    const parentToken = await getTokenFromParent();
+    if (parentToken) {
+      return parentToken;
+    }
+  }
+  return "";
 }
 function createConfig(options = {}) {
-  const apiKey = getApiToken(options.apiKey);
+  const apiKey = options.apiKey;
   const baseUrl = options.baseUrl || (typeof process !== "undefined" ? process.env?.SEACLOUD_BASE_URL : void 0) || "https://proxy-rs.seaverse.ai";
   const fetchImpl = options.fetch || (globalThis.fetch ? globalThis.fetch.bind(globalThis) : void 0);
   if (!fetchImpl) {
     throw new Error("fetch is not available. Please provide a fetch implementation in config or upgrade to Node.js 18+");
   }
   return {
-    apiKey,
+    apiKey: apiKey || "",
+    // 提供默认空字符串，实际请求时会动态获取
     baseUrl,
     fetch: fetchImpl,
     timeout: options.timeout || 3e4
@@ -86,7 +135,7 @@ var SeacloudClient = class {
    */
   async createTask(endpoint, body) {
     const url = `${this.config.baseUrl}${endpoint}`;
-    const currentToken = getApiToken(this.providedApiKey);
+    const currentToken = await getApiToken(this.providedApiKey);
     const controller = new AbortController();
     const timeoutId = setTimeout(() => controller.abort(), this.config.timeout);
     try {
@@ -133,7 +182,7 @@ var SeacloudClient = class {
    */
   async getTaskStatus(endpoint, taskId) {
     const url = `${this.config.baseUrl}${endpoint}/task/${taskId}`;
-    const currentToken = getApiToken(this.providedApiKey);
+    const currentToken = await getApiToken(this.providedApiKey);
     const controller = new AbortController();
     const timeoutId = setTimeout(() => controller.abort(), this.config.timeout);
     try {
@@ -230,6 +279,7 @@ async function llmChatCompletions(params) {
   const client = getClient();
   const config = client.getConfig();
   const url = `${config.baseUrl}/llm/chat/completions`;
+  const token = await getApiToken(config.apiKey);
   const controller = new AbortController();
   const timeoutId = setTimeout(() => controller.abort(), config.timeout);
   try {
@@ -237,7 +287,7 @@ async function llmChatCompletions(params) {
       method: "POST",
       headers: {
         "Content-Type": "application/json",
-        "Authorization": `Bearer ${config.apiKey}`
+        "Authorization": `Bearer ${token}`
       },
       body: JSON.stringify(params),
       signal: controller.signal
@@ -505,6 +555,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 +600,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 +619,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 +647,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 +693,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 +863,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 +974,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("--")) {