@fangyb/ahchat-bridge 0.1.32 → 0.1.33

package/dist/seedanceMcpCli.cjs

@@ -31231,6 +31231,59 @@ var StdioServerTransport = class {
   }
 };
 
+// src/officialMcpQuota.ts
+init_cjs_shims();
+var OfficialMcpQuotaError = class extends Error {
+  constructor(message) {
+    super(message);
+    this.name = "OfficialMcpQuotaError";
+  }
+};
+async function consumeOfficialMcpDailyQuota(serverName, toolName) {
+  const serverApiUrl = process.env.AHCHAT_SERVER_API_URL?.trim();
+  const bridgeToken = process.env.AHCHAT_BRIDGE_TOKEN?.trim();
+  if (!serverApiUrl || !bridgeToken) return;
+  let response;
+  try {
+    response = await fetch(`${serverApiUrl.replace(/\/+$/, "")}/api/mcp/usage/consume`, {
+      method: "POST",
+      headers: {
+        "Content-Type": "application/json",
+        "X-AHChat-Bridge-Token": bridgeToken
+      },
+      body: JSON.stringify({ serverName, toolName })
+    });
+  } catch {
+    throw new OfficialMcpQuotaError("\u6682\u65F6\u65E0\u6CD5\u6821\u9A8C\u5B98\u65B9\u5A92\u4F53 MCP \u6BCF\u65E5\u989D\u5EA6\uFF0C\u8BF7\u7A0D\u540E\u518D\u8BD5\u3002");
+  }
+  if (response.status === 404) return;
+  const body = await readQuotaResponse(response);
+  if (response.status === 429 || body.allowed === false) {
+    throw new OfficialMcpQuotaError(formatQuotaExceededMessage(serverName, body));
+  }
+  if (!response.ok) {
+    throw new OfficialMcpQuotaError(body.message || "\u5B98\u65B9\u5A92\u4F53 MCP \u6BCF\u65E5\u989D\u5EA6\u6821\u9A8C\u5931\u8D25\uFF0C\u8BF7\u7A0D\u540E\u518D\u8BD5\u3002");
+  }
+}
+async function readQuotaResponse(response) {
+  try {
+    const value = await response.json();
+    if (value && typeof value === "object" && !Array.isArray(value)) {
+      return value;
+    }
+  } catch {
+  }
+  return {};
+}
+function formatQuotaExceededMessage(serverName, body) {
+  if (body.message) return body.message;
+  const label = serverName === "seedance" ? "\u751F\u89C6\u9891" : serverName === "seedream" ? "\u751F\u56FE" : "\u5B98\u65B9\u5A92\u4F53";
+  if (typeof body.used === "number" && typeof body.limit === "number") {
+    return `\u4ECA\u5929\u7684${label}\u989D\u5EA6\u5DF2\u7528\u5B8C\uFF08${body.used}/${body.limit}\uFF09\u3002\u8BF7\u660E\u5929\u518D\u8BD5\uFF0C\u6216\u8054\u7CFB\u8FD0\u8425\u8C03\u6574\u989D\u5EA6\u3002`;
+  }
+  return `\u4ECA\u5929\u7684${label}\u989D\u5EA6\u5DF2\u7528\u5B8C\u3002\u8BF7\u660E\u5929\u518D\u8BD5\uFF0C\u6216\u8054\u7CFB\u8FD0\u8425\u8C03\u6574\u989D\u5EA6\u3002`;
+}
+
 // src/seedanceMcpCli.ts
 var SERVER_NAME = "ahchat-seedance-mcp";
 var SERVER_VERSION = "0.1.0";
@@ -31292,7 +31345,7 @@ function createServer() {
   const server = new McpServer(
     { name: SERVER_NAME, version: SERVER_VERSION },
     {
-      instructions: "AHChat official Seedance video generation MCP. Call seedance_create_task to submit a job, then seedance_check_task to poll the task_id. Generated URLs can expire; show or download them promptly."
+      instructions: "AHChat official Seedance video generation MCP. Each user request should create exactly one video task. Call seedance_create_task once to submit a job, then seedance_check_task only when progress is requested. Generated URLs can expire; show or download them promptly."
     }
   );
   server.registerTool(
@@ -31315,7 +31368,7 @@ function createServer() {
     "seedance_create_task",
     {
       title: "Create Seedance video generation task",
-      description: "Submit a Seedance video generation task to the Volcengine Ark API and return the task_id immediately. Poll seedance_check_task afterwards.",
+      description: "Submit exactly one Seedance video generation task to the Volcengine Ark API and return the task_id immediately. In AHChat, do not submit alternate tasks, loop, or sleep afterwards; the media task card tracks status.",
       inputSchema: createTaskInputShape,
       annotations: {
         readOnlyHint: false,
@@ -31328,6 +31381,7 @@ function createServer() {
       if (validationError) return errorResult(validationError);
       try {
         const normalized = normalizeCreateTaskInput(input);
+        await consumeOfficialMcpDailyQuota("seedance", "seedance_create_task");
         const result = await createSeedanceTask(normalized);
         const payload = {
           state: "submitted",
@@ -31500,7 +31554,10 @@ async function parseJsonSafe(response) {
   if (!text) return void 0;
   try {
     return JSON.parse(text);
-  } catch {
+  } catch (error51) {
+    writeStderr(
+      `[${SERVER_NAME}] ARK API returned a non-JSON response body: ${error51 instanceof Error ? error51.message : String(error51)}`
+    );
     return text;
   }
 }
@@ -31579,22 +31636,27 @@ function errorResult(message) {
 function usageGuide() {
   return [
     "AHChat Seedance workflow:",
-    "1. Call seedance_create_task with prompt and optional reference media URLs.",
-    "2. Wait 30-90 seconds, then call seedance_check_task with the returned task_id.",
-    "3. When status is succeeded, use video_url in the chat result. Signed URLs may expire.",
+    "1. Call seedance_create_task with prompt and optional reference media URLs. Each user request creates exactly one video task.",
+    "2. After submission, stop manual waiting. AHChat will track the task in the visible media card.",
+    "3. Only call seedance_check_task once when the user explicitly asks for progress or when diagnostics are needed.",
+    "4. When status is succeeded, keep the text reply short and let the media card show preview, download, copy, and regenerate actions. Signed URLs may expire.",
     "",
     "Common defaults: model=doubao-seedance-2-0-260128, duration=5, ratio=16:9, resolution=720p, generate_audio=true.",
     "For vertical short video, set ratio=9:16. Reference media URLs must be reachable by Volcengine Ark."
   ].join("\n");
 }
+function writeStderr(message) {
+  process.stderr.write(`${message}
+`);
+}
 async function main() {
   const server = createServer();
   const transport = new StdioServerTransport();
   await server.connect(transport);
-  console.error(`${SERVER_NAME} v${SERVER_VERSION} running on stdio`);
+  writeStderr(`${SERVER_NAME} v${SERVER_VERSION} running on stdio`);
 }
 main().catch((error51) => {
   const message = error51 instanceof Error ? error51.stack ?? error51.message : String(error51);
-  console.error(`Fatal error in ${SERVER_NAME}: ${message}`);
+  writeStderr(`Fatal error in ${SERVER_NAME}: ${message}`);
   process.exit(1);
 });