mozi-bot 1.0.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/LICENSE +190 -0
- package/README.md +467 -0
- package/dist/agents/agent.d.ts +118 -0
- package/dist/agents/agent.d.ts.map +1 -0
- package/dist/agents/agent.js +686 -0
- package/dist/agents/agent.js.map +1 -0
- package/dist/agents/compaction.d.ts +78 -0
- package/dist/agents/compaction.d.ts.map +1 -0
- package/dist/agents/compaction.js +350 -0
- package/dist/agents/compaction.js.map +1 -0
- package/dist/agents/failover-error.d.ts +42 -0
- package/dist/agents/failover-error.d.ts.map +1 -0
- package/dist/agents/failover-error.js +171 -0
- package/dist/agents/failover-error.js.map +1 -0
- package/dist/agents/index.d.ts +10 -0
- package/dist/agents/index.d.ts.map +1 -0
- package/dist/agents/index.js +10 -0
- package/dist/agents/index.js.map +1 -0
- package/dist/agents/model-fallback.d.ts +54 -0
- package/dist/agents/model-fallback.d.ts.map +1 -0
- package/dist/agents/model-fallback.js +227 -0
- package/dist/agents/model-fallback.js.map +1 -0
- package/dist/agents/session-store.d.ts +53 -0
- package/dist/agents/session-store.d.ts.map +1 -0
- package/dist/agents/session-store.js +217 -0
- package/dist/agents/session-store.js.map +1 -0
- package/dist/agents/system-prompt.d.ts +29 -0
- package/dist/agents/system-prompt.d.ts.map +1 -0
- package/dist/agents/system-prompt.js +299 -0
- package/dist/agents/system-prompt.js.map +1 -0
- package/dist/channels/common/base.d.ts +40 -0
- package/dist/channels/common/base.d.ts.map +1 -0
- package/dist/channels/common/base.js +23 -0
- package/dist/channels/common/base.js.map +1 -0
- package/dist/channels/common/index.d.ts +21 -0
- package/dist/channels/common/index.d.ts.map +1 -0
- package/dist/channels/common/index.js +66 -0
- package/dist/channels/common/index.js.map +1 -0
- package/dist/channels/dingtalk/api.d.ts +30 -0
- package/dist/channels/dingtalk/api.d.ts.map +1 -0
- package/dist/channels/dingtalk/api.js +149 -0
- package/dist/channels/dingtalk/api.js.map +1 -0
- package/dist/channels/dingtalk/events.d.ts +80 -0
- package/dist/channels/dingtalk/events.d.ts.map +1 -0
- package/dist/channels/dingtalk/events.js +92 -0
- package/dist/channels/dingtalk/events.js.map +1 -0
- package/dist/channels/dingtalk/index.d.ts +51 -0
- package/dist/channels/dingtalk/index.d.ts.map +1 -0
- package/dist/channels/dingtalk/index.js +231 -0
- package/dist/channels/dingtalk/index.js.map +1 -0
- package/dist/channels/dingtalk/stream.d.ts +31 -0
- package/dist/channels/dingtalk/stream.d.ts.map +1 -0
- package/dist/channels/dingtalk/stream.js +116 -0
- package/dist/channels/dingtalk/stream.js.map +1 -0
- package/dist/channels/feishu/api.d.ts +33 -0
- package/dist/channels/feishu/api.d.ts.map +1 -0
- package/dist/channels/feishu/api.js +120 -0
- package/dist/channels/feishu/api.js.map +1 -0
- package/dist/channels/feishu/events.d.ts +87 -0
- package/dist/channels/feishu/events.d.ts.map +1 -0
- package/dist/channels/feishu/events.js +163 -0
- package/dist/channels/feishu/events.js.map +1 -0
- package/dist/channels/feishu/index.d.ts +46 -0
- package/dist/channels/feishu/index.d.ts.map +1 -0
- package/dist/channels/feishu/index.js +207 -0
- package/dist/channels/feishu/index.js.map +1 -0
- package/dist/channels/feishu/websocket.d.ts +32 -0
- package/dist/channels/feishu/websocket.d.ts.map +1 -0
- package/dist/channels/feishu/websocket.js +121 -0
- package/dist/channels/feishu/websocket.js.map +1 -0
- package/dist/channels/index.d.ts +9 -0
- package/dist/channels/index.d.ts.map +1 -0
- package/dist/channels/index.js +9 -0
- package/dist/channels/index.js.map +1 -0
- package/dist/cli/index.d.ts +6 -0
- package/dist/cli/index.d.ts.map +1 -0
- package/dist/cli/index.js +468 -0
- package/dist/cli/index.js.map +1 -0
- package/dist/commands/index.d.ts +54 -0
- package/dist/commands/index.d.ts.map +1 -0
- package/dist/commands/index.js +171 -0
- package/dist/commands/index.js.map +1 -0
- package/dist/config/index.d.ts +201 -0
- package/dist/config/index.d.ts.map +1 -0
- package/dist/config/index.js +261 -0
- package/dist/config/index.js.map +1 -0
- package/dist/gateway/index.d.ts +5 -0
- package/dist/gateway/index.d.ts.map +1 -0
- package/dist/gateway/index.js +5 -0
- package/dist/gateway/index.js.map +1 -0
- package/dist/gateway/server.d.ts +42 -0
- package/dist/gateway/server.d.ts.map +1 -0
- package/dist/gateway/server.js +235 -0
- package/dist/gateway/server.js.map +1 -0
- package/dist/hooks/index.d.ts +168 -0
- package/dist/hooks/index.d.ts.map +1 -0
- package/dist/hooks/index.js +148 -0
- package/dist/hooks/index.js.map +1 -0
- package/dist/index.d.ts +18 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +29 -0
- package/dist/index.js.map +1 -0
- package/dist/memory/index.d.ts +94 -0
- package/dist/memory/index.d.ts.map +1 -0
- package/dist/memory/index.js +282 -0
- package/dist/memory/index.js.map +1 -0
- package/dist/plugins/index.d.ts +57 -0
- package/dist/plugins/index.d.ts.map +1 -0
- package/dist/plugins/index.js +103 -0
- package/dist/plugins/index.js.map +1 -0
- package/dist/providers/anthropic-compatible.d.ts +48 -0
- package/dist/providers/anthropic-compatible.d.ts.map +1 -0
- package/dist/providers/anthropic-compatible.js +380 -0
- package/dist/providers/anthropic-compatible.js.map +1 -0
- package/dist/providers/base.d.ts +26 -0
- package/dist/providers/base.d.ts.map +1 -0
- package/dist/providers/base.js +58 -0
- package/dist/providers/base.js.map +1 -0
- package/dist/providers/custom-openai.d.ts +66 -0
- package/dist/providers/custom-openai.d.ts.map +1 -0
- package/dist/providers/custom-openai.js +255 -0
- package/dist/providers/custom-openai.js.map +1 -0
- package/dist/providers/dashscope.d.ts +15 -0
- package/dist/providers/dashscope.d.ts.map +1 -0
- package/dist/providers/dashscope.js +237 -0
- package/dist/providers/dashscope.js.map +1 -0
- package/dist/providers/deepseek.d.ts +10 -0
- package/dist/providers/deepseek.d.ts.map +1 -0
- package/dist/providers/deepseek.js +57 -0
- package/dist/providers/deepseek.js.map +1 -0
- package/dist/providers/index.d.ts +33 -0
- package/dist/providers/index.d.ts.map +1 -0
- package/dist/providers/index.js +142 -0
- package/dist/providers/index.js.map +1 -0
- package/dist/providers/kimi.d.ts +10 -0
- package/dist/providers/kimi.d.ts.map +1 -0
- package/dist/providers/kimi.js +83 -0
- package/dist/providers/kimi.js.map +1 -0
- package/dist/providers/minimax.d.ts +20 -0
- package/dist/providers/minimax.d.ts.map +1 -0
- package/dist/providers/minimax.js +252 -0
- package/dist/providers/minimax.js.map +1 -0
- package/dist/providers/modelscope.d.ts +15 -0
- package/dist/providers/modelscope.d.ts.map +1 -0
- package/dist/providers/modelscope.js +237 -0
- package/dist/providers/modelscope.js.map +1 -0
- package/dist/providers/openai-compatible.d.ts +57 -0
- package/dist/providers/openai-compatible.d.ts.map +1 -0
- package/dist/providers/openai-compatible.js +193 -0
- package/dist/providers/openai-compatible.js.map +1 -0
- package/dist/providers/stepfun.d.ts +10 -0
- package/dist/providers/stepfun.d.ts.map +1 -0
- package/dist/providers/stepfun.js +125 -0
- package/dist/providers/stepfun.js.map +1 -0
- package/dist/providers/zhipu.d.ts +15 -0
- package/dist/providers/zhipu.d.ts.map +1 -0
- package/dist/providers/zhipu.js +247 -0
- package/dist/providers/zhipu.js.map +1 -0
- package/dist/sessions/index.d.ts +6 -0
- package/dist/sessions/index.d.ts.map +1 -0
- package/dist/sessions/index.js +6 -0
- package/dist/sessions/index.js.map +1 -0
- package/dist/sessions/store.d.ts +45 -0
- package/dist/sessions/store.d.ts.map +1 -0
- package/dist/sessions/store.js +268 -0
- package/dist/sessions/store.js.map +1 -0
- package/dist/sessions/types.d.ts +110 -0
- package/dist/sessions/types.d.ts.map +1 -0
- package/dist/sessions/types.js +6 -0
- package/dist/sessions/types.js.map +1 -0
- package/dist/tools/builtin/apply-patch.d.ts +8 -0
- package/dist/tools/builtin/apply-patch.d.ts.map +1 -0
- package/dist/tools/builtin/apply-patch.js +272 -0
- package/dist/tools/builtin/apply-patch.js.map +1 -0
- package/dist/tools/builtin/bash.d.ts +25 -0
- package/dist/tools/builtin/bash.d.ts.map +1 -0
- package/dist/tools/builtin/bash.js +212 -0
- package/dist/tools/builtin/bash.js.map +1 -0
- package/dist/tools/builtin/browser.d.ts +12 -0
- package/dist/tools/builtin/browser.d.ts.map +1 -0
- package/dist/tools/builtin/browser.js +693 -0
- package/dist/tools/builtin/browser.js.map +1 -0
- package/dist/tools/builtin/filesystem.d.ts +29 -0
- package/dist/tools/builtin/filesystem.d.ts.map +1 -0
- package/dist/tools/builtin/filesystem.js +484 -0
- package/dist/tools/builtin/filesystem.js.map +1 -0
- package/dist/tools/builtin/image.d.ts +13 -0
- package/dist/tools/builtin/image.d.ts.map +1 -0
- package/dist/tools/builtin/image.js +126 -0
- package/dist/tools/builtin/image.js.map +1 -0
- package/dist/tools/builtin/index.d.ts +30 -0
- package/dist/tools/builtin/index.d.ts.map +1 -0
- package/dist/tools/builtin/index.js +55 -0
- package/dist/tools/builtin/index.js.map +1 -0
- package/dist/tools/builtin/process-registry.d.ts +71 -0
- package/dist/tools/builtin/process-registry.d.ts.map +1 -0
- package/dist/tools/builtin/process-registry.js +172 -0
- package/dist/tools/builtin/process-registry.js.map +1 -0
- package/dist/tools/builtin/process-tool.d.ts +8 -0
- package/dist/tools/builtin/process-tool.d.ts.map +1 -0
- package/dist/tools/builtin/process-tool.js +279 -0
- package/dist/tools/builtin/process-tool.js.map +1 -0
- package/dist/tools/builtin/subagent.d.ts +33 -0
- package/dist/tools/builtin/subagent.d.ts.map +1 -0
- package/dist/tools/builtin/subagent.js +185 -0
- package/dist/tools/builtin/subagent.js.map +1 -0
- package/dist/tools/builtin/system.d.ts +11 -0
- package/dist/tools/builtin/system.d.ts.map +1 -0
- package/dist/tools/builtin/system.js +144 -0
- package/dist/tools/builtin/system.js.map +1 -0
- package/dist/tools/builtin/web.d.ts +9 -0
- package/dist/tools/builtin/web.d.ts.map +1 -0
- package/dist/tools/builtin/web.js +87 -0
- package/dist/tools/builtin/web.js.map +1 -0
- package/dist/tools/common.d.ts +45 -0
- package/dist/tools/common.d.ts.map +1 -0
- package/dist/tools/common.js +129 -0
- package/dist/tools/common.js.map +1 -0
- package/dist/tools/index.d.ts +8 -0
- package/dist/tools/index.d.ts.map +1 -0
- package/dist/tools/index.js +8 -0
- package/dist/tools/index.js.map +1 -0
- package/dist/tools/registry.d.ts +33 -0
- package/dist/tools/registry.d.ts.map +1 -0
- package/dist/tools/registry.js +174 -0
- package/dist/tools/registry.js.map +1 -0
- package/dist/tools/types.d.ts +58 -0
- package/dist/tools/types.d.ts.map +1 -0
- package/dist/tools/types.js +11 -0
- package/dist/tools/types.js.map +1 -0
- package/dist/types/index.d.ts +280 -0
- package/dist/types/index.d.ts.map +1 -0
- package/dist/types/index.js +34 -0
- package/dist/types/index.js.map +1 -0
- package/dist/utils/index.d.ts +35 -0
- package/dist/utils/index.d.ts.map +1 -0
- package/dist/utils/index.js +114 -0
- package/dist/utils/index.js.map +1 -0
- package/dist/utils/logger.d.ts +24 -0
- package/dist/utils/logger.d.ts.map +1 -0
- package/dist/utils/logger.js +94 -0
- package/dist/utils/logger.js.map +1 -0
- package/dist/web/index.d.ts +7 -0
- package/dist/web/index.d.ts.map +1 -0
- package/dist/web/index.js +7 -0
- package/dist/web/index.js.map +1 -0
- package/dist/web/static.d.ts +12 -0
- package/dist/web/static.d.ts.map +1 -0
- package/dist/web/static.js +1260 -0
- package/dist/web/static.js.map +1 -0
- package/dist/web/types.d.ts +95 -0
- package/dist/web/types.d.ts.map +1 -0
- package/dist/web/types.js +5 -0
- package/dist/web/types.js.map +1 -0
- package/dist/web/websocket.d.ts +58 -0
- package/dist/web/websocket.d.ts.map +1 -0
- package/dist/web/websocket.js +371 -0
- package/dist/web/websocket.js.map +1 -0
- package/package.json +89 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/gateway/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,cAAc,aAAa,CAAC"}
|
|
@@ -0,0 +1,42 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Gateway 服务器 - HTTP Webhook 处理 + WebChat
|
|
3
|
+
*/
|
|
4
|
+
import { type Express } from "express";
|
|
5
|
+
import type { MoziConfig } from "../types/index.js";
|
|
6
|
+
export declare class Gateway {
|
|
7
|
+
private app;
|
|
8
|
+
private httpServer;
|
|
9
|
+
private config;
|
|
10
|
+
private agent;
|
|
11
|
+
private feishuChannel?;
|
|
12
|
+
private dingtalkChannel?;
|
|
13
|
+
private wsServer?;
|
|
14
|
+
/** 已处理的消息 ID 缓存(用于去重) */
|
|
15
|
+
private processedMessages;
|
|
16
|
+
/** 消息缓存过期时间 (5分钟) */
|
|
17
|
+
private readonly MESSAGE_CACHE_TTL;
|
|
18
|
+
constructor(config: MoziConfig);
|
|
19
|
+
/** 设置中间件 */
|
|
20
|
+
private setupMiddleware;
|
|
21
|
+
/** 设置路由 */
|
|
22
|
+
private setupRoutes;
|
|
23
|
+
/** 处理入站消息 */
|
|
24
|
+
private handleMessage;
|
|
25
|
+
/** 发送回复 */
|
|
26
|
+
private sendReply;
|
|
27
|
+
/** 检查是否为重复消息 */
|
|
28
|
+
private isDuplicateMessage;
|
|
29
|
+
/** 初始化 */
|
|
30
|
+
initialize(): Promise<void>;
|
|
31
|
+
/** 启动服务器 */
|
|
32
|
+
start(): Promise<void>;
|
|
33
|
+
/** 关闭 */
|
|
34
|
+
shutdown(): Promise<void>;
|
|
35
|
+
/** 获取 Express 应用 */
|
|
36
|
+
getApp(): Express;
|
|
37
|
+
}
|
|
38
|
+
/** 创建 Gateway */
|
|
39
|
+
export declare function createGateway(config: MoziConfig): Gateway;
|
|
40
|
+
/** 启动 Gateway 服务器 */
|
|
41
|
+
export declare function startGateway(config: MoziConfig): Promise<Gateway>;
|
|
42
|
+
//# sourceMappingURL=server.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"server.d.ts","sourceRoot":"","sources":["../../src/gateway/server.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAgB,EAAE,KAAK,OAAO,EAA+B,MAAM,SAAS,CAAC;AAE7E,OAAO,KAAK,EAAE,UAAU,EAAyB,MAAM,mBAAmB,CAAC;AAY3E,qBAAa,OAAO;IAClB,OAAO,CAAC,GAAG,CAAU;IACrB,OAAO,CAAC,UAAU,CAAa;IAC/B,OAAO,CAAC,MAAM,CAAa;IAC3B,OAAO,CAAC,KAAK,CAAQ;IACrB,OAAO,CAAC,aAAa,CAAC,CAAgB;IACtC,OAAO,CAAC,eAAe,CAAC,CAAkB;IAC1C,OAAO,CAAC,QAAQ,CAAC,CAAW;IAC5B,yBAAyB;IACzB,OAAO,CAAC,iBAAiB,CAA6B;IACtD,qBAAqB;IACrB,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAiB;gBAEvC,MAAM,EAAE,UAAU;IAU9B,YAAY;IACZ,OAAO,CAAC,eAAe;IAWvB,WAAW;IACX,OAAO,CAAC,WAAW;IA4CnB,aAAa;YACC,aAAa;IA6C3B,WAAW;YACG,SAAS;IAkBvB,gBAAgB;IAChB,OAAO,CAAC,kBAAkB;IAoB1B,UAAU;IACJ,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAwBjC,YAAY;IACN,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;IAsB5B,SAAS;IACH,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAmB/B,oBAAoB;IACpB,MAAM,IAAI,OAAO;CAGlB;AAED,iBAAiB;AACjB,wBAAgB,aAAa,CAAC,MAAM,EAAE,UAAU,GAAG,OAAO,CAEzD;AAED,qBAAqB;AACrB,wBAAsB,YAAY,CAAC,MAAM,EAAE,UAAU,GAAG,OAAO,CAAC,OAAO,CAAC,CAqBvE"}
|
|
@@ -0,0 +1,235 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Gateway 服务器 - HTTP Webhook 处理 + WebChat
|
|
3
|
+
*/
|
|
4
|
+
import express from "express";
|
|
5
|
+
import { createServer } from "http";
|
|
6
|
+
import { createFeishuChannel } from "../channels/feishu/index.js";
|
|
7
|
+
import { createDingtalkChannel } from "../channels/dingtalk/index.js";
|
|
8
|
+
import { registerChannel } from "../channels/common/index.js";
|
|
9
|
+
import { createAgent } from "../agents/agent.js";
|
|
10
|
+
import { initializeProviders } from "../providers/index.js";
|
|
11
|
+
import { getChildLogger, setLogger, createLogger } from "../utils/logger.js";
|
|
12
|
+
import { WsServer } from "../web/websocket.js";
|
|
13
|
+
import { handleStaticRequest } from "../web/static.js";
|
|
14
|
+
const logger = getChildLogger("gateway");
|
|
15
|
+
export class Gateway {
|
|
16
|
+
app;
|
|
17
|
+
httpServer;
|
|
18
|
+
config;
|
|
19
|
+
agent;
|
|
20
|
+
feishuChannel;
|
|
21
|
+
dingtalkChannel;
|
|
22
|
+
wsServer;
|
|
23
|
+
/** 已处理的消息 ID 缓存(用于去重) */
|
|
24
|
+
processedMessages = new Map();
|
|
25
|
+
/** 消息缓存过期时间 (5分钟) */
|
|
26
|
+
MESSAGE_CACHE_TTL = 5 * 60 * 1000;
|
|
27
|
+
constructor(config) {
|
|
28
|
+
this.config = config;
|
|
29
|
+
this.app = express();
|
|
30
|
+
this.httpServer = createServer(this.app);
|
|
31
|
+
this.agent = createAgent(config);
|
|
32
|
+
this.setupMiddleware();
|
|
33
|
+
this.setupRoutes();
|
|
34
|
+
}
|
|
35
|
+
/** 设置中间件 */
|
|
36
|
+
setupMiddleware() {
|
|
37
|
+
this.app.use(express.json());
|
|
38
|
+
this.app.use(express.urlencoded({ extended: true }));
|
|
39
|
+
// 请求日志
|
|
40
|
+
this.app.use((req, res, next) => {
|
|
41
|
+
logger.debug({ method: req.method, path: req.path }, "Incoming request");
|
|
42
|
+
next();
|
|
43
|
+
});
|
|
44
|
+
}
|
|
45
|
+
/** 设置路由 */
|
|
46
|
+
setupRoutes() {
|
|
47
|
+
// 健康检查
|
|
48
|
+
this.app.get("/health", (req, res) => {
|
|
49
|
+
res.json({ status: "ok", timestamp: new Date().toISOString() });
|
|
50
|
+
});
|
|
51
|
+
// 飞书 Webhook
|
|
52
|
+
if (this.config.channels.feishu) {
|
|
53
|
+
this.feishuChannel = createFeishuChannel(this.config.channels.feishu);
|
|
54
|
+
this.feishuChannel.setMessageHandler(this.handleMessage.bind(this));
|
|
55
|
+
this.app.use("/feishu", this.feishuChannel.createRouter());
|
|
56
|
+
registerChannel(this.feishuChannel);
|
|
57
|
+
logger.info("Feishu webhook enabled at /feishu/webhook");
|
|
58
|
+
}
|
|
59
|
+
// 钉钉 Webhook
|
|
60
|
+
if (this.config.channels.dingtalk) {
|
|
61
|
+
this.dingtalkChannel = createDingtalkChannel(this.config.channels.dingtalk);
|
|
62
|
+
this.dingtalkChannel.setMessageHandler(this.handleMessage.bind(this));
|
|
63
|
+
this.app.use("/dingtalk", this.dingtalkChannel.createRouter());
|
|
64
|
+
registerChannel(this.dingtalkChannel);
|
|
65
|
+
logger.info("DingTalk webhook enabled at /dingtalk/webhook");
|
|
66
|
+
}
|
|
67
|
+
// WebChat 静态文件服务 (放在其他路由之后,作为默认处理)
|
|
68
|
+
this.app.use((req, res, next) => {
|
|
69
|
+
const handled = handleStaticRequest(req, res, { config: this.config });
|
|
70
|
+
if (!handled) {
|
|
71
|
+
next();
|
|
72
|
+
}
|
|
73
|
+
});
|
|
74
|
+
// 404 处理
|
|
75
|
+
this.app.use((req, res) => {
|
|
76
|
+
res.status(404).json({ error: "Not found" });
|
|
77
|
+
});
|
|
78
|
+
// 错误处理
|
|
79
|
+
this.app.use((err, req, res, next) => {
|
|
80
|
+
logger.error({ error: err }, "Unhandled error");
|
|
81
|
+
res.status(500).json({ error: "Internal server error" });
|
|
82
|
+
});
|
|
83
|
+
}
|
|
84
|
+
/** 处理入站消息 */
|
|
85
|
+
async handleMessage(context) {
|
|
86
|
+
// 消息去重检查
|
|
87
|
+
if (this.isDuplicateMessage(context.messageId)) {
|
|
88
|
+
logger.debug({ messageId: context.messageId }, "Skipping duplicate message");
|
|
89
|
+
return;
|
|
90
|
+
}
|
|
91
|
+
logger.info({
|
|
92
|
+
channel: context.channelId,
|
|
93
|
+
chatId: context.chatId,
|
|
94
|
+
senderId: context.senderId,
|
|
95
|
+
content: context.content.slice(0, 100),
|
|
96
|
+
}, "Received message");
|
|
97
|
+
// 忽略空消息
|
|
98
|
+
if (!context.content.trim()) {
|
|
99
|
+
return;
|
|
100
|
+
}
|
|
101
|
+
try {
|
|
102
|
+
// 处理消息
|
|
103
|
+
const response = await this.agent.processMessage(context);
|
|
104
|
+
// 发送回复
|
|
105
|
+
await this.sendReply(context, response.content);
|
|
106
|
+
logger.info({
|
|
107
|
+
channel: context.channelId,
|
|
108
|
+
chatId: context.chatId,
|
|
109
|
+
responseLength: response.content.length,
|
|
110
|
+
}, "Reply sent");
|
|
111
|
+
}
|
|
112
|
+
catch (error) {
|
|
113
|
+
logger.error({ error, context }, "Failed to process message");
|
|
114
|
+
// 发送错误提示
|
|
115
|
+
await this.sendReply(context, "抱歉,处理您的消息时出现了错误。请稍后重试。");
|
|
116
|
+
}
|
|
117
|
+
}
|
|
118
|
+
/** 发送回复 */
|
|
119
|
+
async sendReply(context, text) {
|
|
120
|
+
switch (context.channelId) {
|
|
121
|
+
case "feishu":
|
|
122
|
+
if (this.feishuChannel) {
|
|
123
|
+
await this.feishuChannel.sendText(context.chatId, text, context.messageId);
|
|
124
|
+
}
|
|
125
|
+
break;
|
|
126
|
+
case "dingtalk":
|
|
127
|
+
if (this.dingtalkChannel) {
|
|
128
|
+
// 尝试使用 session webhook 回复
|
|
129
|
+
const dingtalkContext = context;
|
|
130
|
+
await this.dingtalkChannel.replyWithSession(dingtalkContext, text);
|
|
131
|
+
}
|
|
132
|
+
break;
|
|
133
|
+
}
|
|
134
|
+
}
|
|
135
|
+
/** 检查是否为重复消息 */
|
|
136
|
+
isDuplicateMessage(messageId) {
|
|
137
|
+
const now = Date.now();
|
|
138
|
+
// 清理过期的消息缓存
|
|
139
|
+
for (const [id, timestamp] of this.processedMessages) {
|
|
140
|
+
if (now - timestamp > this.MESSAGE_CACHE_TTL) {
|
|
141
|
+
this.processedMessages.delete(id);
|
|
142
|
+
}
|
|
143
|
+
}
|
|
144
|
+
// 检查消息是否已处理
|
|
145
|
+
if (this.processedMessages.has(messageId)) {
|
|
146
|
+
return true;
|
|
147
|
+
}
|
|
148
|
+
// 标记为已处理
|
|
149
|
+
this.processedMessages.set(messageId, now);
|
|
150
|
+
return false;
|
|
151
|
+
}
|
|
152
|
+
/** 初始化 */
|
|
153
|
+
async initialize() {
|
|
154
|
+
logger.info("Initializing gateway...");
|
|
155
|
+
// 初始化模型提供商
|
|
156
|
+
initializeProviders(this.config);
|
|
157
|
+
// 初始化 WebSocket 服务器
|
|
158
|
+
this.wsServer = new WsServer({
|
|
159
|
+
server: this.httpServer,
|
|
160
|
+
agent: this.agent,
|
|
161
|
+
config: this.config,
|
|
162
|
+
});
|
|
163
|
+
// 初始化通道
|
|
164
|
+
if (this.feishuChannel) {
|
|
165
|
+
await this.feishuChannel.initialize();
|
|
166
|
+
}
|
|
167
|
+
if (this.dingtalkChannel) {
|
|
168
|
+
await this.dingtalkChannel.initialize();
|
|
169
|
+
}
|
|
170
|
+
logger.info("Gateway initialized");
|
|
171
|
+
}
|
|
172
|
+
/** 启动服务器 */
|
|
173
|
+
async start() {
|
|
174
|
+
await this.initialize();
|
|
175
|
+
const { port, host } = this.config.server;
|
|
176
|
+
this.httpServer.listen(port, host || "0.0.0.0", () => {
|
|
177
|
+
logger.info({ port, host: host || "0.0.0.0" }, "Gateway server started");
|
|
178
|
+
console.log(`\n🚀 Mozi Gateway 已启动`);
|
|
179
|
+
console.log(` 地址: http://${host || "localhost"}:${port}`);
|
|
180
|
+
console.log(` WebChat: http://${host || "localhost"}:${port}/`);
|
|
181
|
+
console.log(` 控制台: http://${host || "localhost"}:${port}/control`);
|
|
182
|
+
console.log(` 健康检查: http://${host || "localhost"}:${port}/health`);
|
|
183
|
+
if (this.feishuChannel) {
|
|
184
|
+
console.log(` 飞书 Webhook: http://${host || "localhost"}:${port}/feishu/webhook`);
|
|
185
|
+
}
|
|
186
|
+
if (this.dingtalkChannel) {
|
|
187
|
+
console.log(` 钉钉 Webhook: http://${host || "localhost"}:${port}/dingtalk/webhook`);
|
|
188
|
+
}
|
|
189
|
+
console.log("");
|
|
190
|
+
});
|
|
191
|
+
}
|
|
192
|
+
/** 关闭 */
|
|
193
|
+
async shutdown() {
|
|
194
|
+
logger.info("Shutting down gateway...");
|
|
195
|
+
if (this.wsServer) {
|
|
196
|
+
this.wsServer.close();
|
|
197
|
+
}
|
|
198
|
+
if (this.feishuChannel) {
|
|
199
|
+
await this.feishuChannel.shutdown();
|
|
200
|
+
}
|
|
201
|
+
if (this.dingtalkChannel) {
|
|
202
|
+
await this.dingtalkChannel.shutdown();
|
|
203
|
+
}
|
|
204
|
+
this.httpServer.close();
|
|
205
|
+
logger.info("Gateway shut down");
|
|
206
|
+
}
|
|
207
|
+
/** 获取 Express 应用 */
|
|
208
|
+
getApp() {
|
|
209
|
+
return this.app;
|
|
210
|
+
}
|
|
211
|
+
}
|
|
212
|
+
/** 创建 Gateway */
|
|
213
|
+
export function createGateway(config) {
|
|
214
|
+
return new Gateway(config);
|
|
215
|
+
}
|
|
216
|
+
/** 启动 Gateway 服务器 */
|
|
217
|
+
export async function startGateway(config) {
|
|
218
|
+
// 设置日志
|
|
219
|
+
setLogger(createLogger({ level: config.logging.level }));
|
|
220
|
+
const gateway = createGateway(config);
|
|
221
|
+
await gateway.start();
|
|
222
|
+
// 优雅关闭
|
|
223
|
+
process.on("SIGINT", async () => {
|
|
224
|
+
console.log("\n收到 SIGINT 信号,正在关闭...");
|
|
225
|
+
await gateway.shutdown();
|
|
226
|
+
process.exit(0);
|
|
227
|
+
});
|
|
228
|
+
process.on("SIGTERM", async () => {
|
|
229
|
+
console.log("\n收到 SIGTERM 信号,正在关闭...");
|
|
230
|
+
await gateway.shutdown();
|
|
231
|
+
process.exit(0);
|
|
232
|
+
});
|
|
233
|
+
return gateway;
|
|
234
|
+
}
|
|
235
|
+
//# sourceMappingURL=server.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"server.js","sourceRoot":"","sources":["../../src/gateway/server.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,OAAsD,MAAM,SAAS,CAAC;AAC7E,OAAO,EAAE,YAAY,EAA6B,MAAM,MAAM,CAAC;AAE/D,OAAO,EAAE,mBAAmB,EAAsB,MAAM,6BAA6B,CAAC;AACtF,OAAO,EAAE,qBAAqB,EAAwB,MAAM,+BAA+B,CAAC;AAC5F,OAAO,EAAE,eAAe,EAAE,MAAM,6BAA6B,CAAC;AAC9D,OAAO,EAAE,WAAW,EAAc,MAAM,oBAAoB,CAAC;AAC7D,OAAO,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAC5D,OAAO,EAAE,cAAc,EAAE,SAAS,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAC7E,OAAO,EAAE,QAAQ,EAAE,MAAM,qBAAqB,CAAC;AAC/C,OAAO,EAAE,mBAAmB,EAAE,MAAM,kBAAkB,CAAC;AAEvD,MAAM,MAAM,GAAG,cAAc,CAAC,SAAS,CAAC,CAAC;AAEzC,MAAM,OAAO,OAAO;IACV,GAAG,CAAU;IACb,UAAU,CAAa;IACvB,MAAM,CAAa;IACnB,KAAK,CAAQ;IACb,aAAa,CAAiB;IAC9B,eAAe,CAAmB;IAClC,QAAQ,CAAY;IAC5B,yBAAyB;IACjB,iBAAiB,GAAG,IAAI,GAAG,EAAkB,CAAC;IACtD,qBAAqB;IACJ,iBAAiB,GAAG,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;IAEnD,YAAY,MAAkB;QAC5B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,GAAG,GAAG,OAAO,EAAE,CAAC;QACrB,IAAI,CAAC,UAAU,GAAG,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACzC,IAAI,CAAC,KAAK,GAAG,WAAW,CAAC,MAAM,CAAC,CAAC;QAEjC,IAAI,CAAC,eAAe,EAAE,CAAC;QACvB,IAAI,CAAC,WAAW,EAAE,CAAC;IACrB,CAAC;IAED,YAAY;IACJ,eAAe;QACrB,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC;QAC7B,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;QAErD,OAAO;QACP,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;YAC9B,MAAM,CAAC,KAAK,CAAC,EAAE,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,IAAI,EAAE,GAAG,CAAC,IAAI,EAAE,EAAE,kBAAkB,CAAC,CAAC;YACzE,IAAI,EAAE,CAAC;QACT,CAAC,CAAC,CAAC;IACL,CAAC;IAED,WAAW;IACH,WAAW;QACjB,OAAO;QACP,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,SAAS,EAAE,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;YACnC,GAAG,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;QAEH,aAAa;QACb,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC;YAChC,IAAI,CAAC,aAAa,GAAG,mBAAmB,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC;YACtE,IAAI,CAAC,aAAa,CAAC,iBAAiB,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YACpE,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,YAAY,EAAE,CAAC,CAAC;YAC3D,eAAe,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YACpC,MAAM,CAAC,IAAI,CAAC,2CAA2C,CAAC,CAAC;QAC3D,CAAC;QAED,aAAa;QACb,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC;YAClC,IAAI,CAAC,eAAe,GAAG,qBAAqB,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;YAC5E,IAAI,CAAC,eAAe,CAAC,iBAAiB,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YACtE,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,WAAW,EAAE,IAAI,CAAC,eAAe,CAAC,YAAY,EAAE,CAAC,CAAC;YAC/D,eAAe,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;YACtC,MAAM,CAAC,IAAI,CAAC,+CAA+C,CAAC,CAAC;QAC/D,CAAC;QAED,mCAAmC;QACnC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE;YAC9B,MAAM,OAAO,GAAG,mBAAmB,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;YACvE,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,IAAI,EAAE,CAAC;YACT,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,SAAS;QACT,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;YACxB,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,OAAO;QACP,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,GAAU,EAAE,GAAY,EAAE,GAAa,EAAE,IAAc,EAAE,EAAE;YACvE,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,iBAAiB,CAAC,CAAC;YAChD,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,uBAAuB,EAAE,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;IACL,CAAC;IAED,aAAa;IACL,KAAK,CAAC,aAAa,CAAC,OAA8B;QACxD,SAAS;QACT,IAAI,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;YAC/C,MAAM,CAAC,KAAK,CAAC,EAAE,SAAS,EAAE,OAAO,CAAC,SAAS,EAAE,EAAE,4BAA4B,CAAC,CAAC;YAC7E,OAAO;QACT,CAAC;QAED,MAAM,CAAC,IAAI,CACT;YACE,OAAO,EAAE,OAAO,CAAC,SAAS;YAC1B,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,QAAQ,EAAE,OAAO,CAAC,QAAQ;YAC1B,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;SACvC,EACD,kBAAkB,CACnB,CAAC;QAEF,QAAQ;QACR,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;YAC5B,OAAO;QACT,CAAC;QAED,IAAI,CAAC;YACH,OAAO;YACP,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;YAE1D,OAAO;YACP,MAAM,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,QAAQ,CAAC,OAAO,CAAC,CAAC;YAEhD,MAAM,CAAC,IAAI,CACT;gBACE,OAAO,EAAE,OAAO,CAAC,SAAS;gBAC1B,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,cAAc,EAAE,QAAQ,CAAC,OAAO,CAAC,MAAM;aACxC,EACD,YAAY,CACb,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,2BAA2B,CAAC,CAAC;YAE9D,SAAS;YACT,MAAM,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,wBAAwB,CAAC,CAAC;QAC1D,CAAC;IACH,CAAC;IAED,WAAW;IACH,KAAK,CAAC,SAAS,CAAC,OAA8B,EAAE,IAAY;QAClE,QAAQ,OAAO,CAAC,SAAS,EAAE,CAAC;YAC1B,KAAK,QAAQ;gBACX,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;oBACvB,MAAM,IAAI,CAAC,aAAa,CAAC,QAAQ,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;gBAC7E,CAAC;gBACD,MAAM;YAER,KAAK,UAAU;gBACb,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;oBACzB,0BAA0B;oBAC1B,MAAM,eAAe,GAAG,OAA8D,CAAC;oBACvF,MAAM,IAAI,CAAC,eAAe,CAAC,gBAAgB,CAAC,eAAe,EAAE,IAAI,CAAC,CAAC;gBACrE,CAAC;gBACD,MAAM;QACV,CAAC;IACH,CAAC;IAED,gBAAgB;IACR,kBAAkB,CAAC,SAAiB;QAC1C,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAEvB,YAAY;QACZ,KAAK,MAAM,CAAC,EAAE,EAAE,SAAS,CAAC,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACrD,IAAI,GAAG,GAAG,SAAS,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBAC7C,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YACpC,CAAC;QACH,CAAC;QAED,YAAY;QACZ,IAAI,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC;YAC1C,OAAO,IAAI,CAAC;QACd,CAAC;QAED,SAAS;QACT,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;QAC3C,OAAO,KAAK,CAAC;IACf,CAAC;IAED,UAAU;IACV,KAAK,CAAC,UAAU;QACd,MAAM,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QAEvC,WAAW;QACX,mBAAmB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAEjC,oBAAoB;QACpB,IAAI,CAAC,QAAQ,GAAG,IAAI,QAAQ,CAAC;YAC3B,MAAM,EAAE,IAAI,CAAC,UAAU;YACvB,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAC,CAAC;QAEH,QAAQ;QACR,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACvB,MAAM,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,CAAC;QACxC,CAAC;QACD,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,MAAM,IAAI,CAAC,eAAe,CAAC,UAAU,EAAE,CAAC;QAC1C,CAAC;QAED,MAAM,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;IACrC,CAAC;IAED,YAAY;IACZ,KAAK,CAAC,KAAK;QACT,MAAM,IAAI,CAAC,UAAU,EAAE,CAAC;QAExB,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;QAE1C,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,IAAI,SAAS,EAAE,GAAG,EAAE;YACnD,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,IAAI,SAAS,EAAE,EAAE,wBAAwB,CAAC,CAAC;YACzE,OAAO,CAAC,GAAG,CAAC,uBAAuB,CAAC,CAAC;YACrC,OAAO,CAAC,GAAG,CAAC,iBAAiB,IAAI,IAAI,WAAW,IAAI,IAAI,EAAE,CAAC,CAAC;YAC5D,OAAO,CAAC,GAAG,CAAC,sBAAsB,IAAI,IAAI,WAAW,IAAI,IAAI,GAAG,CAAC,CAAC;YAClE,OAAO,CAAC,GAAG,CAAC,kBAAkB,IAAI,IAAI,WAAW,IAAI,IAAI,UAAU,CAAC,CAAC;YACrE,OAAO,CAAC,GAAG,CAAC,mBAAmB,IAAI,IAAI,WAAW,IAAI,IAAI,SAAS,CAAC,CAAC;YACrE,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;gBACvB,OAAO,CAAC,GAAG,CAAC,yBAAyB,IAAI,IAAI,WAAW,IAAI,IAAI,iBAAiB,CAAC,CAAC;YACrF,CAAC;YACD,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;gBACzB,OAAO,CAAC,GAAG,CAAC,yBAAyB,IAAI,IAAI,WAAW,IAAI,IAAI,mBAAmB,CAAC,CAAC;YACvF,CAAC;YACD,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAClB,CAAC,CAAC,CAAC;IACL,CAAC;IAED,SAAS;IACT,KAAK,CAAC,QAAQ;QACZ,MAAM,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;QAExC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QACxB,CAAC;QAED,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YACvB,MAAM,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,CAAC;QACtC,CAAC;QACD,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACzB,MAAM,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,CAAC;QACxC,CAAC;QAED,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;QAExB,MAAM,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;IACnC,CAAC;IAED,oBAAoB;IACpB,MAAM;QACJ,OAAO,IAAI,CAAC,GAAG,CAAC;IAClB,CAAC;CACF;AAED,iBAAiB;AACjB,MAAM,UAAU,aAAa,CAAC,MAAkB;IAC9C,OAAO,IAAI,OAAO,CAAC,MAAM,CAAC,CAAC;AAC7B,CAAC;AAED,qBAAqB;AACrB,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,MAAkB;IACnD,OAAO;IACP,SAAS,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;IAEzD,MAAM,OAAO,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC;IACtC,MAAM,OAAO,CAAC,KAAK,EAAE,CAAC;IAEtB,OAAO;IACP,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,IAAI,EAAE;QAC9B,OAAO,CAAC,GAAG,CAAC,wBAAwB,CAAC,CAAC;QACtC,MAAM,OAAO,CAAC,QAAQ,EAAE,CAAC;QACzB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC,CAAC;IAEH,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,KAAK,IAAI,EAAE;QAC/B,OAAO,CAAC,GAAG,CAAC,yBAAyB,CAAC,CAAC;QACvC,MAAM,OAAO,CAAC,QAAQ,EAAE,CAAC;QACzB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC,CAAC,CAAC;IAEH,OAAO,OAAO,CAAC;AACjB,CAAC"}
|
|
@@ -0,0 +1,168 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Hooks 系统 - 事件钩子
|
|
3
|
+
*/
|
|
4
|
+
import type { InboundMessageContext, ChatMessage, ProviderId } from "../types/index.js";
|
|
5
|
+
/** Hook 事件类型 */
|
|
6
|
+
export type HookEventType = "message_received" | "message_sending" | "message_sent" | "agent_start" | "agent_end" | "tool_start" | "tool_end" | "session_start" | "session_end" | "compaction_start" | "compaction_end" | "error";
|
|
7
|
+
/** Hook 事件数据基础 */
|
|
8
|
+
interface HookEventBase {
|
|
9
|
+
type: HookEventType;
|
|
10
|
+
timestamp: number;
|
|
11
|
+
sessionKey?: string;
|
|
12
|
+
}
|
|
13
|
+
/** 消息接收事件 */
|
|
14
|
+
export interface MessageReceivedEvent extends HookEventBase {
|
|
15
|
+
type: "message_received";
|
|
16
|
+
context: InboundMessageContext;
|
|
17
|
+
}
|
|
18
|
+
/** 消息发送事件 */
|
|
19
|
+
export interface MessageSendingEvent extends HookEventBase {
|
|
20
|
+
type: "message_sending";
|
|
21
|
+
channelId: string;
|
|
22
|
+
chatId: string;
|
|
23
|
+
content: string;
|
|
24
|
+
replyToId?: string;
|
|
25
|
+
}
|
|
26
|
+
/** 消息已发送事件 */
|
|
27
|
+
export interface MessageSentEvent extends HookEventBase {
|
|
28
|
+
type: "message_sent";
|
|
29
|
+
channelId: string;
|
|
30
|
+
chatId: string;
|
|
31
|
+
messageId?: string;
|
|
32
|
+
success: boolean;
|
|
33
|
+
}
|
|
34
|
+
/** Agent 开始事件 */
|
|
35
|
+
export interface AgentStartEvent extends HookEventBase {
|
|
36
|
+
type: "agent_start";
|
|
37
|
+
provider: ProviderId;
|
|
38
|
+
model: string;
|
|
39
|
+
messages: ChatMessage[];
|
|
40
|
+
}
|
|
41
|
+
/** Agent 结束事件 */
|
|
42
|
+
export interface AgentEndEvent extends HookEventBase {
|
|
43
|
+
type: "agent_end";
|
|
44
|
+
provider: ProviderId;
|
|
45
|
+
model: string;
|
|
46
|
+
response: string;
|
|
47
|
+
usage?: {
|
|
48
|
+
promptTokens: number;
|
|
49
|
+
completionTokens: number;
|
|
50
|
+
totalTokens: number;
|
|
51
|
+
};
|
|
52
|
+
durationMs: number;
|
|
53
|
+
}
|
|
54
|
+
/** 工具开始事件 */
|
|
55
|
+
export interface ToolStartEvent extends HookEventBase {
|
|
56
|
+
type: "tool_start";
|
|
57
|
+
toolName: string;
|
|
58
|
+
toolCallId: string;
|
|
59
|
+
arguments: unknown;
|
|
60
|
+
}
|
|
61
|
+
/** 工具结束事件 */
|
|
62
|
+
export interface ToolEndEvent extends HookEventBase {
|
|
63
|
+
type: "tool_end";
|
|
64
|
+
toolName: string;
|
|
65
|
+
toolCallId: string;
|
|
66
|
+
result: unknown;
|
|
67
|
+
isError: boolean;
|
|
68
|
+
durationMs: number;
|
|
69
|
+
}
|
|
70
|
+
/** 会话开始事件 */
|
|
71
|
+
export interface SessionStartEvent extends HookEventBase {
|
|
72
|
+
type: "session_start";
|
|
73
|
+
channelId: string;
|
|
74
|
+
chatId: string;
|
|
75
|
+
senderId: string;
|
|
76
|
+
}
|
|
77
|
+
/** 会话结束事件 */
|
|
78
|
+
export interface SessionEndEvent extends HookEventBase {
|
|
79
|
+
type: "session_end";
|
|
80
|
+
messageCount: number;
|
|
81
|
+
totalTokens: number;
|
|
82
|
+
}
|
|
83
|
+
/** 压缩开始事件 */
|
|
84
|
+
export interface CompactionStartEvent extends HookEventBase {
|
|
85
|
+
type: "compaction_start";
|
|
86
|
+
messageCount: number;
|
|
87
|
+
estimatedTokens: number;
|
|
88
|
+
}
|
|
89
|
+
/** 压缩结束事件 */
|
|
90
|
+
export interface CompactionEndEvent extends HookEventBase {
|
|
91
|
+
type: "compaction_end";
|
|
92
|
+
compactedMessages: number;
|
|
93
|
+
summaryLength: number;
|
|
94
|
+
durationMs: number;
|
|
95
|
+
}
|
|
96
|
+
/** 错误事件 */
|
|
97
|
+
export interface ErrorEvent extends HookEventBase {
|
|
98
|
+
type: "error";
|
|
99
|
+
error: Error;
|
|
100
|
+
context?: string;
|
|
101
|
+
}
|
|
102
|
+
/** 所有事件类型 */
|
|
103
|
+
export type HookEvent = MessageReceivedEvent | MessageSendingEvent | MessageSentEvent | AgentStartEvent | AgentEndEvent | ToolStartEvent | ToolEndEvent | SessionStartEvent | SessionEndEvent | CompactionStartEvent | CompactionEndEvent | ErrorEvent;
|
|
104
|
+
/** Hook 处理器 */
|
|
105
|
+
export type HookHandler<T extends HookEvent = HookEvent> = (event: T) => void | Promise<void>;
|
|
106
|
+
/** Hook 处理器带返回值 (用于修改事件) */
|
|
107
|
+
export type HookTransformer<T extends HookEvent = HookEvent> = (event: T) => T | Promise<T>;
|
|
108
|
+
/** 注册 Hook */
|
|
109
|
+
export declare function registerHook<T extends HookEventType>(eventType: T, handler: HookHandler): () => void;
|
|
110
|
+
/** 批量注册 Hooks */
|
|
111
|
+
export declare function registerHooks(hooks: Partial<Record<HookEventType, HookHandler>>): () => void;
|
|
112
|
+
/** 触发 Hook */
|
|
113
|
+
export declare function triggerHook(event: HookEvent): Promise<void>;
|
|
114
|
+
/** 同步触发 Hook (不等待) */
|
|
115
|
+
export declare function triggerHookSync(event: HookEvent): void;
|
|
116
|
+
/** 清除所有 Hooks */
|
|
117
|
+
export declare function clearHooks(): void;
|
|
118
|
+
/** 获取已注册的 Hook 数量 */
|
|
119
|
+
export declare function getHookCount(eventType?: HookEventType): number;
|
|
120
|
+
/** 触发消息接收事件 */
|
|
121
|
+
export declare function emitMessageReceived(context: InboundMessageContext): void;
|
|
122
|
+
/** 触发消息发送事件 */
|
|
123
|
+
export declare function emitMessageSending(params: {
|
|
124
|
+
channelId: string;
|
|
125
|
+
chatId: string;
|
|
126
|
+
content: string;
|
|
127
|
+
replyToId?: string;
|
|
128
|
+
sessionKey?: string;
|
|
129
|
+
}): void;
|
|
130
|
+
/** 触发 Agent 开始事件 */
|
|
131
|
+
export declare function emitAgentStart(params: {
|
|
132
|
+
provider: ProviderId;
|
|
133
|
+
model: string;
|
|
134
|
+
messages: ChatMessage[];
|
|
135
|
+
sessionKey?: string;
|
|
136
|
+
}): void;
|
|
137
|
+
/** 触发 Agent 结束事件 */
|
|
138
|
+
export declare function emitAgentEnd(params: {
|
|
139
|
+
provider: ProviderId;
|
|
140
|
+
model: string;
|
|
141
|
+
response: string;
|
|
142
|
+
usage?: {
|
|
143
|
+
promptTokens: number;
|
|
144
|
+
completionTokens: number;
|
|
145
|
+
totalTokens: number;
|
|
146
|
+
};
|
|
147
|
+
durationMs: number;
|
|
148
|
+
sessionKey?: string;
|
|
149
|
+
}): void;
|
|
150
|
+
/** 触发工具事件 */
|
|
151
|
+
export declare function emitToolStart(params: {
|
|
152
|
+
toolName: string;
|
|
153
|
+
toolCallId: string;
|
|
154
|
+
arguments: unknown;
|
|
155
|
+
sessionKey?: string;
|
|
156
|
+
}): void;
|
|
157
|
+
export declare function emitToolEnd(params: {
|
|
158
|
+
toolName: string;
|
|
159
|
+
toolCallId: string;
|
|
160
|
+
result: unknown;
|
|
161
|
+
isError: boolean;
|
|
162
|
+
durationMs: number;
|
|
163
|
+
sessionKey?: string;
|
|
164
|
+
}): void;
|
|
165
|
+
/** 触发错误事件 */
|
|
166
|
+
export declare function emitError(error: Error, context?: string): void;
|
|
167
|
+
export {};
|
|
168
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/hooks/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAGH,OAAO,KAAK,EAAE,qBAAqB,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAMxF,gBAAgB;AAChB,MAAM,MAAM,aAAa,GACrB,kBAAkB,GAClB,iBAAiB,GACjB,cAAc,GACd,aAAa,GACb,WAAW,GACX,YAAY,GACZ,UAAU,GACV,eAAe,GACf,aAAa,GACb,kBAAkB,GAClB,gBAAgB,GAChB,OAAO,CAAC;AAEZ,kBAAkB;AAClB,UAAU,aAAa;IACrB,IAAI,EAAE,aAAa,CAAC;IACpB,SAAS,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,aAAa;AACb,MAAM,WAAW,oBAAqB,SAAQ,aAAa;IACzD,IAAI,EAAE,kBAAkB,CAAC;IACzB,OAAO,EAAE,qBAAqB,CAAC;CAChC;AAED,aAAa;AACb,MAAM,WAAW,mBAAoB,SAAQ,aAAa;IACxD,IAAI,EAAE,iBAAiB,CAAC;IACxB,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAED,cAAc;AACd,MAAM,WAAW,gBAAiB,SAAQ,aAAa;IACrD,IAAI,EAAE,cAAc,CAAC;IACrB,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,OAAO,EAAE,OAAO,CAAC;CAClB;AAED,iBAAiB;AACjB,MAAM,WAAW,eAAgB,SAAQ,aAAa;IACpD,IAAI,EAAE,aAAa,CAAC;IACpB,QAAQ,EAAE,UAAU,CAAC;IACrB,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,WAAW,EAAE,CAAC;CACzB;AAED,iBAAiB;AACjB,MAAM,WAAW,aAAc,SAAQ,aAAa;IAClD,IAAI,EAAE,WAAW,CAAC;IAClB,QAAQ,EAAE,UAAU,CAAC;IACrB,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE;QACN,YAAY,EAAE,MAAM,CAAC;QACrB,gBAAgB,EAAE,MAAM,CAAC;QACzB,WAAW,EAAE,MAAM,CAAC;KACrB,CAAC;IACF,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,aAAa;AACb,MAAM,WAAW,cAAe,SAAQ,aAAa;IACnD,IAAI,EAAE,YAAY,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,OAAO,CAAC;CACpB;AAED,aAAa;AACb,MAAM,WAAW,YAAa,SAAQ,aAAa;IACjD,IAAI,EAAE,UAAU,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,OAAO,CAAC;IAChB,OAAO,EAAE,OAAO,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,aAAa;AACb,MAAM,WAAW,iBAAkB,SAAQ,aAAa;IACtD,IAAI,EAAE,eAAe,CAAC;IACtB,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,QAAQ,EAAE,MAAM,CAAC;CAClB;AAED,aAAa;AACb,MAAM,WAAW,eAAgB,SAAQ,aAAa;IACpD,IAAI,EAAE,aAAa,CAAC;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;CACrB;AAED,aAAa;AACb,MAAM,WAAW,oBAAqB,SAAQ,aAAa;IACzD,IAAI,EAAE,kBAAkB,CAAC;IACzB,YAAY,EAAE,MAAM,CAAC;IACrB,eAAe,EAAE,MAAM,CAAC;CACzB;AAED,aAAa;AACb,MAAM,WAAW,kBAAmB,SAAQ,aAAa;IACvD,IAAI,EAAE,gBAAgB,CAAC;IACvB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,aAAa,EAAE,MAAM,CAAC;IACtB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,WAAW;AACX,MAAM,WAAW,UAAW,SAAQ,aAAa;IAC/C,IAAI,EAAE,OAAO,CAAC;IACd,KAAK,EAAE,KAAK,CAAC;IACb,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,aAAa;AACb,MAAM,MAAM,SAAS,GACjB,oBAAoB,GACpB,mBAAmB,GACnB,gBAAgB,GAChB,eAAe,GACf,aAAa,GACb,cAAc,GACd,YAAY,GACZ,iBAAiB,GACjB,eAAe,GACf,oBAAoB,GACpB,kBAAkB,GAClB,UAAU,CAAC;AAIf,eAAe;AACf,MAAM,MAAM,WAAW,CAAC,CAAC,SAAS,SAAS,GAAG,SAAS,IAAI,CACzD,KAAK,EAAE,CAAC,KACL,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC;AAE1B,4BAA4B;AAC5B,MAAM,MAAM,eAAe,CAAC,CAAC,SAAS,SAAS,GAAG,SAAS,IAAI,CAC7D,KAAK,EAAE,CAAC,KACL,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;AAOpB,cAAc;AACd,wBAAgB,YAAY,CAAC,CAAC,SAAS,aAAa,EAClD,SAAS,EAAE,CAAC,EACZ,OAAO,EAAE,WAAW,GACnB,MAAM,IAAI,CAiBZ;AAED,iBAAiB;AACjB,wBAAgB,aAAa,CAC3B,KAAK,EAAE,OAAO,CAAC,MAAM,CAAC,aAAa,EAAE,WAAW,CAAC,CAAC,GACjD,MAAM,IAAI,CAcZ;AAED,cAAc;AACd,wBAAsB,WAAW,CAAC,KAAK,EAAE,SAAS,GAAG,OAAO,CAAC,IAAI,CAAC,CAWjE;AAED,sBAAsB;AACtB,wBAAgB,eAAe,CAAC,KAAK,EAAE,SAAS,GAAG,IAAI,CAItD;AAED,iBAAiB;AACjB,wBAAgB,UAAU,IAAI,IAAI,CAEjC;AAED,qBAAqB;AACrB,wBAAgB,YAAY,CAAC,SAAS,CAAC,EAAE,aAAa,GAAG,MAAM,CAS9D;AAgBD,eAAe;AACf,wBAAgB,mBAAmB,CAAC,OAAO,EAAE,qBAAqB,GAAG,IAAI,CAKxE;AAED,eAAe;AACf,wBAAgB,kBAAkB,CAAC,MAAM,EAAE;IACzC,SAAS,EAAE,MAAM,CAAC;IAClB,MAAM,EAAE,MAAM,CAAC;IACf,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB,GAAG,IAAI,CAQP;AAED,oBAAoB;AACpB,wBAAgB,cAAc,CAAC,MAAM,EAAE;IACrC,QAAQ,EAAE,UAAU,CAAC;IACrB,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,WAAW,EAAE,CAAC;IACxB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB,GAAG,IAAI,CAOP;AAED,oBAAoB;AACpB,wBAAgB,YAAY,CAAC,MAAM,EAAE;IACnC,QAAQ,EAAE,UAAU,CAAC;IACrB,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE;QAAE,YAAY,EAAE,MAAM,CAAC;QAAC,gBAAgB,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAA;KAAE,CAAC;IAChF,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB,GAAG,IAAI,CASP;AAED,aAAa;AACb,wBAAgB,aAAa,CAAC,MAAM,EAAE;IACpC,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,OAAO,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB,GAAG,IAAI,CAOP;AAED,wBAAgB,WAAW,CAAC,MAAM,EAAE;IAClC,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,MAAM,EAAE,OAAO,CAAC;IAChB,OAAO,EAAE,OAAO,CAAC;IACjB,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB,GAAG,IAAI,CASP;AAED,aAAa;AACb,wBAAgB,SAAS,CAAC,KAAK,EAAE,KAAK,EAAE,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAM9D"}
|
|
@@ -0,0 +1,148 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Hooks 系统 - 事件钩子
|
|
3
|
+
*/
|
|
4
|
+
import { getChildLogger } from "../utils/logger.js";
|
|
5
|
+
const logger = getChildLogger("hooks");
|
|
6
|
+
/** Hook 注册表 */
|
|
7
|
+
const hookRegistry = new Map();
|
|
8
|
+
// ============== Hook 管理 ==============
|
|
9
|
+
/** 注册 Hook */
|
|
10
|
+
export function registerHook(eventType, handler) {
|
|
11
|
+
const handlers = hookRegistry.get(eventType) ?? [];
|
|
12
|
+
handlers.push(handler);
|
|
13
|
+
hookRegistry.set(eventType, handlers);
|
|
14
|
+
logger.debug({ eventType }, "Hook registered");
|
|
15
|
+
// 返回取消注册函数
|
|
16
|
+
return () => {
|
|
17
|
+
const currentHandlers = hookRegistry.get(eventType);
|
|
18
|
+
if (currentHandlers) {
|
|
19
|
+
const index = currentHandlers.indexOf(handler);
|
|
20
|
+
if (index >= 0) {
|
|
21
|
+
currentHandlers.splice(index, 1);
|
|
22
|
+
}
|
|
23
|
+
}
|
|
24
|
+
};
|
|
25
|
+
}
|
|
26
|
+
/** 批量注册 Hooks */
|
|
27
|
+
export function registerHooks(hooks) {
|
|
28
|
+
const unsubscribers = [];
|
|
29
|
+
for (const [eventType, handler] of Object.entries(hooks)) {
|
|
30
|
+
if (handler) {
|
|
31
|
+
unsubscribers.push(registerHook(eventType, handler));
|
|
32
|
+
}
|
|
33
|
+
}
|
|
34
|
+
return () => {
|
|
35
|
+
for (const unsubscribe of unsubscribers) {
|
|
36
|
+
unsubscribe();
|
|
37
|
+
}
|
|
38
|
+
};
|
|
39
|
+
}
|
|
40
|
+
/** 触发 Hook */
|
|
41
|
+
export async function triggerHook(event) {
|
|
42
|
+
const handlers = hookRegistry.get(event.type);
|
|
43
|
+
if (!handlers || handlers.length === 0)
|
|
44
|
+
return;
|
|
45
|
+
for (const handler of handlers) {
|
|
46
|
+
try {
|
|
47
|
+
await handler(event);
|
|
48
|
+
}
|
|
49
|
+
catch (error) {
|
|
50
|
+
logger.error({ error, eventType: event.type }, "Hook handler error");
|
|
51
|
+
}
|
|
52
|
+
}
|
|
53
|
+
}
|
|
54
|
+
/** 同步触发 Hook (不等待) */
|
|
55
|
+
export function triggerHookSync(event) {
|
|
56
|
+
triggerHook(event).catch((error) => {
|
|
57
|
+
logger.error({ error, eventType: event.type }, "Hook trigger error");
|
|
58
|
+
});
|
|
59
|
+
}
|
|
60
|
+
/** 清除所有 Hooks */
|
|
61
|
+
export function clearHooks() {
|
|
62
|
+
hookRegistry.clear();
|
|
63
|
+
}
|
|
64
|
+
/** 获取已注册的 Hook 数量 */
|
|
65
|
+
export function getHookCount(eventType) {
|
|
66
|
+
if (eventType) {
|
|
67
|
+
return hookRegistry.get(eventType)?.length ?? 0;
|
|
68
|
+
}
|
|
69
|
+
let count = 0;
|
|
70
|
+
for (const handlers of hookRegistry.values()) {
|
|
71
|
+
count += handlers.length;
|
|
72
|
+
}
|
|
73
|
+
return count;
|
|
74
|
+
}
|
|
75
|
+
// ============== 便捷函数 ==============
|
|
76
|
+
/** 创建事件基础数据 */
|
|
77
|
+
function createEventBase(type, sessionKey) {
|
|
78
|
+
return {
|
|
79
|
+
type,
|
|
80
|
+
timestamp: Date.now(),
|
|
81
|
+
sessionKey,
|
|
82
|
+
};
|
|
83
|
+
}
|
|
84
|
+
/** 触发消息接收事件 */
|
|
85
|
+
export function emitMessageReceived(context) {
|
|
86
|
+
triggerHookSync({
|
|
87
|
+
...createEventBase("message_received"),
|
|
88
|
+
context,
|
|
89
|
+
});
|
|
90
|
+
}
|
|
91
|
+
/** 触发消息发送事件 */
|
|
92
|
+
export function emitMessageSending(params) {
|
|
93
|
+
triggerHookSync({
|
|
94
|
+
...createEventBase("message_sending", params.sessionKey),
|
|
95
|
+
channelId: params.channelId,
|
|
96
|
+
chatId: params.chatId,
|
|
97
|
+
content: params.content,
|
|
98
|
+
replyToId: params.replyToId,
|
|
99
|
+
});
|
|
100
|
+
}
|
|
101
|
+
/** 触发 Agent 开始事件 */
|
|
102
|
+
export function emitAgentStart(params) {
|
|
103
|
+
triggerHookSync({
|
|
104
|
+
...createEventBase("agent_start", params.sessionKey),
|
|
105
|
+
provider: params.provider,
|
|
106
|
+
model: params.model,
|
|
107
|
+
messages: params.messages,
|
|
108
|
+
});
|
|
109
|
+
}
|
|
110
|
+
/** 触发 Agent 结束事件 */
|
|
111
|
+
export function emitAgentEnd(params) {
|
|
112
|
+
triggerHookSync({
|
|
113
|
+
...createEventBase("agent_end", params.sessionKey),
|
|
114
|
+
provider: params.provider,
|
|
115
|
+
model: params.model,
|
|
116
|
+
response: params.response,
|
|
117
|
+
usage: params.usage,
|
|
118
|
+
durationMs: params.durationMs,
|
|
119
|
+
});
|
|
120
|
+
}
|
|
121
|
+
/** 触发工具事件 */
|
|
122
|
+
export function emitToolStart(params) {
|
|
123
|
+
triggerHookSync({
|
|
124
|
+
...createEventBase("tool_start", params.sessionKey),
|
|
125
|
+
toolName: params.toolName,
|
|
126
|
+
toolCallId: params.toolCallId,
|
|
127
|
+
arguments: params.arguments,
|
|
128
|
+
});
|
|
129
|
+
}
|
|
130
|
+
export function emitToolEnd(params) {
|
|
131
|
+
triggerHookSync({
|
|
132
|
+
...createEventBase("tool_end", params.sessionKey),
|
|
133
|
+
toolName: params.toolName,
|
|
134
|
+
toolCallId: params.toolCallId,
|
|
135
|
+
result: params.result,
|
|
136
|
+
isError: params.isError,
|
|
137
|
+
durationMs: params.durationMs,
|
|
138
|
+
});
|
|
139
|
+
}
|
|
140
|
+
/** 触发错误事件 */
|
|
141
|
+
export function emitError(error, context) {
|
|
142
|
+
triggerHookSync({
|
|
143
|
+
...createEventBase("error"),
|
|
144
|
+
error,
|
|
145
|
+
context,
|
|
146
|
+
});
|
|
147
|
+
}
|
|
148
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/hooks/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AAEH,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAGpD,MAAM,MAAM,GAAG,cAAc,CAAC,OAAO,CAAC,CAAC;AA0JvC,eAAe;AACf,MAAM,YAAY,GAAG,IAAI,GAAG,EAAqC,CAAC;AAElE,wCAAwC;AAExC,cAAc;AACd,MAAM,UAAU,YAAY,CAC1B,SAAY,EACZ,OAAoB;IAEpB,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;IACnD,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IACvB,YAAY,CAAC,GAAG,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IAEtC,MAAM,CAAC,KAAK,CAAC,EAAE,SAAS,EAAE,EAAE,iBAAiB,CAAC,CAAC;IAE/C,WAAW;IACX,OAAO,GAAG,EAAE;QACV,MAAM,eAAe,GAAG,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACpD,IAAI,eAAe,EAAE,CAAC;YACpB,MAAM,KAAK,GAAG,eAAe,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAC/C,IAAI,KAAK,IAAI,CAAC,EAAE,CAAC;gBACf,eAAe,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;YACnC,CAAC;QACH,CAAC;IACH,CAAC,CAAC;AACJ,CAAC;AAED,iBAAiB;AACjB,MAAM,UAAU,aAAa,CAC3B,KAAkD;IAElD,MAAM,aAAa,GAAsB,EAAE,CAAC;IAE5C,KAAK,MAAM,CAAC,SAAS,EAAE,OAAO,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACzD,IAAI,OAAO,EAAE,CAAC;YACZ,aAAa,CAAC,IAAI,CAAC,YAAY,CAAC,SAA0B,EAAE,OAAO,CAAC,CAAC,CAAC;QACxE,CAAC;IACH,CAAC;IAED,OAAO,GAAG,EAAE;QACV,KAAK,MAAM,WAAW,IAAI,aAAa,EAAE,CAAC;YACxC,WAAW,EAAE,CAAC;QAChB,CAAC;IACH,CAAC,CAAC;AACJ,CAAC;AAED,cAAc;AACd,MAAM,CAAC,KAAK,UAAU,WAAW,CAAC,KAAgB;IAChD,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC9C,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO;IAE/C,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,IAAI,CAAC;YACH,MAAM,OAAO,CAAC,KAAK,CAAC,CAAC;QACvB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,CAAC,IAAI,EAAE,EAAE,oBAAoB,CAAC,CAAC;QACvE,CAAC;IACH,CAAC;AACH,CAAC;AAED,sBAAsB;AACtB,MAAM,UAAU,eAAe,CAAC,KAAgB;IAC9C,WAAW,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,EAAE,EAAE;QACjC,MAAM,CAAC,KAAK,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,CAAC,IAAI,EAAE,EAAE,oBAAoB,CAAC,CAAC;IACvE,CAAC,CAAC,CAAC;AACL,CAAC;AAED,iBAAiB;AACjB,MAAM,UAAU,UAAU;IACxB,YAAY,CAAC,KAAK,EAAE,CAAC;AACvB,CAAC;AAED,qBAAqB;AACrB,MAAM,UAAU,YAAY,CAAC,SAAyB;IACpD,IAAI,SAAS,EAAE,CAAC;QACd,OAAO,YAAY,CAAC,GAAG,CAAC,SAAS,CAAC,EAAE,MAAM,IAAI,CAAC,CAAC;IAClD,CAAC;IACD,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,KAAK,MAAM,QAAQ,IAAI,YAAY,CAAC,MAAM,EAAE,EAAE,CAAC;QAC7C,KAAK,IAAI,QAAQ,CAAC,MAAM,CAAC;IAC3B,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,qCAAqC;AAErC,eAAe;AACf,SAAS,eAAe,CACtB,IAAO,EACP,UAAmB;IAEnB,OAAO;QACL,IAAI;QACJ,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;QACrB,UAAU;KACX,CAAC;AACJ,CAAC;AAED,eAAe;AACf,MAAM,UAAU,mBAAmB,CAAC,OAA8B;IAChE,eAAe,CAAC;QACd,GAAG,eAAe,CAAC,kBAAkB,CAAC;QACtC,OAAO;KACR,CAAC,CAAC;AACL,CAAC;AAED,eAAe;AACf,MAAM,UAAU,kBAAkB,CAAC,MAMlC;IACC,eAAe,CAAC;QACd,GAAG,eAAe,CAAC,iBAAiB,EAAE,MAAM,CAAC,UAAU,CAAC;QACxD,SAAS,EAAE,MAAM,CAAC,SAAS;QAC3B,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,OAAO,EAAE,MAAM,CAAC,OAAO;QACvB,SAAS,EAAE,MAAM,CAAC,SAAS;KAC5B,CAAC,CAAC;AACL,CAAC;AAED,oBAAoB;AACpB,MAAM,UAAU,cAAc,CAAC,MAK9B;IACC,eAAe,CAAC;QACd,GAAG,eAAe,CAAC,aAAa,EAAE,MAAM,CAAC,UAAU,CAAC;QACpD,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,QAAQ,EAAE,MAAM,CAAC,QAAQ;KAC1B,CAAC,CAAC;AACL,CAAC;AAED,oBAAoB;AACpB,MAAM,UAAU,YAAY,CAAC,MAO5B;IACC,eAAe,CAAC;QACd,GAAG,eAAe,CAAC,WAAW,EAAE,MAAM,CAAC,UAAU,CAAC;QAClD,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,KAAK,EAAE,MAAM,CAAC,KAAK;QACnB,UAAU,EAAE,MAAM,CAAC,UAAU;KAC9B,CAAC,CAAC;AACL,CAAC;AAED,aAAa;AACb,MAAM,UAAU,aAAa,CAAC,MAK7B;IACC,eAAe,CAAC;QACd,GAAG,eAAe,CAAC,YAAY,EAAE,MAAM,CAAC,UAAU,CAAC;QACnD,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,UAAU,EAAE,MAAM,CAAC,UAAU;QAC7B,SAAS,EAAE,MAAM,CAAC,SAAS;KAC5B,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,WAAW,CAAC,MAO3B;IACC,eAAe,CAAC;QACd,GAAG,eAAe,CAAC,UAAU,EAAE,MAAM,CAAC,UAAU,CAAC;QACjD,QAAQ,EAAE,MAAM,CAAC,QAAQ;QACzB,UAAU,EAAE,MAAM,CAAC,UAAU;QAC7B,MAAM,EAAE,MAAM,CAAC,MAAM;QACrB,OAAO,EAAE,MAAM,CAAC,OAAO;QACvB,UAAU,EAAE,MAAM,CAAC,UAAU;KAC9B,CAAC,CAAC;AACL,CAAC;AAED,aAAa;AACb,MAAM,UAAU,SAAS,CAAC,KAAY,EAAE,OAAgB;IACtD,eAAe,CAAC;QACd,GAAG,eAAe,CAAC,OAAO,CAAC;QAC3B,KAAK;QACL,OAAO;KACR,CAAC,CAAC;AACL,CAAC"}
|
package/dist/index.d.ts
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Mozi - 国产AI模型 + 国产通讯软件的智能助手
|
|
3
|
+
*
|
|
4
|
+
* 主入口文件
|
|
5
|
+
*/
|
|
6
|
+
export * from "./types/index.js";
|
|
7
|
+
export { loadConfig, validateRequiredConfig } from "./config/index.js";
|
|
8
|
+
export { BaseProvider, DeepSeekProvider, KimiProvider, StepfunProvider, MiniMaxProvider, createDeepSeekProvider, createKimiProvider, createStepfunProvider, createMiniMaxProvider, initializeProviders, getProvider, getAllProviders, findProviderForModel, getAllModels, } from "./providers/index.js";
|
|
9
|
+
export { BaseChannelAdapter, FeishuChannel, DingtalkChannel, createFeishuChannel, createDingtalkChannel, FeishuApiClient, DingtalkApiClient, registerChannel, getChannel, getAllChannels, setGlobalMessageHandler, } from "./channels/index.js";
|
|
10
|
+
export { Agent, createAgent, estimateTokens, estimateMessagesTokens, summarizeInStages, limitHistoryTurns, pruneHistoryForContextShare, FailoverError, runWithModelFallback, } from "./agents/index.js";
|
|
11
|
+
export { type Tool, type ToolCall, type ToolCallResult, type ToolResult, registerTool, registerTools, getTool, getAllTools, filterToolsByPolicy, executeToolCalls, toolsToOpenAIFunctions, createBuiltinTools, jsonResult, errorResult, textResult, } from "./tools/index.js";
|
|
12
|
+
export { type HookEventType, type HookEvent, type HookHandler, registerHook, registerHooks, triggerHook, triggerHookSync, clearHooks, getHookCount, emitMessageReceived, emitMessageSending, emitAgentStart, emitAgentEnd, emitToolStart, emitToolEnd, emitError, } from "./hooks/index.js";
|
|
13
|
+
export { type PluginMeta, type PluginApi, type PluginDefinition, registerPlugin, unregisterPlugin, getLoadedPlugins, isPluginLoaded, unregisterAllPlugins, definePlugin, defineToolPlugin, } from "./plugins/index.js";
|
|
14
|
+
export { type CommandContext, type CommandHandler, type CommandDefinition, registerCommand, registerCommands, getCommand, getAllCommands, isCommand, parseCommand, executeCommand, registerBuiltinCommands, } from "./commands/index.js";
|
|
15
|
+
export { Gateway, createGateway, startGateway } from "./gateway/index.js";
|
|
16
|
+
export { getLogger, createLogger, setLogger, getChildLogger } from "./utils/logger.js";
|
|
17
|
+
export { generateId, getEnvVar, requireEnvVar, delay, retry, truncate, safeJsonParse, deepMerge, computeHmacSha256, aesDecrypt, } from "./utils/index.js";
|
|
18
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,cAAc,kBAAkB,CAAC;AAGjC,OAAO,EAAE,UAAU,EAAE,sBAAsB,EAAE,MAAM,mBAAmB,CAAC;AAGvE,OAAO,EACL,YAAY,EACZ,gBAAgB,EAChB,YAAY,EACZ,eAAe,EACf,eAAe,EACf,sBAAsB,EACtB,kBAAkB,EAClB,qBAAqB,EACrB,qBAAqB,EACrB,mBAAmB,EACnB,WAAW,EACX,eAAe,EACf,oBAAoB,EACpB,YAAY,GACb,MAAM,sBAAsB,CAAC;AAG9B,OAAO,EACL,kBAAkB,EAClB,aAAa,EACb,eAAe,EACf,mBAAmB,EACnB,qBAAqB,EACrB,eAAe,EACf,iBAAiB,EACjB,eAAe,EACf,UAAU,EACV,cAAc,EACd,uBAAuB,GACxB,MAAM,qBAAqB,CAAC;AAG7B,OAAO,EACL,KAAK,EACL,WAAW,EACX,cAAc,EACd,sBAAsB,EACtB,iBAAiB,EACjB,iBAAiB,EACjB,2BAA2B,EAC3B,aAAa,EACb,oBAAoB,GACrB,MAAM,mBAAmB,CAAC;AAG3B,OAAO,EACL,KAAK,IAAI,EACT,KAAK,QAAQ,EACb,KAAK,cAAc,EACnB,KAAK,UAAU,EACf,YAAY,EACZ,aAAa,EACb,OAAO,EACP,WAAW,EACX,mBAAmB,EACnB,gBAAgB,EAChB,sBAAsB,EACtB,kBAAkB,EAClB,UAAU,EACV,WAAW,EACX,UAAU,GACX,MAAM,kBAAkB,CAAC;AAG1B,OAAO,EACL,KAAK,aAAa,EAClB,KAAK,SAAS,EACd,KAAK,WAAW,EAChB,YAAY,EACZ,aAAa,EACb,WAAW,EACX,eAAe,EACf,UAAU,EACV,YAAY,EACZ,mBAAmB,EACnB,kBAAkB,EAClB,cAAc,EACd,YAAY,EACZ,aAAa,EACb,WAAW,EACX,SAAS,GACV,MAAM,kBAAkB,CAAC;AAG1B,OAAO,EACL,KAAK,UAAU,EACf,KAAK,SAAS,EACd,KAAK,gBAAgB,EACrB,cAAc,EACd,gBAAgB,EAChB,gBAAgB,EAChB,cAAc,EACd,oBAAoB,EACpB,YAAY,EACZ,gBAAgB,GACjB,MAAM,oBAAoB,CAAC;AAG5B,OAAO,EACL,KAAK,cAAc,EACnB,KAAK,cAAc,EACnB,KAAK,iBAAiB,EACtB,eAAe,EACf,gBAAgB,EAChB,UAAU,EACV,cAAc,EACd,SAAS,EACT,YAAY,EACZ,cAAc,EACd,uBAAuB,GACxB,MAAM,qBAAqB,CAAC;AAG7B,OAAO,EAAE,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAG1E,OAAO,EAAE,SAAS,EAAE,YAAY,EAAE,SAAS,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AACvF,OAAO,EACL,UAAU,EACV,SAAS,EACT,aAAa,EACb,KAAK,EACL,KAAK,EACL,QAAQ,EACR,aAAa,EACb,SAAS,EACT,iBAAiB,EACjB,UAAU,GACX,MAAM,kBAAkB,CAAC"}
|