agent-bridge-qwenwork 0.0.0-stage → 1.0.0

package/README.md

@@ -1,3 +1,5 @@
-# Temporary Holding Version
+# 千问办公插件
 
-This version is a temporary placeholder for this package. An operational version to replace this has been submitted for review and is awaiting a staged release.
+供 `mu-agent-bridge` 使用的千问办公插件。核心包已内置此插件；独立包适用于单独发布和安装。
+
+安装：`npm install -g agent-bridge-qwenwork`。插件需要 Windows 千问办公开启 CDP 端口 `9226`。

package/config.schema.json

@@ -0,0 +1,14 @@
+{
+  "$schema": "https://json-schema.org/draft/2020-12/schema",
+  "type": "object",
+  "properties": {
+    "aid": { "type": "string", "description": "复用的 EvolCore 已有 AID" },
+    "cdpPort": { "type": "integer", "minimum": 1, "maximum": 65535, "default": 9226 },
+    "httpPort": { "type": "integer", "minimum": 0, "maximum": 65535, "default": 17893 },
+    "subChatId": { "type": "string" },
+    "chatId": { "type": "string" },
+    "workspace": { "type": "string" },
+    "timeoutMs": { "type": "integer", "minimum": 1000, "default": 240000 }
+  },
+  "additionalProperties": false
+}

package/dist/src-ts/agent.js

@@ -0,0 +1,10 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AgentError = void 0;
+class AgentError extends Error {
+    constructor(message, options) {
+        super(message, options);
+        this.name = "AgentError";
+    }
+}
+exports.AgentError = AgentError;

package/dist/src-ts/plugins/qwenwork/cdp-client.js

@@ -0,0 +1,165 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.QwenCdpClient = exports.inputSelector = exports.assistantMessageSelector = void 0;
+exports.freshAssistantMessages = freshAssistantMessages;
+function sleep(ms) { return new Promise((resolve) => setTimeout(resolve, ms)); }
+exports.assistantMessageSelector = '[data-message-id][data-part-type="text"]:not([data-conversation-user-text="true"])';
+exports.inputSelector = '.chat-input-editor-text[contenteditable="true"]';
+function freshAssistantMessages(beforeIds, messages) {
+    return messages.filter((message) => Boolean(message.id) && !beforeIds.has(message.id) && Boolean(message.text?.trim()));
+}
+function wait(ms, signal) {
+    if (signal?.aborted)
+        return Promise.reject(new Error("请求已取消"));
+    return new Promise((resolve, reject) => {
+        const timer = setTimeout(resolve, ms);
+        signal?.addEventListener("abort", () => {
+            clearTimeout(timer);
+            reject(new Error("请求已取消"));
+        }, { once: true });
+    });
+}
+/** CDP/tRPC transport for the QwenWorkCN desktop client. */
+class QwenCdpClient {
+    socket;
+    sequence = 0;
+    pending = new Map();
+    options;
+    constructor(options = {}) {
+        this.options = {
+            port: options.port ?? Number(process.env.QWEN_CDP_PORT ?? "9226"),
+            subChatId: options.subChatId ?? process.env.QWEN_SUBCHAT ?? "muj55ep9426gudjw",
+            chatId: options.chatId ?? process.env.QWEN_CHAT ?? "muj55ep9u6wbv0eo",
+            workspace: options.workspace ?? `${process.env.APPDATA ?? ""}\\QwenWorkCN\\workspace`,
+            timeoutMs: options.timeoutMs ?? 30_000,
+            fetchImpl: options.fetchImpl,
+        };
+    }
+    connected() { return Boolean(this.socket); }
+    async connect() {
+        if (this.socket)
+            return;
+        const fetchImpl = this.options.fetchImpl ?? fetch;
+        const response = await fetchImpl(`http://127.0.0.1:${this.options.port}/json`);
+        if (!response.ok)
+            throw new Error(`CDP /json HTTP ${response.status}`);
+        const targets = await response.json();
+        const page = targets.find((target) => target.type === "page" && (target.url ?? "").includes("index.html#windowId=main"))
+            ?? targets.find((target) => target.type === "page" && (target.url ?? "").includes("index.html"));
+        if (!page?.webSocketDebuggerUrl)
+            throw new Error(`未找到千问主窗口 target（客户端是否以 --remote-debugging-port=${this.options.port} 启动？）`);
+        this.socket = new WebSocket(page.webSocketDebuggerUrl);
+        this.socket.onmessage = (event) => this.onMessage(String(event.data));
+        await new Promise((resolve, reject) => {
+            const socket = this.socket;
+            socket.onopen = () => resolve();
+            socket.onerror = () => reject(new Error("CDP WebSocket 连接失败"));
+        });
+        await this.evaluate(`(() => {
+      if (!window.__ts_bridge_resps) {
+        window.__ts_bridge_resps = [];
+        window.electronTRPC.onMessage(message => window.__ts_bridge_resps.push(message));
+      }
+      return 'ok';
+    })()`);
+    }
+    close() {
+        this.socket?.close();
+        this.socket = undefined;
+        for (const pending of this.pending.values())
+            pending.reject(new Error("CDP 已关闭"));
+        this.pending.clear();
+    }
+    async send(text) {
+        return this.trpc("agent.submitMessage", "mutation", {
+            subChatId: this.options.subChatId,
+            chatId: this.options.chatId,
+            prompt: text,
+            cwd: this.options.workspace,
+        });
+    }
+    async direct(text, timeoutMs = 240_000, signal) {
+        const snapshot = await this.evaluate(`JSON.stringify([...document.querySelectorAll(${JSON.stringify(exports.assistantMessageSelector)})].map(element => ({ id: element.getAttribute('data-message-id'), text: element.innerText })))`);
+        const before = JSON.parse(snapshot || "[]");
+        const beforeIds = new Set(before.map((message) => message.id).filter((id) => Boolean(id)));
+        const focused = await this.evaluate(`(() => {
+      const element = document.querySelector(${JSON.stringify(exports.inputSelector)});
+      if (!element) return false;
+      element.focus();
+      return true;
+    })()`);
+        if (!focused)
+            throw new Error("千问输入框未找到");
+        await this.cdp("Input.insertText", { text });
+        for (const type of ["keyDown", "keyUp"]) {
+            await this.cdp("Input.dispatchKeyEvent", {
+                type, key: "Enter", code: "Enter", text: type === "keyDown" ? "\r" : "", unmodifiedText: type === "keyDown" ? "\r" : "",
+                windowsVirtualKeyCode: 13, nativeVirtualKeyCode: 13,
+            });
+        }
+        let previousReply = "";
+        let stablePolls = 0;
+        const deadline = Date.now() + timeoutMs;
+        while (Date.now() < deadline) {
+            await wait(2_000, signal);
+            const messagesJson = await this.evaluate(`JSON.stringify([...document.querySelectorAll(${JSON.stringify(exports.assistantMessageSelector)})].map(element => ({ id: element.getAttribute('data-message-id'), text: element.innerText })))`);
+            const fresh = freshAssistantMessages(beforeIds, JSON.parse(messagesJson || "[]"));
+            const reply = fresh.at(-1)?.text.trim() ?? "";
+            stablePolls = reply && reply === previousReply ? stablePolls + 1 : 0;
+            previousReply = reply;
+            if (reply && stablePolls >= 3)
+                return reply;
+        }
+        throw new Error("千问办公 UI 超时未返回回复");
+    }
+    onMessage(raw) {
+        let message;
+        try {
+            message = JSON.parse(raw);
+        }
+        catch {
+            return;
+        }
+        if (message.id === undefined)
+            return;
+        const pending = this.pending.get(message.id);
+        if (!pending)
+            return;
+        this.pending.delete(message.id);
+        pending.resolve(message);
+    }
+    async cdp(method, params = {}, timeoutMs = this.options.timeoutMs) {
+        if (!this.socket)
+            throw new Error("CDP 尚未连接");
+        const id = ++this.sequence;
+        const result = new Promise((resolve, reject) => this.pending.set(id, { resolve, reject }));
+        this.socket.send(JSON.stringify({ id, method, params }));
+        return Promise.race([result, sleep(timeoutMs).then(() => { throw new Error(`CDP ${method} 超时`); })]);
+    }
+    async evaluate(expression) {
+        const response = await this.cdp("Runtime.evaluate", { expression, returnByValue: true, awaitPromise: true });
+        if (response.result?.exceptionDetails)
+            throw new Error(`页面 JS 异常: ${JSON.stringify(response.result.exceptionDetails).slice(0, 300)}`);
+        return response.result?.result?.value;
+    }
+    async trpc(path, type, input) {
+        const id = `ts-${Date.now()}-${Math.random().toString(36).slice(2, 8)}`;
+        const operation = { id, type, path, input: { json: input, meta: {} } };
+        await this.evaluate(`(() => { window.electronTRPC.sendMessage({ method: 'request', operation: ${JSON.stringify(operation)} }); return 'sent'; })()`);
+        const deadline = Date.now() + 20_000;
+        while (Date.now() < deadline) {
+            await wait(400);
+            const encoded = await this.evaluate(`(() => {
+        const list = window.__ts_bridge_resps ?? [];
+        const index = list.findIndex(message => message && message.id === ${JSON.stringify(id)});
+        if (index < 0) return null;
+        const [message] = list.splice(index, 1);
+        return JSON.stringify(message);
+      })()`);
+            if (encoded && encoded !== "null")
+                return JSON.parse(String(encoded));
+        }
+        throw new Error(`tRPC ${path} 超时`);
+    }
+}
+exports.QwenCdpClient = QwenCdpClient;

package/dist/src-ts/plugins/qwenwork/index.js

@@ -0,0 +1,121 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.QwenworkPlugin = exports.qwenworkManifest = void 0;
+exports.createQwenworkPlugin = createQwenworkPlugin;
+const openai_adapter_1 = require("./openai-adapter");
+const cdp_client_1 = require("./cdp-client");
+exports.qwenworkManifest = {
+    id: "qwenwork",
+    version: "1.0.0",
+    displayName: "千问办公",
+    entry: "dist/index.js",
+    apiVersion: "1",
+    process: "in-process",
+    capabilities: ["chat", "openai-compatible", "health"],
+    configSchema: "config.schema.json",
+    defaults: { cdpPort: 9226, httpPort: 17893 },
+};
+function asNumber(value, fallback) {
+    return typeof value === "number" && Number.isFinite(value) ? value : fallback;
+}
+/** Dynamic qwenwork plugin. It returns text to the platform and never emits AUN messages itself. */
+class QwenworkPlugin {
+    manifest = exports.qwenworkManifest;
+    injectedClient;
+    client;
+    adapter;
+    configuredCdpPort;
+    activeCdpPort;
+    configuredHttpPort;
+    disposers = [];
+    requestQueue = Promise.resolve();
+    state = "stopped";
+    message = "";
+    constructor(options = {}) {
+        this.injectedClient = options.client;
+        this.configuredCdpPort = options.cdpPort;
+        this.configuredHttpPort = options.httpPort;
+    }
+    async activate(context) {
+        if (this.state === "ready")
+            return;
+        this.state = "starting";
+        try {
+            const config = context.config;
+            this.activeCdpPort = asNumber(config.cdpPort, this.configuredCdpPort ?? exports.qwenworkManifest.defaults.cdpPort);
+            const client = this.injectedClient ?? new cdp_client_1.QwenCdpClient({
+                port: this.activeCdpPort,
+                subChatId: typeof config.subChatId === "string" ? config.subChatId : undefined,
+                chatId: typeof config.chatId === "string" ? config.chatId : undefined,
+                workspace: typeof config.workspace === "string" ? config.workspace : undefined,
+            });
+            this.client = client;
+            await client.connect();
+            this.adapter = new openai_adapter_1.OpenAIAdapter((prompt, signal) => this.enqueue(() => client.direct(prompt, asNumber(config.timeoutMs, 240_000), signal)), {
+                onLog: (message) => context.logger?.info(message),
+                onExchange: (direction, text, peer, agentAid) => context.logger?.info("千问通信", { direction, text, peer, agentAid }),
+            });
+            if (context.http) {
+                this.registerRoutes(context.http);
+            }
+            else {
+                this.configuredHttpPort = asNumber(config.httpPort, this.configuredHttpPort ?? exports.qwenworkManifest.defaults.httpPort);
+                if (this.configuredHttpPort > 0)
+                    await this.adapter.listen(this.configuredHttpPort);
+            }
+            this.state = "ready";
+            this.message = "";
+        }
+        catch (error) {
+            this.state = "failed";
+            this.message = error instanceof Error ? error.message : String(error);
+            context.logger?.error(this.message);
+            await this.cleanup();
+            throw error;
+        }
+    }
+    async deactivate(_reason) {
+        await this.cleanup();
+        this.state = "stopped";
+    }
+    async health() {
+        const details = { cdpPort: this.activeCdpPort ?? exports.qwenworkManifest.defaults.cdpPort, httpPort: this.configuredHttpPort ?? exports.qwenworkManifest.defaults.httpPort };
+        if (this.state === "ready" && this.client && !this.client.connected())
+            return { status: "degraded", message: "千问 CDP 已断开", details };
+        return { status: this.state, ...(this.message ? { message: this.message } : {}), details };
+    }
+    async handle(request) {
+        if (this.state !== "ready" || !this.client)
+            throw new Error("qwenwork 插件未就绪");
+        const text = await this.enqueue(() => this.client.direct(request.task, request.timeoutMs ?? 240_000));
+        return { text };
+    }
+    registerRoutes(http) {
+        if (!this.adapter)
+            return;
+        const handler = (req, res) => this.adapter.handle(req, res);
+        for (const [method, path] of [["GET", "/v1/models"], ["POST", "/v1/chat/completions"], ["OPTIONS", "/v1/chat/completions"]]) {
+            const disposer = http.register({ method, path, handler });
+            if (typeof disposer === "function")
+                this.disposers.push(disposer);
+        }
+    }
+    async cleanup() {
+        for (const dispose of this.disposers.splice(0))
+            dispose();
+        await this.adapter?.close();
+        this.adapter = undefined;
+        this.client?.close();
+        this.client = undefined;
+    }
+    enqueue(operation) {
+        const run = this.requestQueue.then(operation, operation);
+        this.requestQueue = run.then(() => undefined, () => undefined);
+        return run;
+    }
+}
+exports.QwenworkPlugin = QwenworkPlugin;
+function createQwenworkPlugin(options = {}) {
+    return new QwenworkPlugin(options);
+}
+exports.default = createQwenworkPlugin;

package/dist/src-ts/plugins/qwenwork/openai-adapter.js

@@ -0,0 +1,146 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.OpenAIAdapter = void 0;
+exports.messageText = messageText;
+const node_http_1 = require("node:http");
+const node_crypto_1 = require("node:crypto");
+function json(res, status, value) {
+    const body = JSON.stringify(value);
+    res.statusCode = status;
+    res.setHeader("Content-Type", "application/json; charset=utf-8");
+    res.setHeader("Content-Length", Buffer.byteLength(body));
+    res.end(body);
+}
+function messageText(content) {
+    if (typeof content === "string")
+        return content;
+    if (Array.isArray(content)) {
+        return content.map((part) => {
+            if (typeof part === "string")
+                return part;
+            if (part && typeof part === "object" && "text" in part)
+                return String(part.text ?? "");
+            return "";
+        }).join("");
+    }
+    return content == null ? "" : String(content);
+}
+async function readBody(req) {
+    const chunks = [];
+    let size = 0;
+    for await (const chunk of req) {
+        const data = Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk);
+        size += data.length;
+        if (size > 2 * 1024 * 1024)
+            throw new Error("请求体超过 2 MiB");
+        chunks.push(data);
+    }
+    return Buffer.concat(chunks).toString("utf8");
+}
+/** OpenAI-compatible HTTP adapter used by qwenwork and the platform router. */
+class OpenAIAdapter {
+    complete;
+    model;
+    onLog;
+    onExchange;
+    prefix;
+    server;
+    constructor(complete, options = {}) {
+        this.complete = complete;
+        this.model = options.model ?? "qwenwork";
+        this.onLog = options.onLog;
+        this.onExchange = options.onExchange;
+        this.prefix = (options.prefix ?? "/v1").replace(/\/+$/, "") || "/v1";
+    }
+    models() {
+        return { object: "list", data: [{ id: this.model, object: "model", owned_by: "qwenwork" }] };
+    }
+    async handle(req, res) {
+        const requestPath = new URL(req.url ?? "/", "http://127.0.0.1").pathname;
+        res.setHeader("Access-Control-Allow-Origin", "*");
+        if (req.method === "OPTIONS") {
+            res.setHeader("Access-Control-Allow-Headers", "authorization, content-type");
+            res.statusCode = 204;
+            res.end();
+            return;
+        }
+        if (req.method === "GET" && requestPath === `${this.prefix}/models`) {
+            json(res, 200, this.models());
+            return;
+        }
+        if (req.method !== "POST" || requestPath !== `${this.prefix}/chat/completions`) {
+            json(res, 404, { error: { message: "Not found", type: "invalid_request_error" } });
+            return;
+        }
+        const abort = new AbortController();
+        const peer = req.headers["x-aun-peer"];
+        const peerId = typeof peer === "string" && peer.length <= 256 ? peer : undefined;
+        const receivingAid = req.headers["x-evolcore-agent-aid"];
+        const agentAid = typeof receivingAid === "string" && receivingAid.length <= 256 ? receivingAid : undefined;
+        const onClose = () => { if (!res.writableEnded)
+            abort.abort(); };
+        res.once("close", onClose);
+        try {
+            const payload = JSON.parse(await readBody(req));
+            const lastUser = [...(payload.messages ?? [])].reverse().find((message) => message.role === "user");
+            const prompt = messageText(lastUser?.content);
+            if (!prompt.trim()) {
+                json(res, 400, { error: { message: "messages 中没有 user 内容", type: "invalid_request_error" } });
+                return;
+            }
+            this.onLog?.(`chat request model=${payload.model ?? this.model} stream=${Boolean(payload.stream)}`);
+            this.onExchange?.("request", prompt, peerId, agentAid);
+            const content = await this.complete(prompt, abort.signal);
+            this.onExchange?.("reply", content, peerId, agentAid);
+            if (abort.signal.aborted || res.destroyed)
+                return;
+            const now = Math.floor(Date.now() / 1000);
+            const id = `chatcmpl-${(0, node_crypto_1.randomUUID)()}`;
+            const model = payload.model ?? this.model;
+            if (payload.stream) {
+                res.statusCode = 200;
+                res.setHeader("Content-Type", "text/event-stream; charset=utf-8");
+                res.setHeader("Cache-Control", "no-cache");
+                res.setHeader("Connection", "keep-alive");
+                res.write(`data: ${JSON.stringify({ id, object: "chat.completion.chunk", created: now, model, choices: [{ index: 0, delta: { role: "assistant", content }, finish_reason: null }] })}\n\n`);
+                res.write(`data: ${JSON.stringify({ id, object: "chat.completion.chunk", created: now, model, choices: [{ index: 0, delta: {}, finish_reason: "stop" }] })}\n\n`);
+                res.end("data: [DONE]\n\n");
+                return;
+            }
+            json(res, 200, {
+                id, object: "chat.completion", created: now, model,
+                choices: [{ index: 0, message: { role: "assistant", content }, finish_reason: "stop" }],
+                usage: { prompt_tokens: 0, completion_tokens: 0, total_tokens: 0 },
+            });
+        }
+        catch (error) {
+            this.onExchange?.("error", error instanceof Error ? error.message : String(error), peerId, agentAid);
+            if (abort.signal.aborted || res.destroyed)
+                return;
+            json(res, 500, { error: { message: error instanceof Error ? error.message : String(error), type: "server_error" } });
+        }
+        finally {
+            res.off("close", onClose);
+        }
+    }
+    createServer() {
+        return (0, node_http_1.createServer)((req, res) => void this.handle(req, res));
+    }
+    async listen(port = 17893, host = "127.0.0.1") {
+        if (this.server)
+            return;
+        this.server = this.createServer();
+        await new Promise((resolve, reject) => {
+            this.server.once("error", reject);
+            this.server.listen(port, host, () => resolve());
+        });
+    }
+    async close() {
+        const server = this.server;
+        this.server = undefined;
+        if (!server)
+            return;
+        await new Promise((resolve) => server.close(() => resolve()));
+    }
+}
+exports.OpenAIAdapter = OpenAIAdapter;

package/package.json

@@ -1,6 +1,23 @@
 {
   "name": "agent-bridge-qwenwork",
-  "version": "0.0.0-stage",
-  "stub": true,
-  "description": "Temporary package placeholder for staged publishing"
-}
+  "version": "1.0.0",
+  "description": "agent-bridge 的千问办公插件。",
+  "repository": { "type": "git", "url": "https://codeup.aliyun.com/6808da56fb3da14b899afd54/agent_bridge.git", "directory": "packages/qwenwork" },
+  "main": "dist/src-ts/plugins/qwenwork/index.js",
+  "files": [
+    "dist/",
+    "plugin.json",
+    "config.schema.json",
+    "README.md"
+  ],
+  "agentBridge": {
+    "plugin": "./plugin.json"
+  },
+  "publishConfig": {
+    "access": "public",
+    "registry": "https://registry.npmjs.org"
+  },
+  "engines": {
+    "node": ">=22"
+  }
+}

package/plugin.json

@@ -0,0 +1,10 @@
+{
+  "id": "qwenwork",
+  "version": "1.0.0",
+  "displayName": "千问办公",
+  "entry": "dist/src-ts/plugins/qwenwork/index.js",
+  "apiVersion": "1",
+  "process": "in-process",
+  "capabilities": ["chat", "openai-compatible", "health"],
+  "configSchema": "config.schema.json"
+}