@infimind-next/ai-hive-mcp 0.3.3 → 0.4.1

package/dist/managed-token-provider.js

@@ -0,0 +1,31 @@
+import { execFile } from "node:child_process";
+export const MANAGED_AUTH_ERROR = "登录已失效，请回到 AI-HIVE 启动器重新登录";
+export class ManagedBearerTokenProvider {
+    helper;
+    constructor(helper) {
+        this.helper = helper;
+    }
+    async getToken() {
+        return new Promise((resolve, reject) => {
+            execFile(this.helper.command, this.helper.args, {
+                encoding: "utf8",
+                timeout: 10_000,
+                maxBuffer: 16_384,
+                windowsHide: true,
+                env: Object.fromEntries(Object.entries(process.env).filter(([name]) => !/KEY|TOKEN|SECRET|^(ELECTRON_RUN_AS_NODE|NODE_OPTIONS|NODE_PATH)$/i.test(name))),
+            }, (error, stdout) => {
+                const token = stdout.trim();
+                if (error ||
+                    !/^[A-Za-z0-9._~+/-]+={0,2}$/.test(token) ||
+                    token.length > 8_192) {
+                    reject(new Error(MANAGED_AUTH_ERROR));
+                    return;
+                }
+                resolve(token);
+            });
+        });
+    }
+    async refreshAfterUnauthorized() {
+        return this.getToken();
+    }
+}

package/dist/rest-commands.d.ts

@@ -0,0 +1,14 @@
+import { AiHiveApiClient } from "./api-client.js";
+import type { McpClientConfig } from "./config.js";
+export type RestCommandDependencies = {
+    stdout(line: string): void;
+    stderr(line: string): void;
+    createClient(input: {
+        endpoint: string;
+        key: string;
+        timeoutMs: number;
+    }): AiHiveApiClient;
+    newClientRequestId(): string;
+};
+export declare function runRestCommand(config: McpClientConfig, dependencies: RestCommandDependencies): Promise<number>;
+export declare const defaultRestCommandDependencies: RestCommandDependencies;

package/dist/rest-commands.js

@@ -0,0 +1,164 @@
+import { randomUUID } from "node:crypto";
+import { AiHiveApiClient, AiHiveApiError } from "./api-client.js";
+import { StaticBearerTokenProvider } from "./auth-token-provider.js";
+import { SERVER_VERSION } from "./server.js";
+// 这些命令让没有 MCP 宿主的 Agent(纯 shell)也能完成选型、试跑和查账;
+// 走的是同一个 /mcp 门面, 因此 MCP 密钥即可用, 不需要再申请 API 密钥.
+export async function runRestCommand(config, dependencies) {
+    if (!config.key) {
+        dependencies.stderr("缺少 MCP 密钥。请在控制台创建后传入 --key，或设置环境变量 AI_HIVE_MCP_KEY。");
+        return 1;
+    }
+    const client = dependencies.createClient({
+        endpoint: config.endpoint,
+        key: config.key,
+        timeoutMs: config.timeoutMs,
+    });
+    try {
+        switch (config.command) {
+            case "models":
+                return await runModels(config, client, dependencies);
+            case "model":
+                return await runModel(config, client, dependencies);
+            case "usage":
+                return await runUsage(client, dependencies);
+            case "run":
+                return await runPrompt(config, client, dependencies);
+            default:
+                return 1;
+        }
+    }
+    catch (error) {
+        dependencies.stderr(safeErrorMessage(error));
+        return 1;
+    }
+}
+async function runModels(config, client, dependencies) {
+    const result = await client.listModels({
+        ...(config.modelType ? { modelType: config.modelType } : {}),
+        detail: config.detail ?? "lite",
+    });
+    if (config.json) {
+        dependencies.stdout(JSON.stringify(result, null, 2));
+        return 0;
+    }
+    const models = Array.isArray(result) ? result : [];
+    if (models.length === 0) {
+        dependencies.stderr("没有可用的模型，请到控制台确认模型状态。");
+        return 1;
+    }
+    dependencies.stdout("模型 ID\t名称\t类型\t线路");
+    for (const model of models) {
+        dependencies.stdout([
+            text(model.publicModelId),
+            text(model.displayName),
+            text(model.modelType),
+            Array.isArray(model.routingModes)
+                ? model.routingModes
+                    .filter((mode) => typeof mode === "string")
+                    .join(",")
+                : "",
+        ].join("\t"));
+    }
+    dependencies.stdout("价格与参数用 models --json 或 model <模型 ID> 查看。");
+    return 0;
+}
+async function runModel(config, client, dependencies) {
+    if (!config.modelId) {
+        dependencies.stderr("请提供模型 ID，例如：ai-hive-mcp model doubao-seed-1-6");
+        return 1;
+    }
+    const result = await client.getModel(config.modelId, config.routingMode, {
+        announcements: false,
+    });
+    dependencies.stdout(JSON.stringify(result, null, 2));
+    return 0;
+}
+async function runUsage(client, dependencies) {
+    const result = await client.getBillingSummary({});
+    dependencies.stdout(JSON.stringify(result, null, 2));
+    return 0;
+}
+async function runPrompt(config, client, dependencies) {
+    if (!config.modelId || !config.prompt) {
+        dependencies.stderr('请提供模型 ID 和提示词，例如：ai-hive-mcp run doubao-seed-1-6 "写一句欢迎语"');
+        return 1;
+    }
+    const routingMode = config.routingMode ?? "COST_FIRST";
+    // /mcp/chat/text 的严格入参要求 pricingSnapshot, CLI 没有宿主代为补全, 必须自己取一次.
+    const model = (await client.getModel(config.modelId, routingMode, {
+        announcements: false,
+    }));
+    const pricingSnapshot = Array.isArray(model.pricingSnapshot)
+        ? model.pricingSnapshot.find((snapshot) => typeof snapshot === "object" &&
+            snapshot !== null &&
+            snapshot.routingMode === routingMode)
+        : undefined;
+    if (pricingSnapshot === undefined) {
+        dependencies.stderr("该模型当前没有所选线路的报价，请先用 model 命令确认可用线路。");
+        return 1;
+    }
+    const result = (await client.chatText({
+        publicModelId: config.modelId,
+        routingMode,
+        messages: [{ role: "user", content: config.prompt }],
+        pricingSnapshot,
+        clientRequestId: dependencies.newClientRequestId(),
+    }));
+    if (config.json) {
+        dependencies.stdout(JSON.stringify(result, null, 2));
+        return 0;
+    }
+    if (typeof result.content === "string") {
+        dependencies.stdout(result.content);
+    }
+    else {
+        dependencies.stdout(JSON.stringify(result, null, 2));
+    }
+    if (typeof result.chargeMicros === "string") {
+        dependencies.stdout(`本次消费：${formatHoneyMicros(result.chargeMicros)} 蜂蜜`);
+    }
+    return 0;
+}
+// 钱包与公开报价都以蜂蜜计价, 只有充值实付金额才是元.
+function formatHoneyMicros(value) {
+    if (!/^-?\d+$/.test(value)) {
+        return value;
+    }
+    const micros = BigInt(value);
+    const negative = micros < 0n;
+    const magnitude = negative ? -micros : micros;
+    const whole = (magnitude / 1000000n).toString();
+    const fraction = (magnitude % 1000000n)
+        .toString()
+        .padStart(6, "0")
+        .replace(/0+$/, "");
+    return `${negative ? "-" : ""}${whole}${fraction ? `.${fraction}` : ""}`;
+}
+function text(value) {
+    return typeof value === "string" ? value : "";
+}
+function safeErrorMessage(error) {
+    if (error instanceof AiHiveApiError) {
+        return error.message;
+    }
+    if (error instanceof Error && error.message.length > 0) {
+        return error.message;
+    }
+    return "命令执行失败，请稍后重试。";
+}
+export const defaultRestCommandDependencies = {
+    stdout: (line) => console.log(line),
+    stderr: (line) => console.error(line),
+    createClient: ({ endpoint, key, timeoutMs }) => new AiHiveApiClient({
+        endpoint,
+        tokenProvider: new StaticBearerTokenProvider(key),
+        timeoutMs,
+        // 命令行调用没有 MCP 宿主可以代报身份, 自己声明来源, 后台才能区分 CLI 与裸 MCP.
+        clientInfoProvider: () => ({
+            name: "ai-hive-mcp-cli",
+            version: SERVER_VERSION,
+        }),
+    }),
+    newClientRequestId: () => randomUUID(),
+};

package/dist/schemas.d.ts

@@ -9,37 +9,37 @@ export declare const mcpCreateMediaUploadOutputSchema: z.ZodObject<{
         method: z.ZodLiteral<"PUT">;
         headers: z.ZodRecord<z.ZodString, z.ZodString>;
     }, "strict", z.ZodTypeAny, {
-        method: "PUT";
         url: string;
+        method: "PUT";
         headers: Record<string, string>;
     }, {
-        method: "PUT";
         url: string;
+        method: "PUT";
         headers: Record<string, string>;
     }>;
     expiresAt: z.ZodString;
 }, "strict", z.ZodTypeAny, {
-    mimeType: string;
-    expiresAt: string;
     mediaId: string;
     mediaType: "IMAGE" | "VIDEO" | "AUDIO" | "DOCUMENT";
+    mimeType: string;
     sizeBytes: number;
     upload: {
-        method: "PUT";
         url: string;
+        method: "PUT";
         headers: Record<string, string>;
     };
-}, {
-    mimeType: string;
     expiresAt: string;
+}, {
     mediaId: string;
     mediaType: "IMAGE" | "VIDEO" | "AUDIO" | "DOCUMENT";
+    mimeType: string;
     sizeBytes: number;
     upload: {
-        method: "PUT";
         url: string;
+        method: "PUT";
         headers: Record<string, string>;
     };
+    expiresAt: string;
 }>;
 export declare const mcpUploadMediaFromPathInputSchema: z.ZodObject<{
     path: z.ZodEffects<z.ZodString, string, string>;
@@ -70,12 +70,12 @@ export declare const chatTextShape: {
         content: z.ZodEffects<z.ZodString, string, string>;
         mediaIds: z.ZodDefault<z.ZodEffects<z.ZodUnion<[z.ZodArray<z.ZodString, "many">, z.ZodNull]>, string[], string[] | null>>;
     }, "strict", z.ZodTypeAny, {
+        role: "system" | "user" | "assistant";
         content: string;
-        role: "user" | "assistant" | "system";
         mediaIds: string[];
     }, {
+        role: "system" | "user" | "assistant";
         content: string;
-        role: "user" | "assistant" | "system";
         mediaIds?: string[] | null | undefined;
     }>, "many">;
     thinkingEnabled: z.ZodOptional<z.ZodBoolean>;
@@ -109,7 +109,115 @@ export declare const generateVideoShape: {
     clientRequestId: z.ZodOptional<z.ZodEffects<z.ZodString, string, string>>;
     pricingSnapshot: z.ZodOptional<z.ZodUnion<[z.ZodObject<{}, "passthrough", z.ZodTypeAny, z.objectOutputType<{}, z.ZodTypeAny, "passthrough">, z.objectInputType<{}, z.ZodTypeAny, "passthrough">>, z.ZodArray<z.ZodNullable<z.ZodObject<{}, "passthrough", z.ZodTypeAny, z.objectOutputType<{}, z.ZodTypeAny, "passthrough">, z.objectInputType<{}, z.ZodTypeAny, "passthrough">>>, "many">, z.ZodString, z.ZodNull]>>;
 };
+export declare const listGenerationTasksShape: {
+    publicModelId: z.ZodOptional<z.ZodString>;
+    page: z.ZodDefault<z.ZodNumber>;
+    pageSize: z.ZodDefault<z.ZodNumber>;
+};
 export declare const getGenerationTaskShape: {
     taskId: z.ZodString;
 };
 export type McpUploadMediaFromPathInput = z.infer<typeof mcpUploadMediaFromPathInputSchema>;
+export declare const billingQueryShape: {
+    from: z.ZodOptional<z.ZodString>;
+    to: z.ZodOptional<z.ZodString>;
+    taskIds: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+    publicModelId: z.ZodOptional<z.ZodString>;
+    invocationSource: z.ZodOptional<z.ZodEnum<["UNKNOWN", "WEB", "MCP", "WORKBUDDY", "QWENWORK", "DOUBAO_WORK", "MIAOSHOU", "OPENAPI"]>>;
+};
+export declare const listBillingRecordsShape: {
+    pageSize: z.ZodDefault<z.ZodNumber>;
+    cursor: z.ZodOptional<z.ZodString>;
+    from: z.ZodOptional<z.ZodString>;
+    to: z.ZodOptional<z.ZodString>;
+    taskIds: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+    publicModelId: z.ZodOptional<z.ZodString>;
+    invocationSource: z.ZodOptional<z.ZodEnum<["UNKNOWN", "WEB", "MCP", "WORKBUDDY", "QWENWORK", "DOUBAO_WORK", "MIAOSHOU", "OPENAPI"]>>;
+};
+export declare const getBillingRecordShape: {
+    recordId: z.ZodString;
+};
+export type BillingQueryInput = z.infer<z.ZodObject<typeof billingQueryShape>>;
+export type ListBillingRecordsInput = BillingQueryInput & {
+    pageSize?: number;
+    cursor?: string;
+};
+export declare const billingQueryInputSchema: z.ZodEffects<z.ZodObject<{
+    from: z.ZodOptional<z.ZodString>;
+    to: z.ZodOptional<z.ZodString>;
+    taskIds: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+    publicModelId: z.ZodOptional<z.ZodString>;
+    invocationSource: z.ZodOptional<z.ZodEnum<["UNKNOWN", "WEB", "MCP", "WORKBUDDY", "QWENWORK", "DOUBAO_WORK", "MIAOSHOU", "OPENAPI"]>>;
+}, "strict", z.ZodTypeAny, {
+    publicModelId?: string | undefined;
+    from?: string | undefined;
+    to?: string | undefined;
+    taskIds?: string[] | undefined;
+    invocationSource?: "UNKNOWN" | "WEB" | "MCP" | "WORKBUDDY" | "QWENWORK" | "DOUBAO_WORK" | "MIAOSHOU" | "OPENAPI" | undefined;
+}, {
+    publicModelId?: string | undefined;
+    from?: string | undefined;
+    to?: string | undefined;
+    taskIds?: string[] | undefined;
+    invocationSource?: "UNKNOWN" | "WEB" | "MCP" | "WORKBUDDY" | "QWENWORK" | "DOUBAO_WORK" | "MIAOSHOU" | "OPENAPI" | undefined;
+}>, {
+    publicModelId?: string | undefined;
+    from?: string | undefined;
+    to?: string | undefined;
+    taskIds?: string[] | undefined;
+    invocationSource?: "UNKNOWN" | "WEB" | "MCP" | "WORKBUDDY" | "QWENWORK" | "DOUBAO_WORK" | "MIAOSHOU" | "OPENAPI" | undefined;
+}, {
+    publicModelId?: string | undefined;
+    from?: string | undefined;
+    to?: string | undefined;
+    taskIds?: string[] | undefined;
+    invocationSource?: "UNKNOWN" | "WEB" | "MCP" | "WORKBUDDY" | "QWENWORK" | "DOUBAO_WORK" | "MIAOSHOU" | "OPENAPI" | undefined;
+}>;
+export declare const listBillingRecordsInputSchema: z.ZodEffects<z.ZodObject<{
+    pageSize: z.ZodDefault<z.ZodNumber>;
+    cursor: z.ZodOptional<z.ZodString>;
+    from: z.ZodOptional<z.ZodString>;
+    to: z.ZodOptional<z.ZodString>;
+    taskIds: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+    publicModelId: z.ZodOptional<z.ZodString>;
+    invocationSource: z.ZodOptional<z.ZodEnum<["UNKNOWN", "WEB", "MCP", "WORKBUDDY", "QWENWORK", "DOUBAO_WORK", "MIAOSHOU", "OPENAPI"]>>;
+}, "strict", z.ZodTypeAny, {
+    pageSize: number;
+    publicModelId?: string | undefined;
+    from?: string | undefined;
+    to?: string | undefined;
+    taskIds?: string[] | undefined;
+    invocationSource?: "UNKNOWN" | "WEB" | "MCP" | "WORKBUDDY" | "QWENWORK" | "DOUBAO_WORK" | "MIAOSHOU" | "OPENAPI" | undefined;
+    cursor?: string | undefined;
+}, {
+    publicModelId?: string | undefined;
+    pageSize?: number | undefined;
+    from?: string | undefined;
+    to?: string | undefined;
+    taskIds?: string[] | undefined;
+    invocationSource?: "UNKNOWN" | "WEB" | "MCP" | "WORKBUDDY" | "QWENWORK" | "DOUBAO_WORK" | "MIAOSHOU" | "OPENAPI" | undefined;
+    cursor?: string | undefined;
+}>, {
+    pageSize: number;
+    publicModelId?: string | undefined;
+    from?: string | undefined;
+    to?: string | undefined;
+    taskIds?: string[] | undefined;
+    invocationSource?: "UNKNOWN" | "WEB" | "MCP" | "WORKBUDDY" | "QWENWORK" | "DOUBAO_WORK" | "MIAOSHOU" | "OPENAPI" | undefined;
+    cursor?: string | undefined;
+}, {
+    publicModelId?: string | undefined;
+    pageSize?: number | undefined;
+    from?: string | undefined;
+    to?: string | undefined;
+    taskIds?: string[] | undefined;
+    invocationSource?: "UNKNOWN" | "WEB" | "MCP" | "WORKBUDDY" | "QWENWORK" | "DOUBAO_WORK" | "MIAOSHOU" | "OPENAPI" | undefined;
+    cursor?: string | undefined;
+}>;
+export declare const getBillingRecordInputSchema: z.ZodObject<{
+    recordId: z.ZodString;
+}, "strict", z.ZodTypeAny, {
+    recordId: string;
+}, {
+    recordId: string;
+}>;

package/dist/schemas.js

@@ -116,6 +116,85 @@ export const generateVideoShape = {
     clientRequestId: clientRequestIdSchema,
     pricingSnapshot: pricingSnapshotInputSchema,
 };
+export const listGenerationTasksShape = {
+    publicModelId: z.string().min(1).optional(),
+    page: z.number().int().min(1).default(1),
+    pageSize: z.number().int().min(1).max(50).default(20),
+};
 export const getGenerationTaskShape = {
     taskId: z.string().min(1),
 };
+// 独立发布的客户端镜像服务端账单输入, 避免引入 contracts 运行时依赖.
+export const billingQueryShape = {
+    from: z
+        .string()
+        .datetime({ offset: true })
+        .optional()
+        .describe("开始时间，ISO 8601 含时区；与 to 成对传入，包含此时刻；不填默认北京时间今日"),
+    to: z
+        .string()
+        .datetime({ offset: true })
+        .optional()
+        .describe("结束时间，ISO 8601 含时区；与 from 成对传入，不包含此时刻"),
+    taskIds: z
+        .array(z.string().trim().min(1).max(200))
+        .min(1)
+        .max(100)
+        .optional()
+        .describe("按生成任务 ID 查询指定批次，最多 100 个；不能用积分差推断批次消费"),
+    publicModelId: z.string().trim().min(1).max(200).optional(),
+    invocationSource: z
+        .enum([
+        "UNKNOWN",
+        "WEB",
+        "MCP",
+        "WORKBUDDY",
+        "QWENWORK",
+        "DOUBAO_WORK",
+        "MIAOSHOU",
+        "OPENAPI",
+    ])
+        .optional(),
+};
+export const listBillingRecordsShape = {
+    ...billingQueryShape,
+    pageSize: z.number().int().min(1).max(100).default(20),
+    cursor: z.string().min(1).max(8192).optional(),
+};
+export const getBillingRecordShape = {
+    recordId: z
+        .string()
+        .regex(/^(charge|reservation):[^:]+$/)
+        .max(250)
+        .describe("列表返回的账单记录 ID，例如 charge:<id> 或 reservation:<id>"),
+};
+function refineBillingRange(value, ctx) {
+    if ((value.from === undefined) !== (value.to === undefined)) {
+        ctx.addIssue({
+            code: z.ZodIssueCode.custom,
+            message: "from 和 to 必须同时填写",
+            path: [value.from === undefined ? "from" : "to"],
+        });
+        return;
+    }
+    if (value.from !== undefined &&
+        value.to !== undefined &&
+        Date.parse(value.from) >= Date.parse(value.to)) {
+        ctx.addIssue({
+            code: z.ZodIssueCode.custom,
+            message: "结束时间必须晚于开始时间",
+            path: ["to"],
+        });
+    }
+}
+export const billingQueryInputSchema = z
+    .object(billingQueryShape)
+    .strict()
+    .superRefine(refineBillingRange);
+export const listBillingRecordsInputSchema = z
+    .object(listBillingRecordsShape)
+    .strict()
+    .superRefine(refineBillingRange);
+export const getBillingRecordInputSchema = z
+    .object(getBillingRecordShape)
+    .strict();

package/dist/server.d.ts

@@ -5,7 +5,7 @@ import { AiHiveApiClient } from "./api-client.js";
 import { type BearerTokenProvider } from "./auth-token-provider.js";
 import type { McpClientConfig } from "./config.js";
 import { WorkBuddyCredentialStore } from "./workbuddy-credentials.js";
-export declare const SERVER_VERSION = "0.3.3";
+export declare const SERVER_VERSION = "0.4.1";
 declare function content(value: unknown): {
     content: {
         type: "text";
@@ -20,12 +20,13 @@ declare function userInfoContent(value: unknown): {
     }[];
     structuredContent: Record<string, unknown>;
 };
-type ApiClientLike = Pick<AiHiveApiClient, "getUserInfo" | "getMaterialMode" | "quoteVideoMaterials" | "listModels" | "getModel" | "createMediaUpload" | "completeMediaUpload" | "chatText" | "generateImage" | "generateVideo" | "getGenerationTask" | "putUpload">;
+type ApiClientLike = Pick<AiHiveApiClient, "getUserInfo" | "getMaterialMode" | "quoteVideoMaterials" | "listModels" | "getModel" | "createMediaUpload" | "completeMediaUpload" | "chatText" | "generateImage" | "generateVideo" | "getGenerationTask" | "listGenerationTasks" | "putUpload"> & Partial<Pick<AiHiveApiClient, "getBillingSummary" | "listBillingRecords" | "getBillingRecord">>;
 export type ToolDefinition = {
     name: string;
-    title?: string;
+    title: string;
     description: string;
     schema: z.ZodRawShape;
+    inputSchema?: z.AnyZodObject;
     annotations?: ToolAnnotations;
     handler: (args: Record<string, unknown>) => Promise<ReturnType<typeof content> | ReturnType<typeof userInfoContent>>;
 };