seacloud-sdk 0.9.7 → 0.9.9

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
@@ -551,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:
@@ -581,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?"
@@ -604,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"}'
 
@@ -902,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") {
@@ -924,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) {