@vectorx/agent-simulator 1.0.0 → 1.1.0

package/lib/services/agent.service.js

@@ -16,6 +16,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AgentService = void 0;
+const endpoints_1 = require("@vectorx/endpoints");
 const form_data_1 = __importDefault(require("form-data"));
 const inversify_1 = require("inversify");
 const node_fetch_1 = __importDefault(require("node-fetch"));
@@ -24,23 +25,30 @@ const url_parser_1 = require("../utils/url-parser");
 let AgentService = class AgentService {
     constructor(context) {
         this.parsedServerUrl = (0, url_parser_1.parseAgentServerUrl)(context.agentServerUrl);
+        this.agentId = context.agentId;
     }
     /**
-     * 获取Agent信息
+     * 获取 Agent 信息。
+     * 直接请求 agentbase `/agent/info`，不经过开发者 Agent Server（rcb-ff），
+     * 避免开发者代码异常导致获取失败。
      */
     async getAgentInfo() {
-        const healthUrl = `${this.parsedServerUrl.fullUrl}/info`;
-        const response = await (0, node_fetch_1.default)(healthUrl);
+        const agentInfoUrl = `${(0, endpoints_1.getAgentbaseBaseUrl)()}/agent/info`;
+        const response = await (0, node_fetch_1.default)(agentInfoUrl, {
+            headers: {
+                ...(this.agentId ? { "agent-id": this.agentId } : {}),
+            },
+        });
         if (!response.ok) {
-            throw new Error(`无法连接到后端服务: ${response.status} ${response.statusText}`);
+            throw new Error(`agentbase 响应异常: ${response.status} ${response.statusText}`);
         }
         const resData = await response.json();
         if (resData.code !== 0) {
-            throw new Error(`后端服务返回错误: ${resData.message || "未知错误"}`);
+            throw new Error(`agentbase 返回错误: ${resData.message || "未知错误"}`);
         }
         const { agent_info } = resData.data;
         if (!agent_info) {
-            throw new Error("后端服务返回的数据中缺少 agent_info 字段");
+            throw new Error("agentbase 返回数据异常：缺少 agent_info 字段");
         }
         return this.transformToAgentSimulatorInfo(agent_info);
     }
@@ -53,7 +61,10 @@ let AgentService = class AgentService {
             const healthUrl = `${this.parsedServerUrl.fullUrl}/health`;
             const response = await (0, node_fetch_1.default)(healthUrl);
             if (response.ok) {
-                const healthData = await response.json();
+                const contentType = response.headers.get("content-type");
+                const healthData = contentType?.includes("application/json") && response.status !== 204
+                    ? await response.json()
+                    : null;
                 return {
                     status: "normal",
                     health: healthData,
@@ -94,6 +105,22 @@ let AgentService = class AgentService {
             };
         }
     }
+    /**
+     * 获取函数能力元数据（用于检测函数是否支持 WebSocket 等特性）
+     */
+    async getFunctionMeta() {
+        try {
+            const metaUrl = `${this.parsedServerUrl.baseUrl}/@meta`;
+            const response = await (0, node_fetch_1.default)(metaUrl);
+            if (!response.ok) {
+                return { functions: [] };
+            }
+            return (await response.json());
+        }
+        catch {
+            return { functions: [] };
+        }
+    }
     /**
      * 上传文件到云函数
      * 使用 multipart/form-data 格式传输，对齐 OpenAI 文件上传标准
@@ -144,4 +171,3 @@ exports.AgentService = AgentService = __decorate([
     __param(0, (0, inversify_1.inject)(container_1.APP_CONTEXT)),
     __metadata("design:paramtypes", [Object])
 ], AgentService);
-//# sourceMappingURL=agent.service.js.map

package/lib/services/auth.service.d.ts

@@ -46,10 +46,9 @@ export declare class AuthService implements IAuthService {
     /**
      * 检查登录状态（调用下游服务）
      */
-    check(params: {
+    check(_params: {
         secret_id: string;
         nonce: string;
         cli_token: string;
     }): Promise<any>;
 }
-//# sourceMappingURL=auth.service.d.ts.map

package/lib/services/auth.service.js

@@ -43,9 +43,9 @@ var __metadata = (this && this.__metadata) || function (k, v) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AuthService = void 0;
-const fs = __importStar(require("fs"));
-const os = __importStar(require("os"));
-const path = __importStar(require("path"));
+const fs = __importStar(require("node:fs"));
+const os = __importStar(require("node:os"));
+const path = __importStar(require("node:path"));
 const inversify_1 = require("inversify");
 const env_helper_1 = require("../utils/env-helper");
 /**
@@ -88,7 +88,7 @@ let AuthService = class AuthService {
             }
             return loginInfo;
         }
-        catch (error) {
+        catch (_error) {
             return null;
         }
     }
@@ -124,7 +124,7 @@ let AuthService = class AuthService {
     /**
      * 检查登录状态（调用下游服务）
      */
-    async check(params) {
+    async check(_params) {
         // 这里可以调用下游服务进行验证
         // 暂时返回模拟数据
         return {
@@ -138,4 +138,3 @@ exports.AuthService = AuthService = __decorate([
     (0, inversify_1.injectable)(),
     __metadata("design:paramtypes", [])
 ], AuthService);
-//# sourceMappingURL=auth.service.js.map

package/lib/services/simulator.service.d.ts

@@ -20,4 +20,3 @@ export declare class SimulatorService implements ISimulatorService {
      */
     updateAgentStatus(status: "normal" | "stopped" | "error"): void;
 }
-//# sourceMappingURL=simulator.service.d.ts.map

package/lib/services/simulator.service.js

@@ -46,4 +46,3 @@ exports.SimulatorService = SimulatorService = __decorate([
     __param(0, (0, inversify_1.inject)(container_1.APP_CONTEXT)),
     __metadata("design:paramtypes", [Object])
 ], SimulatorService);
-//# sourceMappingURL=simulator.service.js.map

package/lib/services/upload.service.d.ts

@@ -50,4 +50,3 @@ export declare class UploadService implements IUploadService {
         scene: string;
     }): Promise<TokenResponse>;
 }
-//# sourceMappingURL=upload.service.d.ts.map

package/lib/services/upload.service.js

@@ -122,4 +122,3 @@ exports.UploadService = UploadService = __decorate([
     __param(0, (0, inversify_1.inject)(identifiers_1.SERVICE_IDENTIFIERS.AuthService)),
     __metadata("design:paramtypes", [Object])
 ], UploadService);
-//# sourceMappingURL=upload.service.js.map

package/lib/simulator.d.ts

@@ -2,6 +2,10 @@ export interface SimulatorOptions {
     port: number;
     agentServerUrl: string;
     open?: boolean;
+    /** 静默模式：不输出启动日志（用于 useIde 等场景由调用方统一输出） */
+    quiet?: boolean;
+    /** Agent ID（来自 project.config.json），用于直接向 agentbase 查询 Agent 信息 */
+    agentId?: string;
 }
 /**
  * Agent模拟器实现
@@ -19,9 +23,15 @@ export declare class AgentSimulator {
      * 配置并启动模拟器
      */
     start(): Promise<void>;
+    /**
+     * 设置 WebSocket 代理：将 /ws-proxy/* 请求透传到 rcb-ff
+     *
+     * 实现策略：先完成浏览器侧握手，再连接上游 rcb-ff。
+     * 上游连接成功前，客户端消息暂存在队列中，连接就绪后逐条发出。
+     */
+    private setupWebSocketProxy;
     /**
      * 停止模拟器
      */
     stop(): Promise<void>;
 }
-//# sourceMappingURL=simulator.d.ts.map

package/lib/simulator.js

@@ -1,8 +1,10 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AgentSimulator = void 0;
+const ws_1 = require("ws");
 const app_1 = require("./app");
 const logger_1 = require("./utils/logger");
+const url_parser_1 = require("./utils/url-parser");
 /**
  * Agent模拟器实现
  */
@@ -21,6 +23,7 @@ class AgentSimulator {
             port: options.port,
             agentServerUrl: options.agentServerUrl,
             open: options.open,
+            agentId: options.agentId,
         });
     }
     /**
@@ -32,7 +35,9 @@ class AgentSimulator {
         // 启动服务器
         return new Promise((resolve) => {
             this.server = expressApp.listen(config.port, () => {
-                logger_1.Logger.success(`
+                this.setupWebSocketProxy();
+                if (!this.options.quiet) {
+                    logger_1.Logger.success(`
 ┌────────────────────────────────────────────────────┐
 │                                                    │
 │  🚀 Agent模拟器已启动!                             │
@@ -41,15 +46,95 @@ class AgentSimulator {
 │  🔄 Agent服务: ${config.agentServerUrl}      │
 │                                                    │
 └────────────────────────────────────────────────────┘
-        `);
-                // 自动打开浏览器提示
-                if (config.open) {
-                    logger_1.Logger.warning(`请手动在浏览器中打开: ${logger_1.Logger.info(`http://localhost:${config.port}`)}`);
+          `);
+                    // 自动打开浏览器提示
+                    if (config.open) {
+                        logger_1.Logger.warning(`请手动在浏览器中打开: ${logger_1.Logger.info(`http://localhost:${config.port}`)}`);
+                    }
                 }
                 resolve();
             });
         });
     }
+    /**
+     * 设置 WebSocket 代理：将 /ws-proxy/* 请求透传到 rcb-ff
+     *
+     * 实现策略：先完成浏览器侧握手，再连接上游 rcb-ff。
+     * 上游连接成功前，客户端消息暂存在队列中，连接就绪后逐条发出。
+     */
+    setupWebSocketProxy() {
+        if (!this.server)
+            return;
+        const parsed = (0, url_parser_1.parseAgentServerUrl)(this.options.agentServerUrl);
+        const rcbWsBase = `ws://${parsed.hostname}:${parsed.port}`;
+        const proxyWss = new ws_1.WebSocketServer({ noServer: true });
+        this.server.on("upgrade", (req, socket, head) => {
+            if (!req.url?.startsWith("/ws-proxy/"))
+                return;
+            const targetPath = req.url.replace("/ws-proxy", "");
+            const targetUrl = `${rcbWsBase}${targetPath}`;
+            proxyWss.handleUpgrade(req, socket, head, (clientWs) => {
+                const pendingMessages = [];
+                let upstreamReady = false;
+                const upstream = new ws_1.WebSocket(targetUrl, {
+                    headers: { origin: req.headers.origin || "" },
+                });
+                upstream.on("open", () => {
+                    upstreamReady = true;
+                    console.log(`[ws-proxy] upstream connected to ${targetUrl}`);
+                    for (const msg of pendingMessages) {
+                        upstream.send(msg.data, { binary: msg.isBinary });
+                    }
+                    pendingMessages.length = 0;
+                });
+                upstream.on("message", (data, isBinary) => {
+                    console.log(`[ws-proxy] upstream->client: ${String(Buffer.isBuffer(data) ? data.toString() : data).slice(0, 100)}`);
+                    if (clientWs.readyState === ws_1.WebSocket.OPEN) {
+                        clientWs.send(data, { binary: isBinary });
+                    }
+                });
+                clientWs.on("message", (data, isBinary) => {
+                    console.log(`[ws-proxy] client->upstream: ready=${upstreamReady} state=${upstream.readyState} data=${String(Buffer.isBuffer(data) ? data.toString() : data).slice(0, 100)}`);
+                    if (upstreamReady && upstream.readyState === ws_1.WebSocket.OPEN) {
+                        upstream.send(data, { binary: isBinary });
+                    }
+                    else {
+                        pendingMessages.push({ data, isBinary });
+                    }
+                });
+                const safeClose = (ws, code, reason) => {
+                    try {
+                        if (ws.readyState === ws_1.WebSocket.OPEN) {
+                            const safeCode = code && code >= 1000 && code <= 4999 ? code : 1000;
+                            ws.close(safeCode, reason || "");
+                        }
+                    }
+                    catch {
+                        try {
+                            ws.terminate();
+                        }
+                        catch {
+                            /* ignore */
+                        }
+                    }
+                };
+                upstream.on("close", (code, reason) => {
+                    safeClose(clientWs, code, reason?.toString());
+                });
+                clientWs.on("close", (code, reason) => {
+                    safeClose(upstream, code, reason?.toString());
+                });
+                upstream.on("error", (err) => {
+                    console.error(`[ws-proxy] upstream error:`, err?.message || err);
+                    safeClose(clientWs, 1011, "upstream error");
+                });
+                clientWs.on("error", (err) => {
+                    console.error(`[ws-proxy] client error:`, err?.message || err);
+                    safeClose(upstream, 1011, "client error");
+                });
+            });
+        });
+    }
     /**
      * 停止模拟器
      */
@@ -76,4 +161,3 @@ class AgentSimulator {
     }
 }
 exports.AgentSimulator = AgentSimulator;
-//# sourceMappingURL=simulator.js.map

package/lib/utils/env-helper.d.ts

@@ -3,4 +3,3 @@
  * 判断当前是否运行在 IDE 环境中
  */
 export declare const isIDE: boolean;
-//# sourceMappingURL=env-helper.d.ts.map

package/lib/utils/env-helper.js

@@ -6,4 +6,3 @@ exports.isIDE = void 0;
  * 判断当前是否运行在 IDE 环境中
  */
 exports.isIDE = process.env.FROM_SOURCE === "IDE";
-//# sourceMappingURL=env-helper.js.map

package/lib/utils/logger.d.ts

@@ -5,4 +5,3 @@ export declare class Logger {
     static error(message: string, error?: any): void;
     static debug(message: string, data?: any): void;
 }
-//# sourceMappingURL=logger.d.ts.map

package/lib/utils/logger.js

@@ -25,4 +25,3 @@ class Logger {
     }
 }
 exports.Logger = Logger;
-//# sourceMappingURL=logger.js.map

package/lib/utils/request.d.ts

@@ -8,4 +8,3 @@ export declare const request: {
         url: string;
     } & FetchOptions): Promise<any>;
 };
-//# sourceMappingURL=request.d.ts.map

package/lib/utils/request.js

@@ -28,9 +28,13 @@ exports.request = {
                 body = JSON.stringify(data);
             }
         }
-        const res = await (0, node_fetch_1.default)(finalUrl, { method, headers: finalHeaders, body: body, timeout: 1000 * 20 });
+        const res = await (0, node_fetch_1.default)(finalUrl, {
+            method,
+            headers: finalHeaders,
+            body: body,
+            timeout: 1000 * 20,
+        });
         const json = await res.json().catch(() => ({}));
         return { status: res.status, ok: res.ok, data: json };
     },
 };
-//# sourceMappingURL=request.js.map

package/lib/utils/url-parser.d.ts

@@ -11,4 +11,3 @@ export declare function parseAgentServerUrl(url: string): {
     baseUrl: string;
     fullUrl: string;
 };
-//# sourceMappingURL=url-parser.d.ts.map

package/lib/utils/url-parser.js

@@ -18,8 +18,7 @@ function parseAgentServerUrl(url) {
             fullUrl: url,
         };
     }
-    catch (error) {
+    catch (_error) {
         throw new Error(`无效的 Agent Server URL: ${url}`);
     }
 }
-//# sourceMappingURL=url-parser.js.map

package/lib/utils.d.ts

@@ -3,4 +3,3 @@
  * @returns 唯一ID字符串
  */
 export declare const generateId: () => string;
-//# sourceMappingURL=utils.d.ts.map

package/lib/utils.js

@@ -9,4 +9,3 @@ const generateId = () => {
     return Date.now() + "-" + Math.random().toString(36).substr(2, 9);
 };
 exports.generateId = generateId;
-//# sourceMappingURL=utils.js.map

package/lib/webview-ui/Agent.d.ts

@@ -2,4 +2,3 @@ import type { AgentSimulatorInfo } from "./services/agent-service";
 export declare function getSendMessageUri(agentSimulatorInfo: AgentSimulatorInfo): string;
 export declare function getHealthUri(agentSimulatorInfo: AgentSimulatorInfo): string;
 export declare function getQueryTasksUri(agentSimulatorInfo: AgentSimulatorInfo, taskId?: string): string;
-//# sourceMappingURL=Agent.d.ts.map

package/lib/webview-ui/Agent.js

@@ -21,4 +21,3 @@ function getQueryTasksUri(agentSimulatorInfo, taskId) {
     }
     return `${agentSimulatorInfo.agentServerUrl}/query-tasks${taskId ? `/${taskId}` : ""}`;
 }
-//# sourceMappingURL=Agent.js.map

package/lib/webview-ui/App.d.ts

@@ -1,3 +1,2 @@
 declare const CopilotSimulator: () => import("react/jsx-runtime").JSX.Element;
 export default CopilotSimulator;
-//# sourceMappingURL=App.d.ts.map