@cicctencent/agent-server 0.1.1 → 0.2.0

package/dist/cjs/spi/adapters.js

@@ -0,0 +1,341 @@
+"use strict";
+// ============================================================
+// 框架适配器 — 支持不同 Web 框架
+// ============================================================
+/**
+ * ## 设计原则
+ *
+ * agent-server 核心只定义**接口**（能力契约），不依赖具体框架。
+ *
+ * 以下适配器实现是**可选参考实现**，业务层可以：
+ * 1. 直接使用（如果满足需求）
+ * 2. 参考实现自己定制（如果有特殊需求）
+ * 3. 完全自己实现（如果框架不同）
+ *
+ * ### 使用示例
+ *
+ * ```typescript
+ * // 方式 1：直接使用提供的适配器
+ * import { KoaSSEAdapter } from '@cicctencent/agent-server';
+ * const adapter = new KoaSSEAdapter(ctx);
+ *
+ * // 方式 2：继承并扩展
+ * class MySSEAdapter extends KoaSSEAdapter {
+ *   write(data: string) {
+ *     // 添加自定义逻辑（如日志、监控）
+ *     console.log('[SSE]', data.length, 'bytes');
+ *     super.write(data);
+ *   }
+ * }
+ *
+ * // 方式 3：完全自己实现
+ * class CustomSSEAdapter implements SSEResponseAdapter {
+ *   // 完全自定义实现
+ * }
+ * ```
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CustomSSEAdapter = exports.KoaRequestAdapter = exports.ExpressRequestAdapter = exports.NativeHttpSSEAdapter = exports.FastifySSEAdapter = exports.KoaSSEAdapter = exports.ExpressSSEAdapter = void 0;
+exports.adaptSSE = adaptSSE;
+exports.adaptRequest = adaptRequest;
+// ============================================================
+// SSE 响应适配器（参考实现）
+// ============================================================
+/**
+ * Express SSE 适配器（参考实现）
+ *
+ * 业务层可以：
+ * - 直接使用
+ * - 继承扩展
+ * - 参考 implementation 自己实现
+ */
+class ExpressSSEAdapter {
+    res;
+    constructor(res) {
+        this.res = res;
+    }
+    write(data) {
+        this.res.write(data);
+    }
+    end() {
+        this.res.end();
+    }
+    get writableEnded() {
+        return this.res.writableEnded;
+    }
+    setHeader(name, value) {
+        this.res.setHeader(name, value);
+    }
+    flushHeaders() {
+        this.res.flushHeaders?.();
+    }
+    status(code) {
+        this.res.status(code);
+    }
+    json(data) {
+        this.res.json(data);
+    }
+}
+exports.ExpressSSEAdapter = ExpressSSEAdapter;
+/**
+ * Koa/MidwayJS SSE 适配器（参考实现）
+ */
+class KoaSSEAdapter {
+    ctx;
+    constructor(ctx) {
+        this.ctx = ctx;
+        // 禁止 Koa 框架自行处理响应，让 SSE 流完全由适配器控制
+        if (this.ctx && 'respond' in this.ctx) {
+            this.ctx.respond = false;
+        }
+    }
+    write(data) {
+        this.ctx.res.write(data);
+    }
+    end() {
+        this.ctx.res.end();
+    }
+    get writableEnded() {
+        return this.ctx.res.writableEnded;
+    }
+    setHeader(name, value) {
+        this.ctx.set(name, String(value));
+    }
+    flushHeaders() {
+        this.ctx.res.flushHeaders?.();
+    }
+    status(code) {
+        this.ctx.status = code;
+    }
+    json(data) {
+        this.ctx.body = data;
+    }
+}
+exports.KoaSSEAdapter = KoaSSEAdapter;
+/**
+ * Fastify SSE 适配器（参考实现）
+ */
+class FastifySSEAdapter {
+    reply;
+    constructor(reply) {
+        this.reply = reply;
+    }
+    write(data) {
+        this.reply.raw.write(data);
+    }
+    end() {
+        this.reply.raw.end();
+    }
+    get writableEnded() {
+        return this.reply.raw.writableEnded;
+    }
+    setHeader(name, value) {
+        this.reply.header(name, value);
+    }
+    flushHeaders() {
+        this.reply.raw.flushHeaders?.();
+    }
+    status(code) {
+        this.reply.code(code);
+    }
+    json(data) {
+        this.reply.send(data);
+    }
+}
+exports.FastifySSEAdapter = FastifySSEAdapter;
+/**
+ * 原生 HTTP SSE 适配器
+ */
+class NativeHttpSSEAdapter {
+    res;
+    constructor(res) {
+        this.res = res;
+    }
+    write(data) {
+        this.res.write(data);
+    }
+    end() {
+        this.res.end();
+    }
+    get writableEnded() {
+        return this.res.writableEnded;
+    }
+    setHeader(name, value) {
+        this.res.setHeader(name, value);
+    }
+    flushHeaders() {
+        this.res.flushHeaders?.();
+    }
+    status(code) {
+        this.res.statusCode = code;
+    }
+    json(data) {
+        this.res.setHeader('Content-Type', 'application/json');
+        this.res.end(JSON.stringify(data));
+    }
+}
+exports.NativeHttpSSEAdapter = NativeHttpSSEAdapter;
+/**
+ * 自动检测框架类型并创建适配器
+ *
+ * ⚠️ 注意：此函数是便捷工具，但建议业务层显式创建适配器：
+ *
+ * ```typescript
+ * // 推荐：显式创建（更可控）
+ * const adapter = new KoaSSEAdapter(ctx);
+ *
+ * // 不推荐：自动检测（可能误判）
+ * const adapter = adaptSSE(ctx);
+ * ```
+ *
+ * 自动检测可能存在的问题：
+ * - 框架版本差异导致误判
+ * - 自定义响应对象无法识别
+ * - 难以调试
+ */
+function adaptSSE(res) {
+    // Express: res.setHeader, res.write, res.end
+    if (res && typeof res === 'object' && 'setHeader' in res && 'write' in res) {
+        // 检查是否是 Koa Context (有 ctx.res 且 ctx.set)
+        if ('ctx' in res || ('req' in res && 'socket' in res.req)) {
+            return new KoaSSEAdapter(res);
+        }
+        // 检查是否是 Fastify Reply (有 .raw)
+        if ('raw' in res && 'header' in res) {
+            return new FastifySSEAdapter(res);
+        }
+        // Express 或原生 HTTP
+        return new ExpressSSEAdapter(res);
+    }
+    // Koa Context
+    if (res && typeof res === 'object' && 'res' in res && 'set' in res) {
+        return new KoaSSEAdapter(res);
+    }
+    // Fastify Reply
+    if (res && typeof res === 'object' && 'raw' in res && 'header' in res) {
+        return new FastifySSEAdapter(res);
+    }
+    // 原生 HTTP Response
+    if (res && typeof res === 'object' && 'write' in res && 'end' in res) {
+        return new NativeHttpSSEAdapter(res);
+    }
+    throw new Error('Unsupported response type. Please provide a custom SSE adapter.');
+}
+// ============================================================
+// 请求上下文适配器
+// ============================================================
+/**
+ * Express 请求上下文适配器
+ */
+class ExpressRequestAdapter {
+    req;
+    constructor(req) {
+        this.req = req;
+    }
+    getUserId() {
+        return this.req.user?.id ?? this.req.headers['x-user-id'];
+    }
+    getWorkspaceId() {
+        const ws = this.req.headers['x-workspace-id'] ?? this.req.query.workspaceId;
+        return ws ? Number(ws) : undefined;
+    }
+    getHeader(name) {
+        return this.req.headers[name.toLowerCase()];
+    }
+    hasPermission(permission) {
+        return this.req.user?.permissions?.includes(permission) ?? false;
+    }
+    getRawRequest() {
+        return this.req;
+    }
+    getRawResponse() {
+        return this.req.res;
+    }
+}
+exports.ExpressRequestAdapter = ExpressRequestAdapter;
+/**
+ * Koa/MidwayJS 请求上下文适配器
+ */
+class KoaRequestAdapter {
+    ctx;
+    constructor(ctx) {
+        this.ctx = ctx;
+    }
+    getUserId() {
+        return this.ctx.state?.user?.id ?? this.ctx.get('x-user-id');
+    }
+    getWorkspaceId() {
+        const ws = this.ctx.get('x-workspace-id') ?? this.ctx.query.workspaceId;
+        return ws ? Number(ws) : undefined;
+    }
+    getHeader(name) {
+        return this.ctx.get(name);
+    }
+    hasPermission(permission) {
+        return this.ctx.state?.user?.permissions?.includes(permission) ?? false;
+    }
+    getRawRequest() {
+        return this.ctx.req;
+    }
+    getRawResponse() {
+        return this.ctx.res;
+    }
+}
+exports.KoaRequestAdapter = KoaRequestAdapter;
+/**
+ * 自动检测框架类型并创建请求适配器
+ */
+function adaptRequest(req) {
+    // Koa Context
+    if (req && typeof req === 'object' && 'state' in req && 'get' in req) {
+        return new KoaRequestAdapter(req);
+    }
+    // Express Request
+    if (req && typeof req === 'object' && 'headers' in req && 'query' in req) {
+        return new ExpressRequestAdapter(req);
+    }
+    throw new Error('Unsupported request type. Please provide a custom request adapter.');
+}
+// ============================================================
+// 自定义适配器示例
+// ============================================================
+/**
+ * 自定义 SSE 适配器示例
+ * 应用层可参考此实现创建自己的适配器
+ */
+class CustomSSEAdapter {
+    onWrite;
+    onEnd;
+    onFlush;
+    _writableEnded = false;
+    headers = new Map();
+    constructor(onWrite, onEnd, onFlush) {
+        this.onWrite = onWrite;
+        this.onEnd = onEnd;
+        this.onFlush = onFlush;
+    }
+    write(data) {
+        if (this._writableEnded)
+            return;
+        this.onWrite(data);
+    }
+    end() {
+        this._writableEnded = true;
+        this.onEnd();
+    }
+    get writableEnded() {
+        return this._writableEnded;
+    }
+    setHeader(name, value) {
+        this.headers.set(name, String(value));
+    }
+    flushHeaders() {
+        this.onFlush?.();
+    }
+    /** 获取已设置的响应头 */
+    getHeaders() {
+        return Object.fromEntries(this.headers);
+    }
+}
+exports.CustomSSEAdapter = CustomSSEAdapter;
+//# sourceMappingURL=adapters.js.map

package/dist/cjs/spi/adapters.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"adapters.js","sourceRoot":"","sources":["../../../src/spi/adapters.ts"],"names":[],"mappings":";AAAA,+DAA+D;AAC/D,sBAAsB;AACtB,+DAA+D;AAC/D;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA+BG;;;AAiLH,4BA+BC;AAyED,oCAYC;AAjSD,+DAA+D;AAC/D,kBAAkB;AAClB,+DAA+D;AAE/D;;;;;;;GAOG;AACH,MAAa,iBAAiB;IACR;IAApB,YAAoB,GAAQ;QAAR,QAAG,GAAH,GAAG,CAAK;IAAG,CAAC;IAEhC,KAAK,CAAC,IAAY;QAChB,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACvB,CAAC;IAED,GAAG;QACD,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC;IACjB,CAAC;IAED,IAAI,aAAa;QACf,OAAO,IAAI,CAAC,GAAG,CAAC,aAAa,CAAC;IAChC,CAAC;IAED,SAAS,CAAC,IAAY,EAAE,KAAsB;QAC5C,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IAClC,CAAC;IAED,YAAY;QACV,IAAI,CAAC,GAAG,CAAC,YAAY,EAAE,EAAE,CAAC;IAC5B,CAAC;IAED,MAAM,CAAC,IAAY;QACjB,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IACxB,CAAC;IAED,IAAI,CAAC,IAAa;QAChB,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACtB,CAAC;CACF;AA9BD,8CA8BC;AAED;;GAEG;AACH,MAAa,aAAa;IACJ;IAApB,YAAoB,GAAQ;QAAR,QAAG,GAAH,GAAG,CAAK;QAC1B,kCAAkC;QAClC,IAAI,IAAI,CAAC,GAAG,IAAI,SAAS,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;YACtC,IAAI,CAAC,GAAG,CAAC,OAAO,GAAG,KAAK,CAAC;QAC3B,CAAC;IACH,CAAC;IAED,KAAK,CAAC,IAAY;QAChB,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC3B,CAAC;IAED,GAAG;QACD,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC;IACrB,CAAC;IAED,IAAI,aAAa;QACf,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,aAAa,CAAC;IACpC,CAAC;IAED,SAAS,CAAC,IAAY,EAAE,KAAsB;QAC5C,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IACpC,CAAC;IAED,YAAY;QACV,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,YAAY,EAAE,EAAE,CAAC;IAChC,CAAC;IAED,MAAM,CAAC,IAAY;QACjB,IAAI,CAAC,GAAG,CAAC,MAAM,GAAG,IAAI,CAAC;IACzB,CAAC;IAED,IAAI,CAAC,IAAa;QAChB,IAAI,CAAC,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC;IACvB,CAAC;CACF;AAnCD,sCAmCC;AAED;;GAEG;AACH,MAAa,iBAAiB;IACR;IAApB,YAAoB,KAAU;QAAV,UAAK,GAAL,KAAK,CAAK;IAAG,CAAC;IAElC,KAAK,CAAC,IAAY;QAChB,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC7B,CAAC;IAED,GAAG;QACD,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC;IACvB,CAAC;IAED,IAAI,aAAa;QACf,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,aAAa,CAAC;IACtC,CAAC;IAED,SAAS,CAAC,IAAY,EAAE,KAAsB;QAC5C,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IACjC,CAAC;IAED,YAAY;QACV,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,YAAY,EAAE,EAAE,CAAC;IAClC,CAAC;IAED,MAAM,CAAC,IAAY;QACjB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxB,CAAC;IAED,IAAI,CAAC,IAAa;QAChB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxB,CAAC;CACF;AA9BD,8CA8BC;AAED;;GAEG;AACH,MAAa,oBAAoB;IACX;IAApB,YAAoB,GAAQ;QAAR,QAAG,GAAH,GAAG,CAAK;IAAG,CAAC;IAEhC,KAAK,CAAC,IAAY;QAChB,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACvB,CAAC;IAED,GAAG;QACD,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC;IACjB,CAAC;IAED,IAAI,aAAa;QACf,OAAO,IAAI,CAAC,GAAG,CAAC,aAAa,CAAC;IAChC,CAAC;IAED,SAAS,CAAC,IAAY,EAAE,KAAsB;QAC5C,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IAClC,CAAC;IAED,YAAY;QACV,IAAI,CAAC,GAAG,CAAC,YAAY,EAAE,EAAE,CAAC;IAC5B,CAAC;IAED,MAAM,CAAC,IAAY;QACjB,IAAI,CAAC,GAAG,CAAC,UAAU,GAAG,IAAI,CAAC;IAC7B,CAAC;IAED,IAAI,CAAC,IAAa;QAChB,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,cAAc,EAAE,kBAAkB,CAAC,CAAC;QACvD,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC;IACrC,CAAC;CACF;AA/BD,oDA+BC;AAED;;;;;;;;;;;;;;;;;GAiBG;AACH,SAAgB,QAAQ,CAAC,GAAY;IACnC,6CAA6C;IAC7C,IAAI,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,WAAW,IAAI,GAAG,IAAI,OAAO,IAAI,GAAG,EAAE,CAAC;QAC3E,0CAA0C;QAC1C,IAAI,KAAK,IAAK,GAAW,IAAI,CAAC,KAAK,IAAK,GAAW,IAAI,QAAQ,IAAK,GAAW,CAAC,GAAG,CAAC,EAAE,CAAC;YACrF,OAAO,IAAI,aAAa,CAAC,GAAG,CAAC,CAAC;QAChC,CAAC;QACD,+BAA+B;QAC/B,IAAI,KAAK,IAAK,GAAW,IAAI,QAAQ,IAAK,GAAW,EAAE,CAAC;YACtD,OAAO,IAAI,iBAAiB,CAAC,GAAG,CAAC,CAAC;QACpC,CAAC;QACD,mBAAmB;QACnB,OAAO,IAAI,iBAAiB,CAAC,GAAG,CAAC,CAAC;IACpC,CAAC;IAED,cAAc;IACd,IAAI,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,KAAK,IAAI,GAAG,IAAI,KAAK,IAAI,GAAG,EAAE,CAAC;QACnE,OAAO,IAAI,aAAa,CAAC,GAAG,CAAC,CAAC;IAChC,CAAC;IAED,gBAAgB;IAChB,IAAI,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,KAAK,IAAI,GAAG,IAAI,QAAQ,IAAI,GAAG,EAAE,CAAC;QACtE,OAAO,IAAI,iBAAiB,CAAC,GAAG,CAAC,CAAC;IACpC,CAAC;IAED,mBAAmB;IACnB,IAAI,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,OAAO,IAAI,GAAG,IAAI,KAAK,IAAI,GAAG,EAAE,CAAC;QACrE,OAAO,IAAI,oBAAoB,CAAC,GAAG,CAAC,CAAC;IACvC,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,iEAAiE,CAAC,CAAC;AACrF,CAAC;AAED,+DAA+D;AAC/D,WAAW;AACX,+DAA+D;AAE/D;;GAEG;AACH,MAAa,qBAAqB;IACZ;IAApB,YAAoB,GAAQ;QAAR,QAAG,GAAH,GAAG,CAAK;IAAG,CAAC;IAEhC,SAAS;QACP,OAAO,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;IAC5D,CAAC;IAED,cAAc;QACZ,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,gBAAgB,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,WAAW,CAAC;QAC5E,OAAO,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACrC,CAAC;IAED,SAAS,CAAC,IAAY;QACpB,OAAO,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;IAC9C,CAAC;IAED,aAAa,CAAC,UAAkB;QAC9B,OAAO,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,WAAW,EAAE,QAAQ,CAAC,UAAU,CAAC,IAAI,KAAK,CAAC;IACnE,CAAC;IAED,aAAa;QACX,OAAO,IAAI,CAAC,GAAG,CAAC;IAClB,CAAC;IAED,cAAc;QACZ,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;IACtB,CAAC;CACF;AA3BD,sDA2BC;AAED;;GAEG;AACH,MAAa,iBAAiB;IACR;IAApB,YAAoB,GAAQ;QAAR,QAAG,GAAH,GAAG,CAAK;IAAG,CAAC;IAEhC,SAAS;QACP,OAAO,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,IAAI,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,WAAW,CAAC,CAAC;IAC/D,CAAC;IAED,cAAc;QACZ,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,gBAAgB,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,WAAW,CAAC;QACxE,OAAO,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACrC,CAAC;IAED,SAAS,CAAC,IAAY;QACpB,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAC5B,CAAC;IAED,aAAa,CAAC,UAAkB;QAC9B,OAAO,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,CAAC,UAAU,CAAC,IAAI,KAAK,CAAC;IAC1E,CAAC;IAED,aAAa;QACX,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;IACtB,CAAC;IAED,cAAc;QACZ,OAAO,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;IACtB,CAAC;CACF;AA3BD,8CA2BC;AAED;;GAEG;AACH,SAAgB,YAAY,CAAC,GAAY;IACvC,cAAc;IACd,IAAI,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,OAAO,IAAI,GAAG,IAAI,KAAK,IAAI,GAAG,EAAE,CAAC;QACrE,OAAO,IAAI,iBAAiB,CAAC,GAAG,CAAC,CAAC;IACpC,CAAC;IAED,kBAAkB;IAClB,IAAI,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,SAAS,IAAI,GAAG,IAAI,OAAO,IAAI,GAAG,EAAE,CAAC;QACzE,OAAO,IAAI,qBAAqB,CAAC,GAAG,CAAC,CAAC;IACxC,CAAC;IAED,MAAM,IAAI,KAAK,CAAC,oEAAoE,CAAC,CAAC;AACxF,CAAC;AAED,+DAA+D;AAC/D,WAAW;AACX,+DAA+D;AAE/D;;;GAGG;AACH,MAAa,gBAAgB;IAKjB;IACA;IACA;IANF,cAAc,GAAG,KAAK,CAAC;IACvB,OAAO,GAAwB,IAAI,GAAG,EAAE,CAAC;IAEjD,YACU,OAA+B,EAC/B,KAAiB,EACjB,OAAoB;QAFpB,YAAO,GAAP,OAAO,CAAwB;QAC/B,UAAK,GAAL,KAAK,CAAY;QACjB,YAAO,GAAP,OAAO,CAAa;IAC3B,CAAC;IAEJ,KAAK,CAAC,IAAY;QAChB,IAAI,IAAI,CAAC,cAAc;YAAE,OAAO;QAChC,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;IACrB,CAAC;IAED,GAAG;QACD,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC3B,IAAI,CAAC,KAAK,EAAE,CAAC;IACf,CAAC;IAED,IAAI,aAAa;QACf,OAAO,IAAI,CAAC,cAAc,CAAC;IAC7B,CAAC;IAED,SAAS,CAAC,IAAY,EAAE,KAAsB;QAC5C,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IACxC,CAAC;IAED,YAAY;QACV,IAAI,CAAC,OAAO,EAAE,EAAE,CAAC;IACnB,CAAC;IAED,gBAAgB;IAChB,UAAU;QACR,OAAO,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC1C,CAAC;CACF;AApCD,4CAoCC"}

package/dist/cjs/spi/examples/typeorm-example.d.ts

@@ -0,0 +1,208 @@
+import type { ChatStoreProvider, BaseThreadEntity, BaseMessageEntity, BaseSettingsEntity, ListStorageAdapter } from '../types.js';
+/**
+ * BI 项目的 Thread 实体（TypeORM）
+ * 注意：字段可能与 agent-server 默认实体不同
+ */
+interface BIThreadEntity extends BaseThreadEntity {
+    id: number;
+    workspaceId?: number;
+    userId?: number;
+    agentType?: string;
+    modelConfig?: string;
+    createdAt: Date;
+    updatedAt: Date;
+}
+/**
+ * BI 项目的 Message 实体（TypeORM）
+ */
+interface BIMessageEntity extends BaseMessageEntity {
+    id: number;
+    threadId: number;
+    tokens?: number;
+    latency?: number;
+    model?: string;
+    createdAt: Date;
+}
+/**
+ * BI 项目的 Settings 实体
+ */
+interface BISettingsEntity extends BaseSettingsEntity {
+    id: number;
+    userId: number;
+    theme: string;
+    language: string;
+    modelProviders: Array<{
+        id: string;
+        name: string;
+        provider: string;
+        model: string;
+        apiKey?: string;
+        baseUrl?: string;
+    }>;
+    createdAt: Date;
+    updatedAt: Date;
+}
+/**
+ * TypeORM 列表存储适配器
+ * 将 TypeORM Repository 包装为 ListStorageAdapter
+ */
+declare class TypeORMListStorage<T extends {
+    id: number | string;
+}> implements ListStorageAdapter<T> {
+    private data;
+    list(): Promise<T[]>;
+    findById(id: number | string): Promise<T | null>;
+    add(entity: T): Promise<T>;
+    update(id: number | string, data: Partial<T>): Promise<T | null>;
+    remove(id: number | string): Promise<boolean>;
+}
+/**
+ * BI 项目的 ChatStoreProvider 实现
+ * 使用 TypeORM Repository 作为底层存储
+ */
+declare class TypeORMChatStoreProvider implements ChatStoreProvider<BIThreadEntity, BIMessageEntity> {
+    private threadStorage;
+    private messageStorages;
+    constructor();
+    private getMessageStorage;
+    getThreads(workspaceId?: number | string): Promise<BIThreadEntity[]>;
+    getThread(id: number | string): Promise<BIThreadEntity | undefined>;
+    createThread(data: Partial<BIThreadEntity>): Promise<BIThreadEntity>;
+    updateThread(id: number | string, data: Partial<BIThreadEntity>): Promise<BIThreadEntity | undefined>;
+    deleteThread(id: number | string): Promise<boolean>;
+    getMessages(threadId: number | string, options?: {
+        limit?: number;
+        offset?: number;
+    }): Promise<{
+        items: BIMessageEntity[];
+        total: number;
+        hasMore: boolean;
+    }>;
+    addMessage(data: Partial<BIMessageEntity> & {
+        threadId: number | string;
+        role: string;
+        content: string;
+    }): Promise<BIMessageEntity>;
+    updateMessage(threadId: number | string, id: number | string, data: Partial<BIMessageEntity>): Promise<BIMessageEntity | undefined>;
+    deleteMessage(threadId: number | string, id: number | string): Promise<boolean>;
+    clearThreadMessages(threadId: number | string): Promise<void>;
+    buildThreadHistory(threadId: number | string, limit?: number): Promise<any[]>;
+}
+/**
+ * MidwayJS Agent 服务示例
+ * 使用 SPI 适配到 MidwayJS + TypeORM
+ *
+ * 在实际项目中，这会是：
+ *
+ * @Provide()
+ * @Scope(ScopeEnum.Singleton)
+ * export class AgentService {
+ *   @Inject()
+ *   threadRepository: Repository<BIThreadEntity>;
+ *
+ *   @Inject()
+ *   messageRepository: Repository<BIMessageEntity>;
+ *
+ *   private server: AgentServer;
+ *
+ *   @Init()
+ *   async init() {
+ *     const chatStoreProvider = new TypeORMChatStoreProvider(
+ *       this.threadRepository,
+ *       this.messageRepository
+ *     );
+ *
+ *     this.server = createAgentServer({
+ *       llmProvider: await this.createLLMProvider(),
+ *       chatStoreProvider,
+ *       // 使用 Koa 适配器
+ *       createSSEResponseAdapter: (ctx) => new KoaSSEAdapter(ctx),
+ *     });
+ *   }
+ *
+ *   @Get('/chat/stream')
+ *   async chatStream(@Ctx() ctx: Context, @Query() query: { message: string }) {
+ *     const adapter = new KoaSSEAdapter(ctx);
+ *     await this.server.runAgentStream(adapter, { content: query.message });
+ *   }
+ * }
+ */
+export { TypeORMListStorage, TypeORMChatStoreProvider, BIThreadEntity, BIMessageEntity, BISettingsEntity, };
+/**
+ * ## 使用说明
+ *
+ * ### 1. 在 MidwayJS 项目中使用 agent-server
+ *
+ * ```typescript
+ * // src/service/agent.service.ts
+ * import { Provide, Scope, ScopeEnum, Inject, Init, Get, Ctx, Query } from '@midwayjs/core';
+ * import { Context } from '@midwayjs/koa';
+ * import { createAgentServer, KoaSSEAdapter, type AgentServer } from '@cicctencent/agent-server';
+ * import { TypeORMChatStoreProvider } from './spi/typeorm-chat-provider';
+ *
+ * @Provide()
+ * @Scope(ScopeEnum.Singleton)
+ * export class AgentService {
+ *   @Inject()
+ *   threadRepository: Repository<ThreadEntity>;
+ *
+ *   @Inject()
+ *   messageRepository: Repository<MessageEntity>;
+ *
+ *   private server: AgentServer;
+ *
+ *   @Init()
+ *   async init() {
+ *     // 使用 TypeORM 存储适配器
+ *     const chatStoreProvider = new TypeORMChatStoreProvider(
+ *       this.threadRepository,
+ *       this.messageRepository
+ *     );
+ *
+ *     this.server = createAgentServer({
+ *       llmProvider: await this.createLLMProvider(),
+ *       chatStoreProvider, // 替换默认 JSON 文件存储
+ *     });
+ *   }
+ *
+ *   @Get('/chat/stream')
+ *   async chatStream(@Ctx() ctx: Context, @Query() query: { message: string }) {
+ *     // 使用 Koa/MidwayJS SSE 适配器
+ *     const adapter = new KoaSSEAdapter(ctx);
+ *     await this.server.runAgentStream(adapter, { content: query.message });
+ *   }
+ * }
+ * ```
+ *
+ * ### 2. 自定义实体字段
+ *
+ * agent-server 现在支持泛型实体类型，你可以：
+ * - 使用自己的实体结构（如 TypeORM 实体）
+ * - 自定义字段（如 userId, modelConfig 等）
+ * - 使用不同的 ID 类型（number / string / UUID）
+ *
+ * ### 3. 组件级使用（不依赖工厂）
+ *
+ * 如果不想使用 createAgentServer 工厂，可以只导入需要的组件：
+ *
+ * ```typescript
+ * import {
+ *   MemoryService,
+ *   writeSSE,
+ *   createBuiltinTools,
+ *   KoaSSEAdapter
+ * } from '@cicctencent/agent-server';
+ *
+ * // 只使用需要的组件
+ * const memoryService = new MemoryService('./data');
+ * const tools = createBuiltinTools();
+ *
+ * // 在 MidwayJS 中使用 SSE
+ * @Get('/stream')
+ * async stream(@Ctx() ctx: Context) {
+ *   const adapter = new KoaSSEAdapter(ctx);
+ *   writeSSE(adapter, 'start', { message: 'Hello' });
+ * }
+ * ```
+ */ 
+//# sourceMappingURL=typeorm-example.d.ts.map

package/dist/cjs/spi/examples/typeorm-example.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"typeorm-example.d.ts","sourceRoot":"","sources":["../../../../src/spi/examples/typeorm-example.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EACV,iBAAiB,EACjB,gBAAgB,EAChB,iBAAiB,EAIjB,kBAAkB,EAElB,kBAAkB,EACnB,MAAM,aAAa,CAAC;AAMrB;;;GAGG;AACH,UAAU,cAAe,SAAQ,gBAAgB;IAE/C,EAAE,EAAE,MAAM,CAAC;IAEX,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IAErB,SAAS,EAAE,IAAI,CAAC;IAChB,SAAS,EAAE,IAAI,CAAC;CACjB;AAED;;GAEG;AACH,UAAU,eAAgB,SAAQ,iBAAiB;IACjD,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,MAAM,CAAC;IAEjB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,IAAI,CAAC;CACjB;AAED;;GAEG;AACH,UAAU,gBAAiB,SAAQ,kBAAkB;IACnD,EAAE,EAAE,MAAM,CAAC;IACX,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,MAAM,CAAC;IACjB,cAAc,EAAE,KAAK,CAAC;QACpB,EAAE,EAAE,MAAM,CAAC;QACX,IAAI,EAAE,MAAM,CAAC;QACb,QAAQ,EAAE,MAAM,CAAC;QACjB,KAAK,EAAE,MAAM,CAAC;QACd,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,OAAO,CAAC,EAAE,MAAM,CAAC;KAClB,CAAC,CAAC;IACH,SAAS,EAAE,IAAI,CAAC;IAChB,SAAS,EAAE,IAAI,CAAC;CACjB;AAMD;;;GAGG;AACH,cAAM,kBAAkB,CAAC,CAAC,SAAS;IAAE,EAAE,EAAE,MAAM,GAAG,MAAM,CAAA;CAAE,CAAE,YAAW,kBAAkB,CAAC,CAAC,CAAC;IAK1F,OAAO,CAAC,IAAI,CAAW;IAEjB,IAAI,IAAI,OAAO,CAAC,CAAC,EAAE,CAAC;IAKpB,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC,CAAC,GAAG,IAAI,CAAC;IAKhD,GAAG,CAAC,MAAM,EAAE,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC;IAM1B,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,GAAG,IAAI,CAAC;IAYhE,MAAM,CAAC,EAAE,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;CASpD;AAMD;;;GAGG;AACH,cAAM,wBAAyB,YAAW,iBAAiB,CAAC,cAAc,EAAE,eAAe,CAAC;IAE1F,OAAO,CAAC,aAAa,CAAqC;IAC1D,OAAO,CAAC,eAAe,CAA+D;;IAStF,OAAO,CAAC,iBAAiB;IAUnB,UAAU,CAAC,WAAW,CAAC,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC,cAAc,EAAE,CAAC;IAUpE,SAAS,CAAC,EAAE,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC,cAAc,GAAG,SAAS,CAAC;IAOnE,YAAY,CAAC,IAAI,EAAE,OAAO,CAAC,cAAc,CAAC,GAAG,OAAO,CAAC,cAAc,CAAC;IAkBpE,YAAY,CAAC,EAAE,EAAE,MAAM,GAAG,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,cAAc,CAAC,GAAG,OAAO,CAAC,cAAc,GAAG,SAAS,CAAC;IAUrG,YAAY,CAAC,EAAE,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IASnD,WAAW,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,EAAE,OAAO,CAAC,EAAE;QAAE,KAAK,CAAC,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC;QACnG,KAAK,EAAE,eAAe,EAAE,CAAC;QACzB,KAAK,EAAE,MAAM,CAAC;QACd,OAAO,EAAE,OAAO,CAAC;KAClB,CAAC;IAiBI,UAAU,CAAC,IAAI,EAAE,OAAO,CAAC,eAAe,CAAC,GAAG;QAAE,QAAQ,EAAE,MAAM,GAAG,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,eAAe,CAAC;IAqBnI,aAAa,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,EAAE,EAAE,EAAE,MAAM,GAAG,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,eAAe,CAAC,GAAG,OAAO,CAAC,eAAe,GAAG,SAAS,CAAC;IAWnI,aAAa,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,EAAE,EAAE,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAQ/E,mBAAmB,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQ7D,kBAAkB,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;CAOpF;AAMD;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAsCG;AAMH,OAAO,EACL,kBAAkB,EAClB,wBAAwB,EACxB,cAAc,EACd,eAAe,EACf,gBAAgB,GACjB,CAAC;AAEF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA4EG"}