feat: add minimax tts support (#1043)

2026-09-22 13:38:31 +00:00 · 2026-05-13 15:34:10 +07:00
parent 3c2503c4b4
commit 74c9879e8e
10 changed files with 578 additions and 32 deletions
@@ -0,0 +1,116 @@
+import { describe, it, expect, vi, beforeEach } from "vitest";
+
+vi.mock("../../open-sse/utils/proxyFetch.js", () => ({
+  proxyAwareFetch: vi.fn(),
+}));
+
+import { proxyAwareFetch } from "../../open-sse/utils/proxyFetch.js";
+import { getUsageForProvider } from "../../open-sse/services/usage.js";
+
+function usageResponse(modelRemains) {
+  return new Response(
+    JSON.stringify({
+      base_resp: { status_code: 0, status_msg: "success" },
+      model_remains: modelRemains,
+    }),
+    { status: 200, headers: { "Content-Type": "application/json" } }
+  );
+}
+
+describe("MiniMax usage", () => {
+  beforeEach(() => {
+    vi.clearAllMocks();
+  });
+
+  it("parses token-plan TTS quota counts as used counts", async () => {
+    proxyAwareFetch.mockResolvedValueOnce(
+      usageResponse([
+        {
+          model_name: "text to speech hd",
+          current_interval_total_count: 4000,
+          current_interval_usage_count: 25,
+          current_weekly_total_count: 12000,
+          current_weekly_usage_count: 100,
+          end_time: "2026-05-12T10:00:00.000Z",
+          weekly_end_time: "2026-05-19T10:00:00.000Z",
+        },
+      ])
+    );
+
+    const usage = await getUsageForProvider({
+      provider: "minimax",
+      apiKey: "test-key",
+    });
+
+    expect(usage.message).toBeUndefined();
+    expect(usage.quotas["Text to Speech HD (5h)"]).toMatchObject({
+      used: 25,
+      total: 4000,
+      remaining: 3975,
+    });
+    expect(usage.quotas["Text to Speech HD (7d)"]).toMatchObject({
+      used: 100,
+      total: 12000,
+      remaining: 11900,
+    });
+  });
+
+  it("parses coding-plan TTS quota counts as remaining counts", async () => {
+    proxyAwareFetch.mockResolvedValueOnce(
+      usageResponse([
+        {
+          modelName: "Text to Speech HD",
+          currentIntervalTotalCount: 4000,
+          currentIntervalUsageCount: 4000,
+          currentWeeklyTotalCount: 12000,
+          currentWeeklyUsageCount: 11800,
+          remainsTime: 1000,
+          weeklyRemainsTime: 2000,
+        },
+      ])
+    );
+
+    const usage = await getUsageForProvider({
+      provider: "minimax-cn",
+      apiKey: "test-key",
+    });
+
+    expect(usage.message).toBeUndefined();
+    expect(usage.quotas["Text to Speech HD (5h)"]).toMatchObject({
+      used: 0,
+      total: 4000,
+      remaining: 4000,
+    });
+    expect(usage.quotas["Text to Speech HD (7d)"]).toMatchObject({
+      used: 200,
+      total: 12000,
+      remaining: 11800,
+    });
+  });
+
+  it("keeps non-TTS MiniMax quota rows instead of filtering to text only", async () => {
+    proxyAwareFetch.mockResolvedValueOnce(
+      usageResponse([
+        {
+          model_name: "music-2.6",
+          current_interval_total_count: 100,
+          current_interval_usage_count: 5,
+        },
+        {
+          model_name: "image-01",
+          current_interval_total_count: 50,
+          current_interval_usage_count: 2,
+        },
+      ])
+    );
+
+    const usage = await getUsageForProvider({
+      provider: "minimax",
+      apiKey: "test-key",
+    });
+
+    expect(Object.keys(usage.quotas)).toEqual(["Music 2.6 (5h)", "Image 01 (5h)"]);
+    expect(usage.quotas["Music 2.6 (5h)"].used).toBe(5);
+    expect(usage.quotas["Image 01 (5h)"].used).toBe(2);
+  });
+});