@vectorx/functions-framework 0.0.0-beta-20260109055258 → 0.0.0-beta-20260320063305

package/lib/middlewares/middle-env-vars-injection.js

@@ -0,0 +1,65 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.envVarsInjectionMiddleware = envVarsInjectionMiddleware;
+function envVarsInjectionMiddleware() {
+    return (ctx, next) => __awaiter(this, void 0, void 0, function* () {
+        var _a, _b, _c, _d, _e, _f, _g, _h;
+        const envVarsHeader = (_b = (_a = ctx.request) === null || _a === void 0 ? void 0 : _a.headers) === null || _b === void 0 ? void 0 : _b["x-ide-env-vars"];
+        if (envVarsHeader && typeof envVarsHeader === "string") {
+            try {
+                const envVars = JSON.parse(envVarsHeader);
+                if (envVars && typeof envVars === "object") {
+                    for (const [key, value] of Object.entries(envVars)) {
+                        if (key && value !== undefined && value !== null) {
+                            process.env[String(key)] = String(value);
+                        }
+                    }
+                }
+            }
+            catch (e) {
+                console.warn("[envVarsInjection] Failed to parse env vars from header:", e);
+            }
+        }
+        const systemHeadersMapping = {
+            "xhs-cloud-env-id": "XHS_CLOUD_ENV_ID",
+            "xhs-cloud-function-id": "XHS_CLOUD_FUNCTION_ID",
+            "xhs-app-id": "XHS_APP_ID",
+        };
+        for (const [headerKey, envKey] of Object.entries(systemHeadersMapping)) {
+            const headerValue = (_d = (_c = ctx.request) === null || _c === void 0 ? void 0 : _c.headers) === null || _d === void 0 ? void 0 : _d[headerKey];
+            if (headerValue && typeof headerValue === "string") {
+                process.env[envKey] = headerValue;
+            }
+        }
+        const versionMapHeader = (_f = (_e = ctx.request) === null || _e === void 0 ? void 0 : _e.headers) === null || _f === void 0 ? void 0 : _f["xhs-cloud-env-version-map"];
+        const aliyunVersionHeader = (_h = (_g = ctx.request) === null || _g === void 0 ? void 0 : _g.headers) === null || _h === void 0 ? void 0 : _h["x-fc-version-id"];
+        if (versionMapHeader &&
+            typeof versionMapHeader === "string" &&
+            aliyunVersionHeader &&
+            typeof aliyunVersionHeader === "string") {
+            try {
+                const parsed = JSON.parse(versionMapHeader);
+                if (parsed && typeof parsed === "object" && !Array.isArray(parsed)) {
+                    const map = parsed;
+                    const mapped = map[aliyunVersionHeader];
+                    if (mapped !== undefined && mapped !== null) {
+                        process.env.XHS_CLOUD_ENV_VERSION = String(mapped);
+                    }
+                }
+            }
+            catch (e) {
+                console.warn("[envVarsInjection] Failed to parse xhs-cloud-env-version-map:", e);
+            }
+        }
+        yield next();
+    });
+}

package/lib/middlewares/middle-open-gw-request.js

@@ -12,9 +12,16 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.openGwRequestMiddleware = openGwRequestMiddleware;
 const error_1 = require("../error");
 const unified_responder_1 = require("../unified-responder");
-function openGwRequestMiddleware(agentId) {
+function openGwRequestMiddleware(agentId, options) {
     return (ctx, next) => __awaiter(this, void 0, void 0, function* () {
+        if ((options === null || options === void 0 ? void 0 : options.mode) === "fun") {
+            yield next();
+            return;
+        }
         if (process.env.OPEN_PLATFORM_STAGE === "production") {
+            if (!agentId) {
+                return (0, unified_responder_1.sendResponse)(ctx, undefined, (0, error_1.newSysErr)("agentId is required in agent mode"), 500);
+            }
             const openId = ctx.headers["open-id"];
             if (!openId) {
                 return (0, unified_responder_1.sendResponse)(ctx, undefined, (0, error_1.newBadRequestErr)("open-id is required"), 401);

package/lib/request.js

@@ -78,7 +78,7 @@ class Request {
     }
     request(options_1) {
         return __awaiter(this, arguments, void 0, function* (options, enableAbort = false) {
-            const { url, headers: _headers = {}, data, responseType, withCredentials, body, method: _method } = options, headers = Object.assign(Object.assign({}, this.defaultHeaders), (0, helper_1.obj2StrRecord)(_headers)), method = String(_method).toLowerCase() || "get";
+            const { url, headers: _headers = {}, data, responseType, withCredentials, body, method: _method, } = options, headers = Object.assign(Object.assign({}, this.defaultHeaders), (0, helper_1.obj2StrRecord)(_headers)), method = String(_method).toLowerCase() || "get";
             let queryParams = {};
             if (method === "get" || method === "head") {
                 queryParams = Object.assign(Object.assign({}, (data || {})), (body && typeof body === "object" ? body : {}));
@@ -151,7 +151,11 @@ class Request {
                 return Promise.reject(x);
             });
             const ret = {
-                data: stream ? (res.body ? stream_1.Readable.toWeb(res.body) : res.body) : yield res.json(),
+                data: stream
+                    ? res.body
+                        ? stream_1.Readable.toWeb(res.body)
+                        : res.body
+                    : yield res.json(),
                 statusCode: res.status,
                 header: res.headers,
             };

package/lib/router.js

@@ -10,6 +10,9 @@ class Router {
         this.router = (0, radix3_1.createRouter)({
             strictTrailingSlash: false,
         });
+        if (!(config && Array.isArray(config.routes))) {
+            return;
+        }
         for (const route of config.routes) {
             if (route.path) {
                 const p = route.path.endsWith("/") ? route.path : route.path + "/";
@@ -40,6 +43,9 @@ class Router {
 }
 exports.Router = Router;
 function routeFunction(router, path) {
+    if (!router) {
+        return {};
+    }
     const matchedRoute = router.findRoute(path);
     if (matchedRoute) {
         return {

package/lib/server.js

@@ -17,8 +17,10 @@ const http_1 = __importDefault(require("http"));
 const koa_1 = __importDefault(require("koa"));
 const koa_body_1 = require("koa-body");
 const error_1 = require("./error");
+const function_registry_1 = require("./function-registry");
 const middlewares_1 = require("./middlewares");
 const unified_responder_1 = require("./unified-responder");
+const wss_1 = require("./wss");
 function corsMiddleware() {
     return (ctx, next) => __awaiter(this, void 0, void 0, function* () {
         ctx.set("Access-Control-Allow-Origin", "*");
@@ -35,8 +37,22 @@ function corsMiddleware() {
 function createServer(options, router, projectConfig) {
     const app = new koa_1.default();
     app.use(corsMiddleware());
+    app.use((ctx, next) => __awaiter(this, void 0, void 0, function* () {
+        if (ctx.path === "/@meta") {
+            const functions = (0, function_registry_1.getAllRegisteredFunctions)().map((fn) => ({
+                name: fn.name,
+                features: {
+                    websocket: typeof fn.userFunction.handleUpgrade === "function",
+                },
+            }));
+            ctx.body = { functions };
+            return;
+        }
+        yield next();
+    }));
     app.use((0, middlewares_1.logsQueryMiddleware)());
     app.use((0, middlewares_1.contextInjectionMiddleware)(options, projectConfig));
+    app.use((0, middlewares_1.envVarsInjectionMiddleware)());
     app.use((0, middlewares_1.eventIdMiddleware)());
     app.use((0, middlewares_1.apmMiddleware)());
     app.use((0, middlewares_1.asyncContextMiddleware)());
@@ -72,8 +88,10 @@ function createServer(options, router, projectConfig) {
             return (0, unified_responder_1.sendResponse)(ctx, undefined, (0, error_1.newBadRequestErr)(err), 400);
         },
     }));
-    app.use((0, middlewares_1.openGwRequestMiddleware)(projectConfig.agentId));
+    const openGwAgentId = options.mode === "fun" ? undefined : projectConfig.agentId;
+    app.use((0, middlewares_1.openGwRequestMiddleware)(openGwAgentId, { mode: options.mode }));
     app.use((0, middlewares_1.functionRouteMiddleware)(router, projectConfig));
     const httpServer = http_1.default.createServer(app.callback());
+    (0, wss_1.createWebSocketServer)(httpServer, app, router, options, projectConfig);
     return httpServer;
 }

package/lib/sse.js

@@ -90,7 +90,9 @@ class ServerSentEvent extends events_1.EventEmitter {
             const openId = ctx.headers["open-id"] || undefined;
             const eventID = (_d = ctx.state) === null || _d === void 0 ? void 0 : _d.eventID;
             const agentInfo = ((_e = ctx.state) === null || _e === void 0 ? void 0 : _e.agentInfo) || {};
-            const name = [(agentInfo === null || agentInfo === void 0 ? void 0 : agentInfo.agentName) || (agentInfo === null || agentInfo === void 0 ? void 0 : agentInfo.name), agentInfo === null || agentInfo === void 0 ? void 0 : agentInfo.version].filter(Boolean).join("@");
+            const name = [(agentInfo === null || agentInfo === void 0 ? void 0 : agentInfo.agentName) || (agentInfo === null || agentInfo === void 0 ? void 0 : agentInfo.name), agentInfo === null || agentInfo === void 0 ? void 0 : agentInfo.version]
+                .filter(Boolean)
+                .join("@");
             const metadata = {
                 http: { method: ctx.method, path: ctx.path },
                 eventID,
@@ -108,7 +110,11 @@ class ServerSentEvent extends events_1.EventEmitter {
             });
             if (trace) {
                 ctx.state.langfuseTrace = trace;
-                (_g = trace.event) === null || _g === void 0 ? void 0 : _g.call(trace, { name: "sse:start", input: (_j = (_h = ctx.state) === null || _h === void 0 ? void 0 : _h.event) !== null && _j !== void 0 ? _j : null, metadata: { path: ctx.path } });
+                (_g = trace.event) === null || _g === void 0 ? void 0 : _g.call(trace, {
+                    name: "sse:start",
+                    input: (_j = (_h = ctx.state) === null || _h === void 0 ? void 0 : _h.event) !== null && _j !== void 0 ? _j : null,
+                    metadata: { path: ctx.path },
+                });
             }
         }
         catch (_k) { }

package/lib/telemetry/langfuse.js

@@ -6,8 +6,10 @@ const secretKey = process.env.LANGFUSE_SECRET_KEY;
 const publicKey = process.env.LANGFUSE_PUBLIC_KEY;
 const baseUrl = process.env.LANGFUSE_BASEURL;
 exports.langfuse = new langfuse_1.Langfuse({
-    secretKey: "sk-lf-f389bb1e-24db-4b42-b506-8d6118f45044",
-    publicKey: "pk-lf-ee301553-7a64-4f0f-bf6b-a2ac1e5ca2df",
-    baseUrl: "https://xray-langfuse.devops.xiaohongshu.com",
+    secretKey,
+    publicKey,
+    baseUrl,
+    environment: process.env.OPEN_PLATFORM_STAGE || "development",
+    enabled: Boolean(secretKey && publicKey && baseUrl),
 });
 exports.default = exports.langfuse;

package/lib/utils/apm.config.js

@@ -1,7 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.defaultConfig = exports.APM_MEASUREMENT_NAME = void 0;
+const endpoints_1 = require("@vectorx/endpoints");
 exports.APM_MEASUREMENT_NAME = "vectorx_kit_agent_runtime";
 exports.defaultConfig = {
-    endpoint: "https://apm-fe.xiaohongshu.com/api/data",
+    endpoint: (0, endpoints_1.resolveUrl)("@vectorx/functions-framework", "apmDataEndpoint"),
 };

package/lib/utils/console-intercept.js

@@ -43,11 +43,10 @@ class Interceptor {
             anotherConsole[method] = (message, ...optionalParams) => {
                 if ((0, async_context_1.isInFrameworkAsyncContext)()) {
                     callAddUserCodeLog(method, message, ...optionalParams);
-                    process.stdout.write((0, util_1.formatWithOptions)({ colors: false }, message, ...optionalParams));
+                    process.stdout.write(`${(0, util_1.formatWithOptions)({ colors: false }, message, ...optionalParams)}\n`);
                 }
                 else {
                     originalConsole[method](message, ...optionalParams);
-                    process.stdout.write((0, util_1.formatWithOptions)({ colors: false }, message, ...optionalParams));
                 }
             };
         }

package/lib/utils/helper.js

@@ -13,13 +13,13 @@ exports.getDependencyVersion = getDependencyVersion;
 const fs_1 = __importDefault(require("fs"));
 const path_1 = __importDefault(require("path"));
 function deepFreeze(obj) {
-    if (obj === null || typeof obj !== "object") {
+    if (!isPlainObject(obj)) {
         return obj;
     }
-    Object.keys(obj).forEach((prop) => {
-        const value = obj[prop];
-        if (typeof value === "object" && value !== null) {
-            deepFreeze(value);
+    Object.getOwnPropertyNames(obj).forEach((name) => {
+        const prop = obj[name];
+        if (typeof prop === "object" && prop !== null && !Object.isFrozen(prop)) {
+            deepFreeze(prop);
         }
     });
     return Object.freeze(obj);
@@ -69,7 +69,9 @@ function formatUrl(PROTOCOL, url, query) {
 }
 function getDependencyVersion(packageName, searchPath) {
     try {
-        const packagePath = require.resolve(packageName, { paths: [searchPath] });
+        const packagePath = require.resolve(packageName, {
+            paths: [searchPath],
+        });
         let currentPath = path_1.default.dirname(packagePath);
         let packageJsonPath = path_1.default.join(currentPath, "package.json");
         while (!fs_1.default.existsSync(packageJsonPath) && currentPath !== path_1.default.parse(currentPath).root) {

package/lib/wss.js

@@ -0,0 +1,305 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.VectorxWebSocket = void 0;
+exports.createWebSocketServer = createWebSocketServer;
+const events = __importStar(require("events"));
+const http = __importStar(require("http"));
+const koa_compose_1 = __importDefault(require("koa-compose"));
+const ws_1 = require("ws");
+const error_1 = require("./error");
+const function_registry_1 = require("./function-registry");
+const logger_1 = require("./logger");
+const middle_async_context_1 = require("./middlewares/middle-async-context");
+const middle_common_logger_1 = require("./middlewares/middle-common-logger");
+const middle_context_injection_1 = require("./middlewares/middle-context-injection");
+const middle_event_id_1 = require("./middlewares/middle-event-id");
+const router_1 = require("./router");
+const helper_1 = require("./utils/helper");
+const WS_ERROR_CODE = {
+    SERVICE_RESTART: 1012,
+};
+function respondError(socket, e) {
+    respond(socket, 500, `Internal Server Error: ${e && e.message ? e.message : e}`);
+}
+function respond(socket, statusCode, body, type = "text/plain") {
+    const statusMessage = http.STATUS_CODES[statusCode];
+    const head = [
+        `HTTP/1.1 ${statusCode} ${statusMessage}`,
+        `Content-Type: ${type}`,
+        `Content-Length: ${Buffer.byteLength(Buffer.from(body))}`,
+        "Content-Encoding: utf-8",
+        "Connection: close",
+    ].join("\r\n");
+    socket.write(head);
+    socket.write("\r\n\r\n");
+    socket.end(body);
+}
+function setupWebSocketsHeartbeat(wss, interval = 10000) {
+    const SYM_WS_IS_ALIVE = Symbol("WS_IS_ALIVE");
+    let checking = false;
+    let wsNotAliveCount = 0;
+    function doCheck() {
+        if (checking)
+            return;
+        checking = true;
+        wsNotAliveCount = 0;
+        wss.clients.forEach((ws) => {
+            if (Reflect.has(ws, SYM_WS_IS_ALIVE) === false) {
+                ws[SYM_WS_IS_ALIVE] = true;
+                ws.on("pong", function onPong() {
+                    this[SYM_WS_IS_ALIVE] = true;
+                });
+            }
+            if (ws.readyState !== ws.OPEN)
+                return;
+            if (ws[SYM_WS_IS_ALIVE] === false) {
+                wsNotAliveCount += 1;
+                ws.close(1001, "websocket client not alive");
+                return ws.terminate();
+            }
+            ws[SYM_WS_IS_ALIVE] = false;
+            ws.ping();
+        });
+        logger_1.functionsLogger.logError(`wss heartbeat finish, ws not alive count: ${wsNotAliveCount}, total: ${wss.clients.size}`);
+        checking = false;
+    }
+    const aliveCheckTimer = setInterval(doCheck, interval);
+    wss.once("close", () => {
+        clearInterval(aliveCheckTimer);
+    });
+}
+function setupGracefulShutdown(wss) {
+    for (const signal of ["SIGINT", "SIGTERM"]) {
+        process.on(signal, () => __awaiter(this, void 0, void 0, function* () {
+            logger_1.functionsLogger.logError(`received ${signal} signal, shutting down websocket server, ${wss.clients.size} ws clients need to be closed`);
+            wss.close();
+            wss.clients.forEach((ws) => {
+                ws.close(WS_ERROR_CODE.SERVICE_RESTART, "websocket server is shutting down, please try to reconnect");
+            });
+        }));
+    }
+}
+class VectorxWebSocket extends events.EventEmitter {
+    constructor(ws, eventCtx) {
+        super();
+        this.ws = ws;
+        this.ws.onopen = (openEvent) => {
+            this.emit("open", {
+                ctx: eventCtx,
+                type: openEvent.type,
+            });
+        };
+        this.ws.onerror = (errorEvent) => {
+            this.emit("error", {
+                ctx: eventCtx,
+                type: errorEvent.type,
+                message: errorEvent.message,
+                error: errorEvent.error,
+            });
+        };
+        this.ws.onclose = (closeEvent) => {
+            this.emit("close", {
+                ctx: eventCtx,
+                type: closeEvent.type,
+                code: closeEvent.code,
+                reason: closeEvent.reason,
+                wasClean: closeEvent.wasClean,
+            });
+        };
+        this.ws.onmessage = (messageEvent) => {
+            this.emit("message", {
+                ctx: eventCtx,
+                type: messageEvent.type,
+                data: messageEvent.data,
+            });
+        };
+        Object.defineProperty(this, "ws", {
+            configurable: false,
+            enumerable: false,
+            writable: false,
+        });
+    }
+    send(buf) {
+        this.ws.send(buf);
+    }
+    close(code, reason) {
+        this.ws.close(code, reason);
+    }
+}
+exports.VectorxWebSocket = VectorxWebSocket;
+function createWebSocketServer(httpServer, koaApp, router, options, projectConfig) {
+    const hasExportHandleUpgradeFunctions = [];
+    (0, function_registry_1.getAllRegisteredFunctions)().forEach((fn) => {
+        if (typeof fn.userFunction.handleUpgrade === "function") {
+            hasExportHandleUpgradeFunctions.push(fn.name);
+        }
+    });
+    if (hasExportHandleUpgradeFunctions.length === 0) {
+        return;
+    }
+    process.stdout.write(`[vectorx-ff] WebSocket feature is enabled. Enabled functions: ${hasExportHandleUpgradeFunctions.join(", ")}\n`);
+    const wss = new ws_1.WebSocketServer({
+        noServer: true,
+        clientTracking: true,
+        perMessageDeflate: {
+            zlibDeflateOptions: {
+                chunkSize: 1024,
+                memLevel: 7,
+                level: 3,
+            },
+            zlibInflateOptions: {
+                chunkSize: 10 * 1024,
+            },
+            clientNoContextTakeover: true,
+            serverNoContextTakeover: true,
+            serverMaxWindowBits: 10,
+            concurrencyLimit: 10,
+            threshold: 1024,
+        },
+    });
+    setupGracefulShutdown(wss);
+    setupWebSocketsHeartbeat(wss);
+    wss.on("error", (err) => {
+        logger_1.functionsLogger.logError(err, { scene: "ws_server_error" });
+    });
+    wss.on("close", () => {
+        logger_1.functionsLogger.logError("ws server has been closed");
+    });
+    const composedMiddleware = (0, koa_compose_1.default)([
+        (0, middle_event_id_1.eventIdMiddleware)(),
+        (0, middle_async_context_1.asyncContextMiddleware)(),
+        (0, middle_common_logger_1.loggerMiddleware)(),
+        (ctx, next) => __awaiter(this, void 0, void 0, function* () {
+            try {
+                yield next();
+            }
+            catch (err) {
+                logger_1.functionsLogger.logError(err, { scene: "ws_middleware_error" });
+            }
+        }),
+        (0, middle_context_injection_1.contextInjectionMiddleware)(options, projectConfig),
+    ]);
+    httpServer.on("upgrade", (request, socket, head) => __awaiter(this, void 0, void 0, function* () {
+        const { handleFunction } = (0, router_1.routeFunction)(router, request.url || "/");
+        const userFunction = handleFunction;
+        if (typeof userFunction === "undefined" || typeof userFunction.handleUpgrade !== "function") {
+            socket.destroy();
+            return;
+        }
+        request.socket.setKeepAlive(true);
+        request.socket.setNoDelay(true);
+        request.socket.setTimeout(0);
+        try {
+            const ctx = koaApp.createContext(request, {});
+            yield composedMiddleware(ctx, () => {
+                return new Promise((resolve) => {
+                    if (typeof userFunction.handleUpgrade !== "function") {
+                        respond(socket, 404, "Not Found");
+                        return;
+                    }
+                    const upgradeContext = {
+                        ctxId: ctx.state.eventID,
+                        eventID: ctx.state.eventID,
+                        eventType: "websocket.upgrade",
+                        timestamp: new Date().toISOString(),
+                        httpContext: {
+                            url: request.url,
+                            httpMethod: request.method,
+                            headers: request.headers,
+                        },
+                    };
+                    ctx.state.contextInjected = upgradeContext;
+                    try {
+                        Promise.resolve(userFunction.handleUpgrade(upgradeContext))
+                            .then((result) => {
+                            if (!result.allowWebSocket) {
+                                respond(socket, result.statusCode || 404, result.body || "Not Found", result.contentType);
+                                return;
+                            }
+                            wss.handleUpgrade(request, socket, head, (ws) => __awaiter(this, void 0, void 0, function* () {
+                                const context = {
+                                    ctxId: ctx.state.eventID,
+                                    eventID: ctx.state.eventID,
+                                    eventType: "websocket.connection",
+                                    timestamp: new Date().toISOString(),
+                                    httpContext: {
+                                        url: request.url,
+                                        httpMethod: request.method,
+                                        headers: request.headers,
+                                    },
+                                    ws: new VectorxWebSocket(ws, { ctxId: ctx.state.eventID }),
+                                };
+                                (0, helper_1.deepFreeze)(context);
+                                process.nextTick(() => {
+                                    try {
+                                        userFunction(null, Object.assign({}, context));
+                                    }
+                                    catch (_e) {
+                                    }
+                                });
+                                resolve();
+                            }));
+                        })
+                            .catch((e) => {
+                            logger_1.functionsLogger.logError(`handleUpgrade promise-catch error: ${(0, error_1.decorateErrorStack)(e)}`);
+                            respondError(socket, e);
+                        });
+                    }
+                    catch (e) {
+                        logger_1.functionsLogger.logError(`handleUpgrade try-catch error: ${(0, error_1.decorateErrorStack)(e)}`);
+                        respondError(socket, e);
+                    }
+                });
+            });
+        }
+        catch (e) {
+            logger_1.functionsLogger.logError(e, { scene: "ws_upgrade_error" });
+            respondError(socket, e);
+        }
+    }));
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@vectorx/functions-framework",
-  "version": "0.0.0-beta-20260109055258",
+  "version": "0.0.0-beta-20260320063305",
   "description": "VectorX Functions Framework",
   "main": "lib/index.js",
   "types": "types/index.d.ts",
@@ -22,15 +22,15 @@
   "author": "",
   "license": "ISC",
   "engines": {
-    "node": ">=18.0.0"
+    "node": ">=20.0.0"
   },
   "dependencies": {
-    "@vectorx/ai-types": "0.0.0-beta-20260109055258",
     "async_hooks": "^1.0.0",
     "chalk": "4",
     "commander": "^12.1.0",
     "dotenv": "^16.5.0",
     "koa": "^2.14.2",
+    "koa-compose": "^4.1.0",
     "koa-body": "^6.0.1",
     "koa-bodyparser": "^4.4.1",
     "koa-router": "^12.0.1",
@@ -41,8 +41,11 @@
     "radix3": "^1.1.2",
     "raw-body": "^2.5.2",
     "uuid": "^9.0.1",
+    "ws": "^8.18.0",
     "winston": "^3.11.0",
-    "winston-daily-rotate-file": "^4.7.1"
+    "winston-daily-rotate-file": "^4.7.1",
+    "@vectorx/ai-types": "0.0.0-beta-20260320063305",
+    "@vectorx/endpoints": "0.0.0-beta-20260320063305"
   },
   "devDependencies": {
     "@types/jest": "^29.5.12",
@@ -52,6 +55,7 @@
     "@types/node": "^20.11.24",
     "@types/supertest": "^6.0.2",
     "@types/uuid": "^9.0.8",
+    "@types/ws": "^8.5.12",
     "@typescript-eslint/eslint-plugin": "^7.1.0",
     "@typescript-eslint/parser": "^7.1.0",
     "eslint": "^8.57.0",
@@ -71,6 +75,5 @@
     "test:ci": "jest --ci --coverage",
     "lint": "eslint src --ext .ts",
     "run:demo": "node ./bin/rcb-ff.js"
-  },
-  "readme": "# VectorX Functions Framework\n\nVectorX Functions Framework 是一个用于构建和运行云函数的框架，提供了完整的日志系统支持。\n\n## 日志系统\n\n框架内置了完整的日志系统，支持请求日志记录和用户代码日志记录。\n\n### 配置\n\n在初始化框架时，可以配置日志系统：\n\n```typescript\nimport { createAgentServerFramework } from '@vectorx/functions-framework';\n\nconst framework = createAgentServerFramework({\n    port: 3000,\n    logging: {\n        dirname: '/path/to/logs',  // 日志存储目录\n        maxSize: '20m',           // 单个日志文件最大大小\n        maxFiles: 14              // 保留的日志文件数量\n    }\n});\n```\n\n### 日志类型\n\n框架支持两种类型的日志：\n\n- `ACCESS`: 访问日志，记录所有 HTTP 请求\n- `USERCODE`: 用户代码日志，记录用户函数中的日志\n\n### 日志查询 API\n\n框架提供了日志查询 API，用于查看和调试日志：\n\n```\nGET /@logs\n```\n\n#### 查询参数\n\n| 参数名 | 类型 | 必填 | 说明 |\n|--------|------|------|------|\n| type | string | 否 | 日志类型，可选值：ACCESS/USERCODE，默认 ACCESS |\n| limit | number | 否 | 返回的日志条数限制，默认 100 |\n| eventId | string | 否 | 按 eventId 过滤日志 |\n\n#### 示例\n\n1. 查询所有访问日志：\n```bash\ncurl 'http://localhost:3000/@logs?type=ACCESS'\n```\n\n2. 查询特定 eventId 的日志：\n```bash\ncurl 'http://localhost:3000/@logs?eventId=550e8400-e29b-41d4-a716-446655440000'\n```\n\n3. 限制返回条数：\n```bash\ncurl 'http://localhost:3000/@logs?limit=10'\n```\n\n4. 组合查询：\n```bash\ncurl 'http://localhost:3000/@logs?type=USERCODE&eventId=550e8400-e29b-41d4-a716-446655440000&limit=20'\n```\n\n#### 响应格式\n\n成功响应：\n```json\n{\n    \"success\": true,\n    \"data\": [\n        {\n            \"@timestamp\": \"2024-03-21T10:30:00.000Z\",\n            \"level\": \"info\",\n            \"eventId\": \"550e8400-e29b-41d4-a716-446655440000\",\n            \"message\": \"Request started\",\n            \"method\": \"GET\",\n            \"url\": \"/api/example\"\n        }\n    ]\n}\n```\n\n错误响应：\n```json\n{\n    \"success\": false,\n    \"error\": \"错误信息\"\n}\n```\n\n### 在代码中使用日志\n\n```typescript\nimport { functionsLogger } from '@vectorx/functions-framework';\nimport { LogLevel } from '@vectorx/functions-framework';\n\n// 记录访问日志\nfunctionsLogger.logAccesslog(LogLevel.INFO, 'Custom message', {\n    customField: 'value'\n});\n\n// 记录用户代码日志\nfunctionsLogger.logUserCodelog([\n    { message: 'User log message', level: 'info' }\n]);\n```\n\n### 日志格式\n\n每条日志都包含以下字段：\n\n- `@timestamp`: 日志时间戳\n- `level`: 日志级别\n- `eventId`: 请求 ID\n- `message`: 日志消息\n- 其他自定义字段\n\n## 开发\n\n### 运行测试\n\n```bash\nnpm test\n```\n\n### 构建\n\n```bash\nnpm run build\n``` \n\n\n### 环境管理\n读取 .env 文件，校验 env 文件的可用性，获取当前运行环境\n"
+  }
 }

package/types/framework.d.ts

@@ -1,4 +1,5 @@
 declare function getDependencyVersion(packageName: string, searchPath: string): string | null;
+export type FrameworkMode = "agent" | "fun";
 export interface FrameworkOptions {
     port: number;
     timeoutMS: number;
@@ -18,6 +19,7 @@ export interface FrameworkOptions {
     redLangfuseConfig?: {
         enable?: boolean;
     };
+    mode?: FrameworkMode;
 }
 export type FunctionName = string;
 declare class AgentServerFramework {
@@ -27,6 +29,9 @@ declare class AgentServerFramework {
     constructor(options: FrameworkOptions);
     startServer(): Promise<import("http").Server<typeof import("http").IncomingMessage, typeof import("http").ServerResponse>>;
     getRegisteredFunction(fnName: string): Function;
+    private loadAgentModeConfigs;
+    private loadFunModeConfigs;
+    private loadAndRegisterFunctions;
 }
 export { getDependencyVersion };
 export declare function createAgentServerFramework(options: FrameworkOptions): AgentServerFramework;