weflow-cli 1.5.0 → 1.6.1
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/ARCHITECTURE.md +105 -0
- package/CHANGELOG.md +121 -0
- package/CONTRIBUTING.md +45 -0
- package/LICENSE +21 -21
- package/OPERATIONS.md +272 -0
- package/README.en.md +281 -0
- package/README.md +334 -149
- package/SECURITY.md +47 -0
- package/bin/weflow-cli.ts +4050 -581
- package/dist/bin/weflow-cli.js +4310 -588
- package/dist/bin/weflow-cli.js.map +1 -1
- package/dist/mcp-server/index.js +178 -236
- package/dist/mcp-server/index.js.map +1 -1
- package/dist/src/core/dbPathService.d.ts +15 -1
- package/dist/src/core/dbPathService.d.ts.map +1 -1
- package/dist/src/core/dbPathService.js +177 -19
- package/dist/src/core/dbPathService.js.map +1 -1
- package/dist/src/core/keyService.d.ts +8 -0
- package/dist/src/core/keyService.d.ts.map +1 -1
- package/dist/src/core/keyService.js +127 -22
- package/dist/src/core/keyService.js.map +1 -1
- package/dist/src/core/ntCore.d.ts +59 -1
- package/dist/src/core/ntCore.d.ts.map +1 -1
- package/dist/src/core/ntCore.js +201 -65
- package/dist/src/core/ntCore.js.map +1 -1
- package/dist/src/core/sqlcipherCore.d.ts.map +1 -1
- package/dist/src/core/sqlcipherCore.js +18 -1
- package/dist/src/core/sqlcipherCore.js.map +1 -1
- package/dist/src/core/wcdbCore.d.ts.map +1 -1
- package/dist/src/core/wcdbCore.js +3 -6
- package/dist/src/core/wcdbCore.js.map +1 -1
- package/dist/src/services/assistantDaemon.d.ts +19 -0
- package/dist/src/services/assistantDaemon.d.ts.map +1 -0
- package/dist/src/services/assistantDaemon.js +117 -0
- package/dist/src/services/assistantDaemon.js.map +1 -0
- package/dist/src/services/assistantMemory.d.ts +36 -0
- package/dist/src/services/assistantMemory.d.ts.map +1 -0
- package/dist/src/services/assistantMemory.js +182 -0
- package/dist/src/services/assistantMemory.js.map +1 -0
- package/dist/src/services/assistantPrivacy.d.ts +25 -0
- package/dist/src/services/assistantPrivacy.d.ts.map +1 -0
- package/dist/src/services/assistantPrivacy.js +77 -0
- package/dist/src/services/assistantPrivacy.js.map +1 -0
- package/dist/src/services/assistantRouting.d.ts +18 -0
- package/dist/src/services/assistantRouting.d.ts.map +1 -0
- package/dist/src/services/assistantRouting.js +68 -0
- package/dist/src/services/assistantRouting.js.map +1 -0
- package/dist/src/services/assistantService.d.ts +37 -0
- package/dist/src/services/assistantService.d.ts.map +1 -0
- package/dist/src/services/assistantService.js +284 -0
- package/dist/src/services/assistantService.js.map +1 -0
- package/dist/src/services/assistantTools.d.ts +24 -0
- package/dist/src/services/assistantTools.d.ts.map +1 -0
- package/dist/src/services/assistantTools.js +582 -0
- package/dist/src/services/assistantTools.js.map +1 -0
- package/dist/src/services/chatService.d.ts +44 -0
- package/dist/src/services/chatService.d.ts.map +1 -1
- package/dist/src/services/chatService.js +105 -4
- package/dist/src/services/chatService.js.map +1 -1
- package/dist/src/services/configService.d.ts +55 -0
- package/dist/src/services/configService.d.ts.map +1 -1
- package/dist/src/services/configService.js +147 -6
- package/dist/src/services/configService.js.map +1 -1
- package/dist/src/services/evidenceService.d.ts +30 -0
- package/dist/src/services/evidenceService.d.ts.map +1 -0
- package/dist/src/services/evidenceService.js +98 -0
- package/dist/src/services/evidenceService.js.map +1 -0
- package/dist/src/services/exportService.d.ts +13 -20
- package/dist/src/services/exportService.d.ts.map +1 -1
- package/dist/src/services/exportService.js +113 -47
- package/dist/src/services/exportService.js.map +1 -1
- package/dist/src/services/initKeyService.d.ts +19 -0
- package/dist/src/services/initKeyService.d.ts.map +1 -0
- package/dist/src/services/initKeyService.js +80 -0
- package/dist/src/services/initKeyService.js.map +1 -0
- package/dist/src/services/messageContract.d.ts +23 -0
- package/dist/src/services/messageContract.d.ts.map +1 -0
- package/dist/src/services/messageContract.js +49 -0
- package/dist/src/services/messageContract.js.map +1 -0
- package/dist/src/services/messageQuery.d.ts +7 -0
- package/dist/src/services/messageQuery.d.ts.map +1 -0
- package/dist/src/services/messageQuery.js +30 -0
- package/dist/src/services/messageQuery.js.map +1 -0
- package/dist/src/services/wechat-formatter.js +14 -14
- package/dist/src/services/wechatMessageService.d.ts.map +1 -1
- package/dist/src/services/wechatMessageService.js +31 -3
- package/dist/src/services/wechatMessageService.js.map +1 -1
- package/dist/src/services/whitelistService.d.ts +52 -4
- package/dist/src/services/whitelistService.d.ts.map +1 -1
- package/dist/src/services/whitelistService.js +183 -17
- package/dist/src/services/whitelistService.js.map +1 -1
- package/dist/src/types.d.ts +20 -0
- package/dist/src/types.d.ts.map +1 -1
- package/dist/src/utils/dateRange.d.ts +15 -0
- package/dist/src/utils/dateRange.d.ts.map +1 -0
- package/dist/src/utils/dateRange.js +48 -0
- package/dist/src/utils/dateRange.js.map +1 -0
- package/dist/src/utils/mcpSecurity.d.ts +5 -0
- package/dist/src/utils/mcpSecurity.d.ts.map +1 -0
- package/dist/src/utils/mcpSecurity.js +39 -0
- package/dist/src/utils/mcpSecurity.js.map +1 -0
- package/dist/src/utils/packageRoot.d.ts +2 -0
- package/dist/src/utils/packageRoot.d.ts.map +1 -0
- package/dist/src/utils/packageRoot.js +16 -0
- package/dist/src/utils/packageRoot.js.map +1 -0
- package/dist/src/utils/python.d.ts +2 -0
- package/dist/src/utils/python.d.ts.map +1 -0
- package/dist/src/utils/python.js +72 -0
- package/dist/src/utils/python.js.map +1 -0
- package/dist/src/utils/pythonProcessEnv.d.ts +3 -0
- package/dist/src/utils/pythonProcessEnv.d.ts.map +1 -0
- package/dist/src/utils/pythonProcessEnv.js +46 -0
- package/dist/src/utils/pythonProcessEnv.js.map +1 -0
- package/dist/src/utils/talkerUtils.d.ts +4 -1
- package/dist/src/utils/talkerUtils.d.ts.map +1 -1
- package/dist/src/utils/talkerUtils.js +9 -9
- package/dist/src/utils/talkerUtils.js.map +1 -1
- package/dist/src/utils/wechatEmoji.d.ts +5 -0
- package/dist/src/utils/wechatEmoji.d.ts.map +1 -0
- package/dist/src/utils/wechatEmoji.js +35 -0
- package/dist/src/utils/wechatEmoji.js.map +1 -0
- package/docs/AI_INTERFACE.md +189 -0
- package/docs/BRANCHES.md +87 -0
- package/docs/DATA_CONTRACT.md +74 -0
- package/docs/DECISIONS.md +270 -0
- package/docs/DEEPSEEK_V4_FLASH/346/226/275/345/267/245/346/226/207/346/241/243.md +234 -0
- package/docs/EVIDENCE_GUIDE.md +96 -0
- package/docs/MCP.md +95 -0
- package/docs/PARTNERS.md +41 -0
- package/docs/PROJECT_STATE.md +96 -0
- package/docs/RELEASING.md +124 -0
- package/docs/ROADMAP.md +191 -0
- package/docs/SETUP.md +103 -0
- package/docs/images/weflow-architecture-gpt-image-2.png +0 -0
- package/docs/images/weflow-architecture.png +0 -0
- package/docs/images/weflow-architecture.svg +65 -0
- package/mcp-server/index.ts +169 -269
- package/package.json +81 -72
- package/requirements-3x.txt +4 -0
- package/requirements-voice.txt +15 -0
- package/requirements.txt +9 -0
- package/resources/js/marked.min.js +69 -0
- package/resources/wechat-emoji/666.png +0 -0
- package/resources/wechat-emoji/Awesome.png +0 -0
- package/resources/wechat-emoji/Concerned.png +0 -0
- package/resources/wechat-emoji/Cry.png +0 -0
- package/resources/wechat-emoji/Emm.png +0 -0
- package/resources/wechat-emoji/Facepalm.png +0 -0
- package/resources/wechat-emoji/Grin.png +0 -0
- package/resources/wechat-emoji/OK.png +0 -0
- package/resources/wechat-emoji/Respect.png +0 -0
- package/resources/wechat-emoji/Sick.png +0 -0
- package/resources/wechat-emoji/Sleep.png +0 -0
- package/resources/wechat-emoji/Smile.png +0 -0
- package/resources/wechat-emoji//344/272/262/344/272/262.png +0 -0
- package/resources/wechat-emoji//344/276/277/344/276/277.png +0 -0
- package/resources/wechat-emoji//345/201/267/347/254/221.png +0 -0
- package/resources/wechat-emoji//345/202/262/346/205/242.png +0 -0
- package/resources/wechat-emoji//345/206/215/350/247/201.png +0 -0
- package/resources/wechat-emoji//345/207/213/350/260/242.png +0 -0
- package/resources/wechat-emoji//345/212/240/346/262/271.png +0 -0
- package/resources/wechat-emoji//345/213/276/345/274/225.png +0 -0
- package/resources/wechat-emoji//345/217/221/345/221/206.png +0 -0
- package/resources/wechat-emoji//345/217/221/346/200/222.png +0 -0
- package/resources/wechat-emoji//345/217/221/346/212/226.png +0 -0
- package/resources/wechat-emoji//345/217/257/346/200/234.png +0 -0
- package/resources/wechat-emoji//345/217/263/345/223/274/345/223/274.png +0 -0
- package/resources/wechat-emoji//345/217/271/346/260/224.png +0 -0
- package/resources/wechat-emoji//345/220/203/347/223/234.png +0 -0
- package/resources/wechat-emoji//345/220/210/345/215/201.png +0 -0
- package/resources/wechat-emoji//345/220/220.png +0 -0
- package/resources/wechat-emoji//345/221/262/347/211/231.png +0 -0
- package/resources/wechat-emoji//345/222/222/351/252/202.png +0 -0
- package/resources/wechat-emoji//345/222/226/345/225/241.png +0 -0
- package/resources/wechat-emoji//345/223/207.png +0 -0
- package/resources/wechat-emoji//345/225/244/351/205/222.png +0 -0
- package/resources/wechat-emoji//345/230/230.png +0 -0
- package/resources/wechat-emoji//345/230/264/345/224/207.png +0 -0
- package/resources/wechat-emoji//345/230/277/345/223/210.png +0 -0
- package/resources/wechat-emoji//345/233/247.png +0 -0
- package/resources/wechat-emoji//345/233/260.png +0 -0
- package/resources/wechat-emoji//345/235/217/347/254/221.png +0 -0
- package/resources/wechat-emoji//345/244/247/345/223/255.png +0 -0
- package/resources/wechat-emoji//345/244/251/345/225/212.png +0 -0
- package/resources/wechat-emoji//345/244/252/351/230/263.png +0 -0
- package/resources/wechat-emoji//345/244/261/346/234/233.png +0 -0
- package/resources/wechat-emoji//345/245/270/347/254/221.png +0 -0
- package/resources/wechat-emoji//345/245/275/347/232/204.png +0 -0
- package/resources/wechat-emoji//345/247/224/345/261/210.png +0 -0
- package/resources/wechat-emoji//345/256/263/347/276/236.png +0 -0
- package/resources/wechat-emoji//345/260/264/345/260/254.png +0 -0
- package/resources/wechat-emoji//345/272/206/347/245/235.png +0 -0
- package/resources/wechat-emoji//345/274/261.png +0 -0
- package/resources/wechat-emoji//345/274/272.png +0 -0
- package/resources/wechat-emoji//345/276/227/346/204/217.png +0 -0
- package/resources/wechat-emoji//345/276/256/347/254/221.png +0 -0
- package/resources/wechat-emoji//345/277/203/347/242/216.png +0 -0
- package/resources/wechat-emoji//345/277/253/345/223/255/344/272/206.png +0 -0
- package/resources/wechat-emoji//346/201/220/346/203/247.png +0 -0
- package/resources/wechat-emoji//346/202/240/351/227/262.png +0 -0
- package/resources/wechat-emoji//346/203/212/346/201/220.png +0 -0
- package/resources/wechat-emoji//346/203/212/350/256/266.png +0 -0
- package/resources/wechat-emoji//346/204/211/345/277/253.png +0 -0
- package/resources/wechat-emoji//346/206/250/347/254/221.png +0 -0
- package/resources/wechat-emoji//346/211/223/350/204/270.png +0 -0
- package/resources/wechat-emoji//346/212/223/347/213/202.png +0 -0
- package/resources/wechat-emoji//346/212/240/351/274/273.png +0 -0
- package/resources/wechat-emoji//346/212/261/346/213/263.png +0 -0
- package/resources/wechat-emoji//346/213/245/346/212/261.png +0 -0
- package/resources/wechat-emoji//346/213/263/345/244/264.png +0 -0
- package/resources/wechat-emoji//346/215/202/350/204/270.png +0 -0
- package/resources/wechat-emoji//346/217/241/346/211/213.png +0 -0
- package/resources/wechat-emoji//346/222/207/345/230/264.png +0 -0
- package/resources/wechat-emoji//346/223/246/346/261/227.png +0 -0
- package/resources/wechat-emoji//346/225/262/346/211/223.png +0 -0
- package/resources/wechat-emoji//346/227/240/350/257/255.png +0 -0
- package/resources/wechat-emoji//346/227/272/346/237/264.png +0 -0
- package/resources/wechat-emoji//346/231/225.png +0 -0
- package/resources/wechat-emoji//346/234/210/344/272/256.png +0 -0
- package/resources/wechat-emoji//346/234/272/346/231/272.png +0 -0
- package/resources/wechat-emoji//346/261/227.png +0 -0
- package/resources/wechat-emoji//346/265/201/346/263/252.png +0 -0
- package/resources/wechat-emoji//347/202/270/345/274/271.png +0 -0
- package/resources/wechat-emoji//347/203/237/350/212/261.png +0 -0
- package/resources/wechat-emoji//347/210/206/347/253/271.png +0 -0
- package/resources/wechat-emoji//347/210/261/345/277/203.png +0 -0
- package/resources/wechat-emoji//347/214/252/345/244/264.png +0 -0
- package/resources/wechat-emoji//347/216/253/347/221/260.png +0 -0
- package/resources/wechat-emoji//347/224/237/347/227/205.png +0 -0
- package/resources/wechat-emoji//347/226/221/351/227/256.png +0 -0
- package/resources/wechat-emoji//347/231/274.png +0 -0
- package/resources/wechat-emoji//347/231/275/347/234/274.png +0 -0
- package/resources/wechat-emoji//347/232/261/347/234/211.png +0 -0
- package/resources/wechat-emoji//347/235/241.png +0 -0
- package/resources/wechat-emoji//347/240/264/346/266/225/344/270/272/347/254/221.png +0 -0
- package/resources/wechat-emoji//347/244/274/347/211/251.png +0 -0
- package/resources/wechat-emoji//347/244/276/344/274/232/347/244/276/344/274/232.png +0 -0
- package/resources/wechat-emoji//347/246/217.png +0 -0
- package/resources/wechat-emoji//347/254/221/350/204/270.png +0 -0
- package/resources/wechat-emoji//347/272/242/345/214/205.png +0 -0
- package/resources/wechat-emoji//347/277/273/347/231/275/347/234/274.png +0 -0
- package/resources/wechat-emoji//350/200/266.png +0 -0
- package/resources/wechat-emoji//350/203/234/345/210/251.png +0 -0
- package/resources/wechat-emoji//350/204/270/347/272/242.png +0 -0
- package/resources/wechat-emoji//350/211/262.png +0 -0
- package/resources/wechat-emoji//350/213/246/346/266/251.png +0 -0
- package/resources/wechat-emoji//350/217/234/345/210/200.png +0 -0
- package/resources/wechat-emoji//350/233/213/347/263/225.png +0 -0
- package/resources/wechat-emoji//350/241/260.png +0 -0
- package/resources/wechat-emoji//350/243/202/345/274/200.png +0 -0
- package/resources/wechat-emoji//350/256/251/346/210/221/347/234/213/347/234/213.png +0 -0
- package/resources/wechat-emoji//350/260/203/347/232/256.png +0 -0
- package/resources/wechat-emoji//350/267/263/350/267/263.png +0 -0
- package/resources/wechat-emoji//350/275/254/345/234/210.png +0 -0
- package/resources/wechat-emoji//351/204/231/350/247/206.png +0 -0
- package/resources/wechat-emoji//351/227/255/345/230/264.png +0 -0
- package/resources/wechat-emoji//351/230/264/351/231/251.png +0 -0
- package/resources/wechat-emoji//351/232/276/350/277/207.png +0 -0
- package/resources/wechat-emoji//351/252/267/351/253/205.png +0 -0
- package/resources/wechat-emoji//351/274/223/346/216/214.png +0 -0
- package/scripts/_batch_link_sources.py +73 -73
- package/scripts/_utils.py +29 -0
- package/scripts/annual_report.py +796 -796
- package/scripts/auto_tag.py +128 -128
- package/scripts/biz_daily.py +210 -53
- package/scripts/chat_report.py +10 -0
- package/scripts/chat_stats.py +702 -702
- package/scripts/classify_daily.py +17 -11
- package/scripts/create_reading_notes.py +370 -370
- package/scripts/daily_stats.py +132 -0
- package/scripts/enrich_backlinks.py +163 -163
- package/scripts/export_chat_html.py +1704 -107
- package/scripts/extract_todos.py +445 -404
- package/scripts/fav_server.py +119 -12
- package/scripts/fix_topics.py +111 -111
- package/scripts/generate_ai_report.py +1 -1
- package/scripts/generate_html.py +597 -277
- package/scripts/inspection_records.py +354 -0
- package/scripts/mcp_bridge.py +398 -398
- package/scripts/nt_decrypt.py +702 -48
- package/scripts/pipeline.py +27 -19
- package/scripts/promote_all.py +312 -310
- package/scripts/promote_ideas.py +299 -297
- package/scripts/rag_chat.py +194 -191
- package/scripts/semantic_search.py +435 -431
- package/scripts/sync_fav.py +201 -146
- package/scripts/sync_weread.py +192 -192
- package/scripts/vault_rag.py +140 -139
- package/scripts/vault_search.py +141 -141
- package/scripts/wechat_emoji.py +132 -0
- package/scripts/wechat_emoticon.py +433 -0
- package/scripts/wechat_image.py +258 -0
- package/scripts/wechat_voice.py +327 -0
- package/src/core/dbPathService.ts +163 -18
- package/src/core/keyService.ts +137 -25
- package/src/core/ntCore.ts +215 -62
- package/src/core/sqlcipherCore.ts +17 -1
- package/src/core/wcdbCore.ts +3 -7
- package/src/services/assistantDaemon.ts +111 -0
- package/src/services/assistantMemory.ts +193 -0
- package/src/services/assistantPrivacy.ts +87 -0
- package/src/services/assistantRouting.ts +92 -0
- package/src/services/assistantService.ts +295 -0
- package/src/services/assistantTools.ts +549 -0
- package/src/services/chatService.ts +133 -4
- package/src/services/configService.ts +194 -6
- package/src/services/evidenceService.ts +129 -0
- package/src/services/exportService.ts +113 -48
- package/src/services/initKeyService.ts +83 -0
- package/src/services/messageContract.ts +77 -0
- package/src/services/messageQuery.ts +38 -0
- package/src/services/wechat-formatter.ts +348 -348
- package/src/services/wechatMessageService.ts +25 -3
- package/src/services/wereadService.ts +308 -308
- package/src/services/whitelistService.ts +209 -22
- package/src/types.ts +20 -0
- package/src/utils/dateRange.ts +58 -0
- package/src/utils/mcpSecurity.ts +39 -0
- package/src/utils/packageRoot.ts +14 -0
- package/src/utils/python.ts +73 -0
- package/src/utils/pythonProcessEnv.ts +45 -0
- package/src/utils/talkerUtils.ts +14 -2
- package/dist/src/utils/errors.d.ts +0 -25
- package/dist/src/utils/errors.d.ts.map +0 -1
- package/dist/src/utils/errors.js +0 -43
- package/dist/src/utils/errors.js.map +0 -1
- package/dist/src/utils/pythonRunner.d.ts +0 -27
- package/dist/src/utils/pythonRunner.d.ts.map +0 -1
- package/dist/src/utils/pythonRunner.js +0 -62
- package/dist/src/utils/pythonRunner.js.map +0 -1
- package/src/utils/errors.ts +0 -42
- package/src/utils/pythonRunner.ts +0 -81
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
export declare function isDaemonAlive(): {
|
|
2
|
+
alive: boolean;
|
|
3
|
+
pid: number | null;
|
|
4
|
+
};
|
|
5
|
+
export declare function appendLog(line: string): void;
|
|
6
|
+
/** 日志轮转: 超过 1MB 保留尾部 256KB, 防止常驻进程日志无限增长 */
|
|
7
|
+
export declare function rotateLogIfNeeded(): void;
|
|
8
|
+
export declare function tailLog(n?: number): string;
|
|
9
|
+
/** 后台启动守护进程 (detached, 脱离终端生命周期) */
|
|
10
|
+
export declare function startDaemon(): {
|
|
11
|
+
started: boolean;
|
|
12
|
+
pid?: number;
|
|
13
|
+
error?: string;
|
|
14
|
+
};
|
|
15
|
+
export declare function stopDaemon(): {
|
|
16
|
+
stopped: boolean;
|
|
17
|
+
message: string;
|
|
18
|
+
};
|
|
19
|
+
//# sourceMappingURL=assistantDaemon.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"assistantDaemon.d.ts","sourceRoot":"","sources":["../../../src/services/assistantDaemon.ts"],"names":[],"mappings":"AAgBA,wBAAgB,aAAa,IAAI;IAAE,KAAK,EAAE,OAAO,CAAC;IAAC,GAAG,EAAE,MAAM,GAAG,IAAI,CAAA;CAAE,CAWtE;AAED,wBAAgB,SAAS,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI,CAK5C;AAED,4CAA4C;AAC5C,wBAAgB,iBAAiB,IAAI,IAAI,CAUxC;AAED,wBAAgB,OAAO,CAAC,CAAC,SAAK,GAAG,MAAM,CAOtC;AAED,oCAAoC;AACpC,wBAAgB,WAAW,IAAI;IAAE,OAAO,EAAE,OAAO,CAAC;IAAC,GAAG,CAAC,EAAE,MAAM,CAAC;IAAC,KAAK,CAAC,EAAE,MAAM,CAAA;CAAE,CAqChF;AAED,wBAAgB,UAAU,IAAI;IAAE,OAAO,EAAE,OAAO,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,CAYlE"}
|
|
@@ -0,0 +1,117 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 助手守护进程管理 — start(后台常驻) / stop / status / run(前台)
|
|
3
|
+
* pid 文件: ~/.weflow-cli/assistant.pid, 日志: ~/.weflow-cli/assistant.log
|
|
4
|
+
*/
|
|
5
|
+
import { join } from 'path';
|
|
6
|
+
import { existsSync, readFileSync, writeFileSync, rmSync, mkdirSync, appendFileSync, statSync } from 'fs';
|
|
7
|
+
import os from 'os';
|
|
8
|
+
import { spawn } from 'child_process';
|
|
9
|
+
import { resolvePackageRoot } from '../utils/packageRoot.js';
|
|
10
|
+
import { createPythonProcessEnv } from '../utils/pythonProcessEnv.js';
|
|
11
|
+
const DIR = join(os.homedir(), '.weflow-cli');
|
|
12
|
+
const PID_FILE = join(DIR, 'assistant.pid');
|
|
13
|
+
const LOG_FILE = join(DIR, 'assistant.log');
|
|
14
|
+
const MARKER = 'WEFLOW_ASSISTANT_DAEMON=1';
|
|
15
|
+
export function isDaemonAlive() {
|
|
16
|
+
if (!existsSync(PID_FILE))
|
|
17
|
+
return { alive: false, pid: null };
|
|
18
|
+
try {
|
|
19
|
+
const pid = parseInt(readFileSync(PID_FILE, 'utf8').trim());
|
|
20
|
+
// Windows: process.kill(pid, 0) 对不存在进程抛错; 对存在进程返回 undefined
|
|
21
|
+
process.kill(pid, 0);
|
|
22
|
+
return { alive: true, pid };
|
|
23
|
+
}
|
|
24
|
+
catch {
|
|
25
|
+
rmSync(PID_FILE, { force: true });
|
|
26
|
+
return { alive: false, pid: null };
|
|
27
|
+
}
|
|
28
|
+
}
|
|
29
|
+
export function appendLog(line) {
|
|
30
|
+
try {
|
|
31
|
+
if (!existsSync(DIR))
|
|
32
|
+
mkdirSync(DIR, { recursive: true });
|
|
33
|
+
appendFileSync(LOG_FILE, line + '\n', 'utf8');
|
|
34
|
+
}
|
|
35
|
+
catch { /* 日志失败不阻断 */ }
|
|
36
|
+
}
|
|
37
|
+
/** 日志轮转: 超过 1MB 保留尾部 256KB, 防止常驻进程日志无限增长 */
|
|
38
|
+
export function rotateLogIfNeeded() {
|
|
39
|
+
try {
|
|
40
|
+
if (!existsSync(LOG_FILE))
|
|
41
|
+
return;
|
|
42
|
+
if (statSync(LOG_FILE).size < 1024 * 1024)
|
|
43
|
+
return;
|
|
44
|
+
const buf = readFileSync(LOG_FILE);
|
|
45
|
+
const tail = buf.subarray(Math.max(0, buf.length - 256 * 1024));
|
|
46
|
+
const cut = tail.indexOf('\n'); // 对齐到行首
|
|
47
|
+
writeFileSync(LOG_FILE, `--- rotated at ${new Date().toLocaleString('zh-CN')} ---\n` + tail.subarray(cut + 1));
|
|
48
|
+
appendLog(`[轮转] 日志已截断 (原 ${statSync(LOG_FILE).size} 字节)`);
|
|
49
|
+
}
|
|
50
|
+
catch { /* 轮转失败不阻断 */ }
|
|
51
|
+
}
|
|
52
|
+
export function tailLog(n = 20) {
|
|
53
|
+
try {
|
|
54
|
+
const lines = readFileSync(LOG_FILE, 'utf8').trim().split('\n');
|
|
55
|
+
return lines.slice(-n).join('\n');
|
|
56
|
+
}
|
|
57
|
+
catch {
|
|
58
|
+
return '(暂无日志)';
|
|
59
|
+
}
|
|
60
|
+
}
|
|
61
|
+
/** 后台启动守护进程 (detached, 脱离终端生命周期) */
|
|
62
|
+
export function startDaemon() {
|
|
63
|
+
const { alive, pid } = isDaemonAlive();
|
|
64
|
+
if (alive)
|
|
65
|
+
return { started: false, pid: pid, error: `已在运行 (pid ${pid})` };
|
|
66
|
+
const packageRoot = resolvePackageRoot(import.meta.url);
|
|
67
|
+
const compiledEntry = join(packageRoot, 'dist', 'bin', 'weflow-cli.js');
|
|
68
|
+
const sourceEntry = join(packageRoot, 'bin', 'weflow-cli.ts');
|
|
69
|
+
const entry = existsSync(compiledEntry) ? compiledEntry : sourceEntry;
|
|
70
|
+
if (!existsSync(entry))
|
|
71
|
+
return { started: false, error: `入口不存在: ${entry}` };
|
|
72
|
+
try {
|
|
73
|
+
appendLog(`--- ${new Date().toLocaleString('zh-CN')} daemon starting ---`);
|
|
74
|
+
const entryArgs = entry === sourceEntry ? ['--import', 'tsx', entry] : [entry];
|
|
75
|
+
const child = spawn(process.execPath, [...entryArgs, 'assistant', 'run'], {
|
|
76
|
+
detached: true,
|
|
77
|
+
stdio: ['ignore', 'pipe', 'pipe'],
|
|
78
|
+
env: createPythonProcessEnv({ [MARKER]: '1' }),
|
|
79
|
+
windowsHide: true,
|
|
80
|
+
});
|
|
81
|
+
// 收集子进程输出写入日志
|
|
82
|
+
child.stdout?.on('data', (d) => appendLog(d.toString().trim()));
|
|
83
|
+
child.stderr?.on('data', (d) => appendLog('[stderr] ' + d.toString().trim()));
|
|
84
|
+
child.on('exit', (code) => {
|
|
85
|
+
appendLog(`--- daemon exited (code ${code}) ---`);
|
|
86
|
+
try {
|
|
87
|
+
if (existsSync(PID_FILE))
|
|
88
|
+
rmSync(PID_FILE, { force: true });
|
|
89
|
+
}
|
|
90
|
+
catch { }
|
|
91
|
+
});
|
|
92
|
+
if (child.pid) {
|
|
93
|
+
writeFileSync(PID_FILE, String(child.pid), 'utf8');
|
|
94
|
+
return { started: true, pid: child.pid };
|
|
95
|
+
}
|
|
96
|
+
return { started: false, error: '无法获取子进程 pid' };
|
|
97
|
+
}
|
|
98
|
+
catch (e) {
|
|
99
|
+
return { started: false, error: e.message };
|
|
100
|
+
}
|
|
101
|
+
}
|
|
102
|
+
export function stopDaemon() {
|
|
103
|
+
const { alive, pid } = isDaemonAlive();
|
|
104
|
+
if (!alive)
|
|
105
|
+
return { stopped: false, message: '守护进程未在运行' };
|
|
106
|
+
try {
|
|
107
|
+
// Windows 下 SIGTERM/SIGKILL 等价 terminate
|
|
108
|
+
process.kill(pid, 'SIGTERM');
|
|
109
|
+
rmSync(PID_FILE, { force: true });
|
|
110
|
+
appendLog(`--- ${new Date().toLocaleString('zh-CN')} daemon stopped manually ---`);
|
|
111
|
+
return { stopped: true, message: `已停止 (pid ${pid})` };
|
|
112
|
+
}
|
|
113
|
+
catch (e) {
|
|
114
|
+
return { stopped: false, message: `停止失败: ${e.message}` };
|
|
115
|
+
}
|
|
116
|
+
}
|
|
117
|
+
//# sourceMappingURL=assistantDaemon.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"assistantDaemon.js","sourceRoot":"","sources":["../../../src/services/assistantDaemon.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAA;AAC3B,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,EAAE,SAAS,EAAE,cAAc,EAAE,QAAQ,EAAE,MAAM,IAAI,CAAA;AACzG,OAAO,EAAE,MAAM,IAAI,CAAA;AACnB,OAAO,EAAE,KAAK,EAAqB,MAAM,eAAe,CAAA;AACxD,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAA;AAC5D,OAAO,EAAE,sBAAsB,EAAE,MAAM,8BAA8B,CAAA;AAErE,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,aAAa,CAAC,CAAA;AAC7C,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,eAAe,CAAC,CAAA;AAC3C,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,eAAe,CAAC,CAAA;AAC3C,MAAM,MAAM,GAAG,2BAA2B,CAAA;AAE1C,MAAM,UAAU,aAAa;IAC3B,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC;QAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,CAAA;IAC7D,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,QAAQ,CAAC,YAAY,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC,CAAA;QAC3D,4DAA4D;QAC5D,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,CAAC,CAAA;QACpB,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,CAAA;IAC7B,CAAC;IAAC,MAAM,CAAC;QACP,MAAM,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAA;QACjC,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,IAAI,EAAE,CAAA;IACpC,CAAC;AACH,CAAC;AAED,MAAM,UAAU,SAAS,CAAC,IAAY;IACpC,IAAI,CAAC;QACH,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;YAAE,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAA;QACzD,cAAc,CAAC,QAAQ,EAAE,IAAI,GAAG,IAAI,EAAE,MAAM,CAAC,CAAA;IAC/C,CAAC;IAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC;AAC3B,CAAC;AAED,4CAA4C;AAC5C,MAAM,UAAU,iBAAiB;IAC/B,IAAI,CAAC;QACH,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC;YAAE,OAAM;QACjC,IAAI,QAAQ,CAAC,QAAQ,CAAC,CAAC,IAAI,GAAG,IAAI,GAAG,IAAI;YAAE,OAAM;QACjD,MAAM,GAAG,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAA;QAClC,MAAM,IAAI,GAAG,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,CAAC,CAAA;QAC/D,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAA,CAAC,QAAQ;QACvC,aAAa,CAAC,QAAQ,EAAE,kBAAkB,IAAI,IAAI,EAAE,CAAC,cAAc,CAAC,OAAO,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,GAAG,CAAC,CAAC,CAAC,CAAA;QAC9G,SAAS,CAAC,iBAAiB,QAAQ,CAAC,QAAQ,CAAC,CAAC,IAAI,MAAM,CAAC,CAAA;IAC3D,CAAC;IAAC,MAAM,CAAC,CAAC,aAAa,CAAC,CAAC;AAC3B,CAAC;AAED,MAAM,UAAU,OAAO,CAAC,CAAC,GAAG,EAAE;IAC5B,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,YAAY,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAA;QAC/D,OAAO,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IACnC,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,QAAQ,CAAA;IACjB,CAAC;AACH,CAAC;AAED,oCAAoC;AACpC,MAAM,UAAU,WAAW;IACzB,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,aAAa,EAAE,CAAA;IACtC,IAAI,KAAK;QAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,GAAI,EAAE,KAAK,EAAE,aAAa,GAAG,GAAG,EAAE,CAAA;IAE3E,MAAM,WAAW,GAAG,kBAAkB,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAA;IACvD,MAAM,aAAa,GAAG,IAAI,CAAC,WAAW,EAAE,MAAM,EAAE,KAAK,EAAE,eAAe,CAAC,CAAA;IACvE,MAAM,WAAW,GAAG,IAAI,CAAC,WAAW,EAAE,KAAK,EAAE,eAAe,CAAC,CAAA;IAC7D,MAAM,KAAK,GAAG,UAAU,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,WAAW,CAAA;IAErE,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC;QAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,KAAK,EAAE,EAAE,CAAA;IAE3E,IAAI,CAAC;QACH,SAAS,CAAC,OAAO,IAAI,IAAI,EAAE,CAAC,cAAc,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAAA;QAC1E,MAAM,SAAS,GAAG,KAAK,KAAK,WAAW,CAAC,CAAC,CAAC,CAAC,UAAU,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAA;QAC9E,MAAM,KAAK,GAAiB,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,GAAG,SAAS,EAAE,WAAW,EAAE,KAAK,CAAC,EAAE;YACtF,QAAQ,EAAE,IAAI;YACd,KAAK,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,CAAC;YACjC,GAAG,EAAE,sBAAsB,CAAC,EAAE,CAAC,MAAM,CAAC,EAAE,GAAG,EAAE,CAAC;YAC9C,WAAW,EAAE,IAAI;SAClB,CAAC,CAAA;QAEF,cAAc;QACd,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,CAAS,EAAE,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC,CAAA;QACvE,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,CAAS,EAAE,EAAE,CAAC,SAAS,CAAC,WAAW,GAAG,CAAC,CAAC,QAAQ,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC,CAAA;QACrF,KAAK,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE;YACxB,SAAS,CAAC,2BAA2B,IAAI,OAAO,CAAC,CAAA;YACjD,IAAI,CAAC;gBAAC,IAAI,UAAU,CAAC,QAAQ,CAAC;oBAAE,MAAM,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAA;YAAC,CAAC;YAAC,MAAM,CAAC,CAAA,CAAC;QAC9E,CAAC,CAAC,CAAA;QAEF,IAAI,KAAK,CAAC,GAAG,EAAE,CAAC;YACd,aAAa,CAAC,QAAQ,EAAE,MAAM,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,MAAM,CAAC,CAAA;YAClD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,CAAC,GAAG,EAAE,CAAA;QAC1C,CAAC;QACD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,aAAa,EAAE,CAAA;IACjD,CAAC;IAAC,OAAO,CAAM,EAAE,CAAC;QAChB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,OAAO,EAAE,CAAA;IAC7C,CAAC;AACH,CAAC;AAED,MAAM,UAAU,UAAU;IACxB,MAAM,EAAE,KAAK,EAAE,GAAG,EAAE,GAAG,aAAa,EAAE,CAAA;IACtC,IAAI,CAAC,KAAK;QAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,CAAA;IAC1D,IAAI,CAAC;QACH,yCAAyC;QACzC,OAAO,CAAC,IAAI,CAAC,GAAI,EAAE,SAAS,CAAC,CAAA;QAC7B,MAAM,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAA;QACjC,SAAS,CAAC,OAAO,IAAI,IAAI,EAAE,CAAC,cAAc,CAAC,OAAO,CAAC,8BAA8B,CAAC,CAAA;QAClF,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,YAAY,GAAG,GAAG,EAAE,CAAA;IACvD,CAAC;IAAC,OAAO,CAAM,EAAE,CAAC;QAChB,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,OAAO,EAAE,SAAS,CAAC,CAAC,OAAO,EAAE,EAAE,CAAA;IAC1D,CAAC;AACH,CAAC"}
|
|
@@ -0,0 +1,36 @@
|
|
|
1
|
+
export interface ChatTurn {
|
|
2
|
+
role: 'user' | 'assistant';
|
|
3
|
+
content: string;
|
|
4
|
+
}
|
|
5
|
+
export interface Fact {
|
|
6
|
+
content: string;
|
|
7
|
+
ts: number;
|
|
8
|
+
}
|
|
9
|
+
/** LLM 调用器抽象 (由 assistantService 注入, 避免循环依赖) */
|
|
10
|
+
export type LlmCaller = (messages: ChatTurn[], maxTokens?: number) => Promise<string>;
|
|
11
|
+
export declare class AssistantMemory {
|
|
12
|
+
private states;
|
|
13
|
+
/** 本进程修改过的用户 (save 时只回写这些用户, 其余保留磁盘最新值) */
|
|
14
|
+
private dirtyUsers;
|
|
15
|
+
constructor();
|
|
16
|
+
private normalize;
|
|
17
|
+
private load;
|
|
18
|
+
save(): void;
|
|
19
|
+
private state;
|
|
20
|
+
reset(userId: string): void;
|
|
21
|
+
addTurn(userId: string, role: 'user' | 'assistant', content: string): void;
|
|
22
|
+
/** L1 窗口 (喂给 LLM 的近期原文) */
|
|
23
|
+
workingWindow(userId: string): ChatTurn[];
|
|
24
|
+
summary(userId: string): string;
|
|
25
|
+
facts(userId: string): Fact[];
|
|
26
|
+
userCount(): number;
|
|
27
|
+
/** L2: 窗口溢出时把最旧一半压进摘要 */
|
|
28
|
+
compressIfNeeded(userId: string, llm: LlmCaller): Promise<boolean>;
|
|
29
|
+
/** L3: 定期从最近对话提取持久事实 */
|
|
30
|
+
extractFactsIfNeeded(userId: string, llm: LlmCaller): Promise<number>;
|
|
31
|
+
/** L3 检索: 关键词命中 (为未来向量检索预留同一接口) */
|
|
32
|
+
searchFacts(userId: string, keyword: string): Fact[];
|
|
33
|
+
/** L3 写入 (save_memory 工具用, 立即生效) */
|
|
34
|
+
addFact(userId: string, content: string): boolean;
|
|
35
|
+
}
|
|
36
|
+
//# sourceMappingURL=assistantMemory.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"assistantMemory.d.ts","sourceRoot":"","sources":["../../../src/services/assistantMemory.ts"],"names":[],"mappings":"AAkBA,MAAM,WAAW,QAAQ;IAAG,IAAI,EAAE,MAAM,GAAG,WAAW,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE;AACzE,MAAM,WAAW,IAAI;IAAG,OAAO,EAAE,MAAM,CAAC;IAAC,EAAE,EAAE,MAAM,CAAA;CAAE;AASrD,gDAAgD;AAChD,MAAM,MAAM,SAAS,GAAG,CAAC,QAAQ,EAAE,QAAQ,EAAE,EAAE,SAAS,CAAC,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,CAAC,CAAA;AAErF,qBAAa,eAAe;IAC1B,OAAO,CAAC,MAAM,CAAiC;IAC/C,2CAA2C;IAC3C,OAAO,CAAC,UAAU,CAAoB;;IAItC,OAAO,CAAC,SAAS;IASjB,OAAO,CAAC,IAAI;IAWZ,IAAI,IAAI,IAAI;IA2BZ,OAAO,CAAC,KAAK;IAOb,KAAK,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI;IAM3B,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,WAAW,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI;IAO1E,2BAA2B;IAC3B,aAAa,CAAC,MAAM,EAAE,MAAM,GAAG,QAAQ,EAAE;IAIzC,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM;IAC/B,KAAK,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,EAAE;IAC7B,SAAS,IAAI,MAAM;IAEnB,yBAAyB;IACnB,gBAAgB,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC;IA0BxE,wBAAwB;IAClB,oBAAoB,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,SAAS,GAAG,OAAO,CAAC,MAAM,CAAC;IAgC3E,mCAAmC;IACnC,WAAW,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,IAAI,EAAE;IAOpD,oCAAoC;IACpC,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO;CASlD"}
|
|
@@ -0,0 +1,182 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 三层记忆 (参考 LangGraph checkpointer + Mem0 范式):
|
|
3
|
+
* L1 工作记忆 最近 N 轮原文 (滑入滑出)
|
|
4
|
+
* L2 滚动摘要 滑出的旧轮次压缩成摘要, 跨窗口保留脉络
|
|
5
|
+
* L3 长期事实 从对话中提取的持久信息 (偏好/项目/关系), 注入系统提示
|
|
6
|
+
* 持久化: ~/.weflow-cli/assistant_memory.json
|
|
7
|
+
*/
|
|
8
|
+
import { join } from 'path';
|
|
9
|
+
import { existsSync, readFileSync, writeFileSync, mkdirSync, renameSync } from 'fs';
|
|
10
|
+
import os from 'os';
|
|
11
|
+
const MEMORY_FILE = join(os.homedir(), '.weflow-cli', 'assistant_memory.json');
|
|
12
|
+
const WORKING_MAX = 16; // 超过则最旧一半转摘要
|
|
13
|
+
const SUMMARY_MAX = 800;
|
|
14
|
+
const FACTS_MAX = 30;
|
|
15
|
+
const FACT_EXTRACT_EVERY = 6;
|
|
16
|
+
export class AssistantMemory {
|
|
17
|
+
states = {};
|
|
18
|
+
/** 本进程修改过的用户 (save 时只回写这些用户, 其余保留磁盘最新值) */
|
|
19
|
+
dirtyUsers = new Set();
|
|
20
|
+
constructor() { this.load(); }
|
|
21
|
+
normalize(s) {
|
|
22
|
+
return {
|
|
23
|
+
working: Array.isArray(s.working) ? s.working : [],
|
|
24
|
+
summary: typeof s.summary === 'string' ? s.summary : '',
|
|
25
|
+
facts: Array.isArray(s.facts) ? s.facts : [],
|
|
26
|
+
turnCount: s.turnCount || 0,
|
|
27
|
+
};
|
|
28
|
+
}
|
|
29
|
+
load() {
|
|
30
|
+
try {
|
|
31
|
+
if (existsSync(MEMORY_FILE)) {
|
|
32
|
+
const raw = JSON.parse(readFileSync(MEMORY_FILE, 'utf8'));
|
|
33
|
+
for (const [uid, s] of Object.entries(raw)) {
|
|
34
|
+
this.states[uid] = this.normalize(s);
|
|
35
|
+
}
|
|
36
|
+
}
|
|
37
|
+
}
|
|
38
|
+
catch {
|
|
39
|
+
this.states = {};
|
|
40
|
+
}
|
|
41
|
+
}
|
|
42
|
+
save() {
|
|
43
|
+
try {
|
|
44
|
+
// 读改写合并: MCP server / assistant 守护进程并发写同一文件,
|
|
45
|
+
// 只回写本进程修改过的用户, 防止旧快照覆盖其他进程刚写入的数据 (同 configService 策略)
|
|
46
|
+
const merged = {};
|
|
47
|
+
try {
|
|
48
|
+
if (existsSync(MEMORY_FILE)) {
|
|
49
|
+
const disk = JSON.parse(readFileSync(MEMORY_FILE, 'utf8'));
|
|
50
|
+
for (const [uid, s] of Object.entries(disk)) {
|
|
51
|
+
if (this.dirtyUsers.has(uid))
|
|
52
|
+
continue;
|
|
53
|
+
merged[uid] = this.normalize(s);
|
|
54
|
+
}
|
|
55
|
+
}
|
|
56
|
+
}
|
|
57
|
+
catch { /* 磁盘损坏 → 只落内存态 */ }
|
|
58
|
+
for (const uid of this.dirtyUsers) {
|
|
59
|
+
if (this.states[uid])
|
|
60
|
+
merged[uid] = this.states[uid];
|
|
61
|
+
}
|
|
62
|
+
const dir = join(os.homedir(), '.weflow-cli');
|
|
63
|
+
if (!existsSync(dir))
|
|
64
|
+
mkdirSync(dir, { recursive: true });
|
|
65
|
+
const tmp = `${MEMORY_FILE}.tmp`;
|
|
66
|
+
writeFileSync(tmp, JSON.stringify(merged), 'utf8');
|
|
67
|
+
renameSync(tmp, MEMORY_FILE);
|
|
68
|
+
this.states = merged;
|
|
69
|
+
this.dirtyUsers.clear();
|
|
70
|
+
}
|
|
71
|
+
catch { /* 持久化失败不阻断对话 */ }
|
|
72
|
+
}
|
|
73
|
+
state(userId) {
|
|
74
|
+
if (!this.states[userId]) {
|
|
75
|
+
this.states[userId] = { working: [], summary: '', facts: [], turnCount: 0 };
|
|
76
|
+
}
|
|
77
|
+
return this.states[userId];
|
|
78
|
+
}
|
|
79
|
+
reset(userId) {
|
|
80
|
+
delete this.states[userId];
|
|
81
|
+
this.dirtyUsers.add(userId);
|
|
82
|
+
this.save();
|
|
83
|
+
}
|
|
84
|
+
addTurn(userId, role, content) {
|
|
85
|
+
this.dirtyUsers.add(userId);
|
|
86
|
+
const s = this.state(userId);
|
|
87
|
+
s.working.push({ role, content });
|
|
88
|
+
if (role === 'user')
|
|
89
|
+
s.turnCount++;
|
|
90
|
+
}
|
|
91
|
+
/** L1 窗口 (喂给 LLM 的近期原文) */
|
|
92
|
+
workingWindow(userId) {
|
|
93
|
+
return this.state(userId).working;
|
|
94
|
+
}
|
|
95
|
+
summary(userId) { return this.state(userId).summary; }
|
|
96
|
+
facts(userId) { return this.state(userId).facts; }
|
|
97
|
+
userCount() { return Object.keys(this.states).length; }
|
|
98
|
+
/** L2: 窗口溢出时把最旧一半压进摘要 */
|
|
99
|
+
async compressIfNeeded(userId, llm) {
|
|
100
|
+
const s = this.state(userId);
|
|
101
|
+
if (s.working.length <= WORKING_MAX)
|
|
102
|
+
return false;
|
|
103
|
+
this.dirtyUsers.add(userId);
|
|
104
|
+
const out = s.working.slice(0, Math.floor(s.working.length / 2));
|
|
105
|
+
s.working = s.working.slice(out.length);
|
|
106
|
+
try {
|
|
107
|
+
const prompt = `将以下对话压缩为摘要,与已有摘要合并。保留:关键事实、决定、人名、时间、待办。输出不超过500字,直接输出摘要正文,不要任何前后缀。
|
|
108
|
+
|
|
109
|
+
[已有摘要]
|
|
110
|
+
${s.summary || '(无)'}
|
|
111
|
+
|
|
112
|
+
[新增对话]
|
|
113
|
+
${out.map(t => `${t.role === 'user' ? '用户' : '助手'}: ${t.content}`).join('\n')}`;
|
|
114
|
+
const merged = await llm([{ role: 'user', content: prompt }], 600);
|
|
115
|
+
s.summary = merged.trim().slice(0, SUMMARY_MAX);
|
|
116
|
+
}
|
|
117
|
+
catch {
|
|
118
|
+
// LLM 失败时降级: 粗暴截断拼接, 保住不丢
|
|
119
|
+
const brief = out.map(t => `${t.role === 'user' ? '用户' : '助手'}: ${t.content.slice(0, 40)}`).join(' / ');
|
|
120
|
+
s.summary = `${s.summary} ${brief}`.slice(-SUMMARY_MAX);
|
|
121
|
+
}
|
|
122
|
+
return true;
|
|
123
|
+
}
|
|
124
|
+
/** L3: 定期从最近对话提取持久事实 */
|
|
125
|
+
async extractFactsIfNeeded(userId, llm) {
|
|
126
|
+
const s = this.state(userId);
|
|
127
|
+
if (s.turnCount === 0 || s.turnCount % FACT_EXTRACT_EVERY !== 0)
|
|
128
|
+
return 0;
|
|
129
|
+
const recent = s.working.slice(-FACT_EXTRACT_EVERY);
|
|
130
|
+
if (!recent.length)
|
|
131
|
+
return 0;
|
|
132
|
+
this.dirtyUsers.add(userId);
|
|
133
|
+
try {
|
|
134
|
+
const prompt = `从对话中提取关于用户的持久事实(背景/偏好/项目/人际关系/惯例)。每条不超过40字。只输出 JSON 字符串数组,最多3条,没有则输出 []。不要提取临时性内容。
|
|
135
|
+
|
|
136
|
+
[对话]
|
|
137
|
+
${recent.map(t => `${t.role === 'user' ? '用户' : '助手'}: ${t.content}`).join('\n')}`;
|
|
138
|
+
const raw = await llm([{ role: 'user', content: prompt }], 300);
|
|
139
|
+
const m = raw.match(/\[[\s\S]*\]/);
|
|
140
|
+
if (!m)
|
|
141
|
+
return 0;
|
|
142
|
+
const items = JSON.parse(m[0]);
|
|
143
|
+
let added = 0;
|
|
144
|
+
for (const item of items) {
|
|
145
|
+
if (typeof item !== 'string' || !item.trim())
|
|
146
|
+
continue;
|
|
147
|
+
const fact = item.trim().slice(0, 60);
|
|
148
|
+
// 去重: 与既有事实包含关系判断
|
|
149
|
+
if (s.facts.some(f => f.content.includes(fact) || fact.includes(f.content)))
|
|
150
|
+
continue;
|
|
151
|
+
s.facts.push({ content: fact, ts: Date.now() });
|
|
152
|
+
added++;
|
|
153
|
+
}
|
|
154
|
+
if (s.facts.length > FACTS_MAX)
|
|
155
|
+
s.facts = s.facts.slice(-FACTS_MAX);
|
|
156
|
+
return added;
|
|
157
|
+
}
|
|
158
|
+
catch {
|
|
159
|
+
return 0;
|
|
160
|
+
}
|
|
161
|
+
}
|
|
162
|
+
/** L3 检索: 关键词命中 (为未来向量检索预留同一接口) */
|
|
163
|
+
searchFacts(userId, keyword) {
|
|
164
|
+
const kw = keyword.trim();
|
|
165
|
+
if (!kw)
|
|
166
|
+
return [];
|
|
167
|
+
return this.state(userId).facts.filter(f => f.content.includes(kw) || kw.split(/\s+/).some(w => w.length >= 2 && f.content.includes(w)));
|
|
168
|
+
}
|
|
169
|
+
/** L3 写入 (save_memory 工具用, 立即生效) */
|
|
170
|
+
addFact(userId, content) {
|
|
171
|
+
this.dirtyUsers.add(userId);
|
|
172
|
+
const s = this.state(userId);
|
|
173
|
+
if (s.facts.some(f => f.content.includes(content) || content.includes(f.content)))
|
|
174
|
+
return false;
|
|
175
|
+
s.facts.push({ content, ts: Date.now() });
|
|
176
|
+
if (s.facts.length > FACTS_MAX)
|
|
177
|
+
s.facts = s.facts.slice(-FACTS_MAX);
|
|
178
|
+
this.save();
|
|
179
|
+
return true;
|
|
180
|
+
}
|
|
181
|
+
}
|
|
182
|
+
//# sourceMappingURL=assistantMemory.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"assistantMemory.js","sourceRoot":"","sources":["../../../src/services/assistantMemory.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AACH,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAA;AAC3B,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,aAAa,EAAE,SAAS,EAAE,UAAU,EAAE,MAAM,IAAI,CAAA;AACnF,OAAO,EAAE,MAAM,IAAI,CAAA;AAEnB,MAAM,WAAW,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,uBAAuB,CAAC,CAAA;AAE9E,MAAM,WAAW,GAAG,EAAE,CAAA,CAAC,aAAa;AACpC,MAAM,WAAW,GAAG,GAAG,CAAA;AACvB,MAAM,SAAS,GAAG,EAAE,CAAA;AACpB,MAAM,kBAAkB,GAAG,CAAC,CAAA;AAe5B,MAAM,OAAO,eAAe;IAClB,MAAM,GAA+B,EAAE,CAAA;IAC/C,2CAA2C;IACnC,UAAU,GAAG,IAAI,GAAG,EAAU,CAAA;IAEtC,gBAAgB,IAAI,CAAC,IAAI,EAAE,CAAA,CAAC,CAAC;IAErB,SAAS,CAAC,CAAM;QACtB,OAAO;YACL,OAAO,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;YAClD,OAAO,EAAE,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE;YACvD,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE;YAC5C,SAAS,EAAE,CAAC,CAAC,SAAS,IAAI,CAAC;SAC5B,CAAA;IACH,CAAC;IAEO,IAAI;QACV,IAAI,CAAC;YACH,IAAI,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;gBAC5B,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC,CAAA;gBACzD,KAAK,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAM,GAAG,CAAC,EAAE,CAAC;oBAChD,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAA;gBACtC,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YAAC,IAAI,CAAC,MAAM,GAAG,EAAE,CAAA;QAAC,CAAC;IAC9B,CAAC;IAED,IAAI;QACF,IAAI,CAAC;YACH,6CAA6C;YAC7C,uDAAuD;YACvD,MAAM,MAAM,GAA+B,EAAE,CAAA;YAC7C,IAAI,CAAC;gBACH,IAAI,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;oBAC5B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,WAAW,EAAE,MAAM,CAAC,CAAC,CAAA;oBAC1D,KAAK,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,IAAI,MAAM,CAAC,OAAO,CAAM,IAAI,CAAC,EAAE,CAAC;wBACjD,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC;4BAAE,SAAQ;wBACtC,MAAM,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAA;oBACjC,CAAC;gBACH,CAAC;YACH,CAAC;YAAC,MAAM,CAAC,CAAC,kBAAkB,CAAC,CAAC;YAC9B,KAAK,MAAM,GAAG,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;gBAClC,IAAI,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC;oBAAE,MAAM,CAAC,GAAG,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAA;YACtD,CAAC;YACD,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,aAAa,CAAC,CAAA;YAC7C,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;gBAAE,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAA;YACzD,MAAM,GAAG,GAAG,GAAG,WAAW,MAAM,CAAA;YAChC,aAAa,CAAC,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,MAAM,CAAC,CAAA;YAClD,UAAU,CAAC,GAAG,EAAE,WAAW,CAAC,CAAA;YAC5B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAA;YACpB,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAA;QACzB,CAAC;QAAC,MAAM,CAAC,CAAC,gBAAgB,CAAC,CAAC;IAC9B,CAAC;IAEO,KAAK,CAAC,MAAc;QAC1B,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,EAAE,CAAC;YACzB,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC,EAAE,CAAA;QAC7E,CAAC;QACD,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAA;IAC5B,CAAC;IAED,KAAK,CAAC,MAAc;QAClB,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAA;QAC1B,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,CAAA;QAC3B,IAAI,CAAC,IAAI,EAAE,CAAA;IACb,CAAC;IAED,OAAO,CAAC,MAAc,EAAE,IAA0B,EAAE,OAAe;QACjE,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,CAAA;QAC3B,MAAM,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAA;QAC5B,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAA;QACjC,IAAI,IAAI,KAAK,MAAM;YAAE,CAAC,CAAC,SAAS,EAAE,CAAA;IACpC,CAAC;IAED,2BAA2B;IAC3B,aAAa,CAAC,MAAc;QAC1B,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,OAAO,CAAA;IACnC,CAAC;IAED,OAAO,CAAC,MAAc,IAAY,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,OAAO,CAAA,CAAC,CAAC;IACrE,KAAK,CAAC,MAAc,IAAY,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,KAAK,CAAA,CAAC,CAAC;IACjE,SAAS,KAAa,OAAO,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,CAAA,CAAC,CAAC;IAE9D,yBAAyB;IACzB,KAAK,CAAC,gBAAgB,CAAC,MAAc,EAAE,GAAc;QACnD,MAAM,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAA;QAC5B,IAAI,CAAC,CAAC,OAAO,CAAC,MAAM,IAAI,WAAW;YAAE,OAAO,KAAK,CAAA;QACjD,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,CAAA;QAE3B,MAAM,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAA;QAChE,CAAC,CAAC,OAAO,GAAG,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,CAAA;QAEvC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG;;;EAGnB,CAAC,CAAC,OAAO,IAAI,KAAK;;;EAGlB,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAA;YACzE,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,EAAE,GAAG,CAAC,CAAA;YAClE,CAAC,CAAC,OAAO,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,WAAW,CAAC,CAAA;QACjD,CAAC;QAAC,MAAM,CAAC;YACP,0BAA0B;YAC1B,MAAM,KAAK,GAAG,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAA;YACvG,CAAC,CAAC,OAAO,GAAG,GAAG,CAAC,CAAC,OAAO,IAAI,KAAK,EAAE,CAAC,KAAK,CAAC,CAAC,WAAW,CAAC,CAAA;QACzD,CAAC;QACD,OAAO,IAAI,CAAA;IACb,CAAC;IAED,wBAAwB;IACxB,KAAK,CAAC,oBAAoB,CAAC,MAAc,EAAE,GAAc;QACvD,MAAM,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAA;QAC5B,IAAI,CAAC,CAAC,SAAS,KAAK,CAAC,IAAI,CAAC,CAAC,SAAS,GAAG,kBAAkB,KAAK,CAAC;YAAE,OAAO,CAAC,CAAA;QACzE,MAAM,MAAM,GAAG,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,kBAAkB,CAAC,CAAA;QACnD,IAAI,CAAC,MAAM,CAAC,MAAM;YAAE,OAAO,CAAC,CAAA;QAC5B,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,CAAA;QAE3B,IAAI,CAAC;YACH,MAAM,MAAM,GAAG;;;EAGnB,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAA;YAC5E,MAAM,GAAG,GAAG,MAAM,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,EAAE,GAAG,CAAC,CAAA;YAC/D,MAAM,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC,aAAa,CAAC,CAAA;YAClC,IAAI,CAAC,CAAC;gBAAE,OAAO,CAAC,CAAA;YAChB,MAAM,KAAK,GAAc,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;YACzC,IAAI,KAAK,GAAG,CAAC,CAAA;YACb,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE;oBAAE,SAAQ;gBACtD,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAA;gBACrC,kBAAkB;gBAClB,IAAI,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;oBAAE,SAAQ;gBACrF,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,EAAE,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAA;gBAC/C,KAAK,EAAE,CAAA;YACT,CAAC;YACD,IAAI,CAAC,CAAC,KAAK,CAAC,MAAM,GAAG,SAAS;gBAAE,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,CAAA;YACnE,OAAO,KAAK,CAAA;QACd,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,CAAC,CAAA;QACV,CAAC;IACH,CAAC;IAED,mCAAmC;IACnC,WAAW,CAAC,MAAc,EAAE,OAAe;QACzC,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,EAAE,CAAA;QACzB,IAAI,CAAC,EAAE;YAAE,OAAO,EAAE,CAAA;QAClB,OAAO,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CACzC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAA;IAChG,CAAC;IAED,oCAAoC;IACpC,OAAO,CAAC,MAAc,EAAE,OAAe;QACrC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,CAAA;QAC3B,MAAM,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAA;QAC5B,IAAI,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;YAAE,OAAO,KAAK,CAAA;QAC/F,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAA;QACzC,IAAI,CAAC,CAAC,KAAK,CAAC,MAAM,GAAG,SAAS;YAAE,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,SAAS,CAAC,CAAA;QACnE,IAAI,CAAC,IAAI,EAAE,CAAA;QACX,OAAO,IAAI,CAAA;IACb,CAAC;CACF"}
|
|
@@ -0,0 +1,25 @@
|
|
|
1
|
+
export type PrivacyMode = 'open' | 'balanced' | 'strict';
|
|
2
|
+
export declare function redactText(text: string, mode: PrivacyMode, localInference: boolean): {
|
|
3
|
+
safe: string;
|
|
4
|
+
redactions: number;
|
|
5
|
+
};
|
|
6
|
+
export declare function maskMessageBodyText(body: string, mode: PrivacyMode, localInference: boolean): string;
|
|
7
|
+
export declare class PrivacyGate {
|
|
8
|
+
mode(): PrivacyMode;
|
|
9
|
+
/** 本地引擎数据不出机器, 无需脱敏 */
|
|
10
|
+
isLocalInference(): boolean;
|
|
11
|
+
/**
|
|
12
|
+
* 出境前脱敏。仅用于「工具从数据库挖出的第三方数据」;
|
|
13
|
+
* 用户自己输入的话按原意发送 (用户对自己的话有处置权)。
|
|
14
|
+
*/
|
|
15
|
+
redact(text: string): {
|
|
16
|
+
safe: string;
|
|
17
|
+
redactions: number;
|
|
18
|
+
};
|
|
19
|
+
/** strict 模式: 第三方聊天正文不出境, 只保留元数据形态 */
|
|
20
|
+
maskMessageBody(body: string): string;
|
|
21
|
+
/** 审计日志 — 只记事件与字节量, 绝不记内容 */
|
|
22
|
+
audit(event: string, bytes?: number, extra?: string): void;
|
|
23
|
+
}
|
|
24
|
+
export declare const privacyGate: PrivacyGate;
|
|
25
|
+
//# sourceMappingURL=assistantPrivacy.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"assistantPrivacy.d.ts","sourceRoot":"","sources":["../../../src/services/assistantPrivacy.ts"],"names":[],"mappings":"AAkBA,MAAM,MAAM,WAAW,GAAG,MAAM,GAAG,UAAU,GAAG,QAAQ,CAAA;AAcxD,wBAAgB,UAAU,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,cAAc,EAAE,OAAO,GAAG;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,UAAU,EAAE,MAAM,CAAA;CAAE,CAUzH;AAED,wBAAgB,mBAAmB,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,cAAc,EAAE,OAAO,GAAG,MAAM,CAGpG;AAED,qBAAa,WAAW;IACtB,IAAI,IAAI,WAAW;IAKnB,uBAAuB;IACvB,gBAAgB,IAAI,OAAO;IAK3B;;;OAGG;IACH,MAAM,CAAC,IAAI,EAAE,MAAM,GAAG;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,MAAM,CAAA;KAAE;IAI1D,sCAAsC;IACtC,eAAe,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM;IAIrC,6BAA6B;IAC7B,KAAK,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,SAAI,EAAE,KAAK,SAAK,GAAG,IAAI;CASlD;AAED,eAAO,MAAM,WAAW,aAAoB,CAAA"}
|
|
@@ -0,0 +1,77 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 隐私关卡 — 本地/云端数据出境的唯一检查点。
|
|
3
|
+
*
|
|
4
|
+
* 原则:
|
|
5
|
+
* - 数据库查询、记忆存储、工具执行全部在本机 (可信区)
|
|
6
|
+
* - 只有经 redact() 处理后的文本才允许发往云端 LLM
|
|
7
|
+
* - 本地推理引擎 (Ollama 等) 不出境, 跳过脱敏以保留完整能力
|
|
8
|
+
*
|
|
9
|
+
* 模式 (config: assistantPrivacy):
|
|
10
|
+
* - open 不脱敏 (自担风险)
|
|
11
|
+
* - balanced 默认; 工具输出中的 PII (电话/证件/邮箱/密钥/链接) 打码
|
|
12
|
+
* - strict balanced 基础上, 第三方聊天正文不出境 (仅保留时间/方向/类型)
|
|
13
|
+
*/
|
|
14
|
+
import { join } from 'path';
|
|
15
|
+
import { existsSync, mkdirSync, appendFileSync } from 'fs';
|
|
16
|
+
import os from 'os';
|
|
17
|
+
import { configService } from './configService.js';
|
|
18
|
+
const AUDIT_FILE = join(os.homedir(), '.weflow-cli', 'assistant_audit.log');
|
|
19
|
+
const PII_RULES = [
|
|
20
|
+
{ re: /(?<!\d)1[3-9]\d{9}(?!\d)/g, mask: '[电话]' },
|
|
21
|
+
{ re: /(?<!\d)\d{17}[\dXx](?!\d)/g, mask: '[证件号]' },
|
|
22
|
+
{ re: /[a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,}/g, mask: '[邮箱]' },
|
|
23
|
+
{ re: /sk-[a-zA-Z0-9]{16,}/g, mask: '[密钥]' },
|
|
24
|
+
{ re: /https?:\/\/\S+/g, mask: '[链接]' },
|
|
25
|
+
];
|
|
26
|
+
export function redactText(text, mode, localInference) {
|
|
27
|
+
if (mode === 'open' || localInference) {
|
|
28
|
+
return { safe: text, redactions: 0 };
|
|
29
|
+
}
|
|
30
|
+
let count = 0;
|
|
31
|
+
let safe = text;
|
|
32
|
+
for (const { re, mask } of PII_RULES) {
|
|
33
|
+
safe = safe.replace(re, () => { count++; return mask; });
|
|
34
|
+
}
|
|
35
|
+
return { safe, redactions: count };
|
|
36
|
+
}
|
|
37
|
+
export function maskMessageBodyText(body, mode, localInference) {
|
|
38
|
+
if (mode !== 'strict' || localInference)
|
|
39
|
+
return body;
|
|
40
|
+
return `[内容${body.length}字已按严格模式屏蔽]`;
|
|
41
|
+
}
|
|
42
|
+
export class PrivacyGate {
|
|
43
|
+
mode() {
|
|
44
|
+
const m = configService.get('assistantPrivacy');
|
|
45
|
+
return m === 'open' || m === 'strict' ? m : 'balanced';
|
|
46
|
+
}
|
|
47
|
+
/** 本地引擎数据不出机器, 无需脱敏 */
|
|
48
|
+
isLocalInference() {
|
|
49
|
+
const engine = configService.get('aiEngine');
|
|
50
|
+
return engine === 'ollama' || engine === 'lmstudio' || engine === 'local';
|
|
51
|
+
}
|
|
52
|
+
/**
|
|
53
|
+
* 出境前脱敏。仅用于「工具从数据库挖出的第三方数据」;
|
|
54
|
+
* 用户自己输入的话按原意发送 (用户对自己的话有处置权)。
|
|
55
|
+
*/
|
|
56
|
+
redact(text) {
|
|
57
|
+
return redactText(text, this.mode(), this.isLocalInference());
|
|
58
|
+
}
|
|
59
|
+
/** strict 模式: 第三方聊天正文不出境, 只保留元数据形态 */
|
|
60
|
+
maskMessageBody(body) {
|
|
61
|
+
return maskMessageBodyText(body, this.mode(), this.isLocalInference());
|
|
62
|
+
}
|
|
63
|
+
/** 审计日志 — 只记事件与字节量, 绝不记内容 */
|
|
64
|
+
audit(event, bytes = 0, extra = '') {
|
|
65
|
+
try {
|
|
66
|
+
const dir = join(os.homedir(), '.weflow-cli');
|
|
67
|
+
if (!existsSync(dir))
|
|
68
|
+
mkdirSync(dir, { recursive: true });
|
|
69
|
+
const line = `[${new Date().toLocaleString('zh-CN')}] ${event}` +
|
|
70
|
+
(bytes ? ` ${bytes}B` : '') + (extra ? ` ${extra}` : '');
|
|
71
|
+
appendFileSync(AUDIT_FILE, line + '\n', 'utf8');
|
|
72
|
+
}
|
|
73
|
+
catch { /* 审计失败不阻断主流程 */ }
|
|
74
|
+
}
|
|
75
|
+
}
|
|
76
|
+
export const privacyGate = new PrivacyGate();
|
|
77
|
+
//# sourceMappingURL=assistantPrivacy.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"assistantPrivacy.js","sourceRoot":"","sources":["../../../src/services/assistantPrivacy.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AACH,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAA;AAC3B,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,cAAc,EAAE,MAAM,IAAI,CAAA;AAC1D,OAAO,EAAE,MAAM,IAAI,CAAA;AACnB,OAAO,EAAE,aAAa,EAAE,MAAM,oBAAoB,CAAA;AAIlD,MAAM,UAAU,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,aAAa,EAAE,qBAAqB,CAAC,CAAA;AAI3E,MAAM,SAAS,GAAW;IACxB,EAAE,EAAE,EAAE,2BAA2B,EAAE,IAAI,EAAE,MAAM,EAAE;IACjD,EAAE,EAAE,EAAE,4BAA4B,EAAE,IAAI,EAAE,OAAO,EAAE;IACnD,EAAE,EAAE,EAAE,iDAAiD,EAAE,IAAI,EAAE,MAAM,EAAE;IACvE,EAAE,EAAE,EAAE,sBAAsB,EAAE,IAAI,EAAE,MAAM,EAAE;IAC5C,EAAE,EAAE,EAAE,iBAAiB,EAAE,IAAI,EAAE,MAAM,EAAE;CACxC,CAAA;AAED,MAAM,UAAU,UAAU,CAAC,IAAY,EAAE,IAAiB,EAAE,cAAuB;IACjF,IAAI,IAAI,KAAK,MAAM,IAAI,cAAc,EAAE,CAAC;QACtC,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC,EAAE,CAAA;IACtC,CAAC;IACD,IAAI,KAAK,GAAG,CAAC,CAAA;IACb,IAAI,IAAI,GAAG,IAAI,CAAA;IACf,KAAK,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,IAAI,SAAS,EAAE,CAAC;QACrC,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,GAAG,EAAE,GAAG,KAAK,EAAE,CAAC,CAAC,OAAO,IAAI,CAAA,CAAC,CAAC,CAAC,CAAA;IACzD,CAAC;IACD,OAAO,EAAE,IAAI,EAAE,UAAU,EAAE,KAAK,EAAE,CAAA;AACpC,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,IAAY,EAAE,IAAiB,EAAE,cAAuB;IAC1F,IAAI,IAAI,KAAK,QAAQ,IAAI,cAAc;QAAE,OAAO,IAAI,CAAA;IACpD,OAAO,MAAM,IAAI,CAAC,MAAM,YAAY,CAAA;AACtC,CAAC;AAED,MAAM,OAAO,WAAW;IACtB,IAAI;QACF,MAAM,CAAC,GAAG,aAAa,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAA;QAC/C,OAAO,CAAC,KAAK,MAAM,IAAI,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,CAAA;IACxD,CAAC;IAED,uBAAuB;IACvB,gBAAgB;QACd,MAAM,MAAM,GAAG,aAAa,CAAC,GAAG,CAAC,UAAU,CAAC,CAAA;QAC5C,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,UAAU,IAAI,MAAM,KAAK,OAAO,CAAA;IAC3E,CAAC;IAED;;;OAGG;IACH,MAAM,CAAC,IAAY;QACjB,OAAO,UAAU,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,IAAI,CAAC,gBAAgB,EAAE,CAAC,CAAA;IAC/D,CAAC;IAED,sCAAsC;IACtC,eAAe,CAAC,IAAY;QAC1B,OAAO,mBAAmB,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE,IAAI,CAAC,gBAAgB,EAAE,CAAC,CAAA;IACxE,CAAC;IAED,6BAA6B;IAC7B,KAAK,CAAC,KAAa,EAAE,KAAK,GAAG,CAAC,EAAE,KAAK,GAAG,EAAE;QACxC,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,aAAa,CAAC,CAAA;YAC7C,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC;gBAAE,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAA;YACzD,MAAM,IAAI,GAAG,IAAI,IAAI,IAAI,EAAE,CAAC,cAAc,CAAC,OAAO,CAAC,KAAK,KAAK,EAAE;gBAC7D,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAA;YAC1D,cAAc,CAAC,UAAU,EAAE,IAAI,GAAG,IAAI,EAAE,MAAM,CAAC,CAAA;QACjD,CAAC;QAAC,MAAM,CAAC,CAAC,gBAAgB,CAAC,CAAC;IAC9B,CAAC;CACF;AAED,MAAM,CAAC,MAAM,WAAW,GAAG,IAAI,WAAW,EAAE,CAAA"}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
import type { WechatInboundMessage } from '../types.js';
|
|
2
|
+
export type AssistantAccessReason = 'allowed' | 'direct-not-whitelisted' | 'group-not-whitelisted' | 'group-sender-missing' | 'group-sender-not-whitelisted' | 'group-mention-required';
|
|
3
|
+
export interface AssistantAccessPolicy {
|
|
4
|
+
directWhitelist: string;
|
|
5
|
+
groupWhitelist: string;
|
|
6
|
+
requireGroupMention: boolean;
|
|
7
|
+
}
|
|
8
|
+
export interface AssistantAccessDecision {
|
|
9
|
+
allowed: boolean;
|
|
10
|
+
reason: AssistantAccessReason;
|
|
11
|
+
}
|
|
12
|
+
export declare function evaluateAssistantAccess(message: Pick<WechatInboundMessage, 'conversationType' | 'conversationId' | 'senderId' | 'mentionedBot'>, policy: AssistantAccessPolicy): AssistantAccessDecision;
|
|
13
|
+
/**
|
|
14
|
+
* Accept group routing only when the upstream payload explicitly identifies a group.
|
|
15
|
+
* Unknown payloads remain direct messages so a platform change cannot silently enable groups.
|
|
16
|
+
*/
|
|
17
|
+
export declare function resolveInboundRouting(record: Record<string, any>, botId?: string): Pick<WechatInboundMessage, 'conversationType' | 'conversationId' | 'senderId' | 'mentionedBot'>;
|
|
18
|
+
//# sourceMappingURL=assistantRouting.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"assistantRouting.d.ts","sourceRoot":"","sources":["../../../src/services/assistantRouting.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAA;AAEvD,MAAM,MAAM,qBAAqB,GAC7B,SAAS,GACT,wBAAwB,GACxB,uBAAuB,GACvB,sBAAsB,GACtB,8BAA8B,GAC9B,wBAAwB,CAAA;AAE5B,MAAM,WAAW,qBAAqB;IACpC,eAAe,EAAE,MAAM,CAAA;IACvB,cAAc,EAAE,MAAM,CAAA;IACtB,mBAAmB,EAAE,OAAO,CAAA;CAC7B;AAED,MAAM,WAAW,uBAAuB;IACtC,OAAO,EAAE,OAAO,CAAA;IAChB,MAAM,EAAE,qBAAqB,CAAA;CAC9B;AAMD,wBAAgB,uBAAuB,CACrC,OAAO,EAAE,IAAI,CAAC,oBAAoB,EAAE,kBAAkB,GAAG,gBAAgB,GAAG,UAAU,GAAG,cAAc,CAAC,EACxG,MAAM,EAAE,qBAAqB,GAC5B,uBAAuB,CAmBzB;AAoBD;;;GAGG;AACH,wBAAgB,qBAAqB,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,KAAK,SAAK,GAAG,IAAI,CAAC,oBAAoB,EAAE,kBAAkB,GAAG,gBAAgB,GAAG,UAAU,GAAG,cAAc,CAAC,CAoB9K"}
|
|
@@ -0,0 +1,68 @@
|
|
|
1
|
+
function parseList(value) {
|
|
2
|
+
return new Set(value.split(/[,;\s]+/).map(item => item.trim()).filter(Boolean));
|
|
3
|
+
}
|
|
4
|
+
export function evaluateAssistantAccess(message, policy) {
|
|
5
|
+
const directWhitelist = parseList(policy.directWhitelist);
|
|
6
|
+
if (message.conversationType !== 'group') {
|
|
7
|
+
return directWhitelist.has(message.senderId)
|
|
8
|
+
? { allowed: true, reason: 'allowed' }
|
|
9
|
+
: { allowed: false, reason: 'direct-not-whitelisted' };
|
|
10
|
+
}
|
|
11
|
+
if (!parseList(policy.groupWhitelist).has(message.conversationId)) {
|
|
12
|
+
return { allowed: false, reason: 'group-not-whitelisted' };
|
|
13
|
+
}
|
|
14
|
+
if (!message.senderId)
|
|
15
|
+
return { allowed: false, reason: 'group-sender-missing' };
|
|
16
|
+
if (!directWhitelist.has(message.senderId)) {
|
|
17
|
+
return { allowed: false, reason: 'group-sender-not-whitelisted' };
|
|
18
|
+
}
|
|
19
|
+
if (policy.requireGroupMention && !message.mentionedBot) {
|
|
20
|
+
return { allowed: false, reason: 'group-mention-required' };
|
|
21
|
+
}
|
|
22
|
+
return { allowed: true, reason: 'allowed' };
|
|
23
|
+
}
|
|
24
|
+
function firstString(record, keys) {
|
|
25
|
+
for (const key of keys) {
|
|
26
|
+
const value = record[key];
|
|
27
|
+
if (typeof value === 'string' && value.trim())
|
|
28
|
+
return value.trim();
|
|
29
|
+
}
|
|
30
|
+
return '';
|
|
31
|
+
}
|
|
32
|
+
function isBotMentioned(record, botId) {
|
|
33
|
+
if (record.is_at_bot === true || record.is_mentioned === true || record.mentioned_bot === true)
|
|
34
|
+
return true;
|
|
35
|
+
if (!botId)
|
|
36
|
+
return false;
|
|
37
|
+
for (const key of ['at_user_ids', 'mentioned_user_ids', 'mention_list']) {
|
|
38
|
+
const value = record[key];
|
|
39
|
+
if (Array.isArray(value) && value.some(item => String(item) === botId))
|
|
40
|
+
return true;
|
|
41
|
+
}
|
|
42
|
+
return false;
|
|
43
|
+
}
|
|
44
|
+
/**
|
|
45
|
+
* Accept group routing only when the upstream payload explicitly identifies a group.
|
|
46
|
+
* Unknown payloads remain direct messages so a platform change cannot silently enable groups.
|
|
47
|
+
*/
|
|
48
|
+
export function resolveInboundRouting(record, botId = '') {
|
|
49
|
+
const fromUserId = firstString(record, ['from_user_id', 'fromUserId']);
|
|
50
|
+
const groupId = firstString(record, ['chatroom_id', 'chat_room_id', 'group_id', 'groupId', 'room_id', 'roomId'])
|
|
51
|
+
|| (fromUserId.endsWith('@chatroom') ? fromUserId : '');
|
|
52
|
+
if (!groupId) {
|
|
53
|
+
return {
|
|
54
|
+
conversationType: 'direct',
|
|
55
|
+
conversationId: fromUserId,
|
|
56
|
+
senderId: fromUserId,
|
|
57
|
+
mentionedBot: false,
|
|
58
|
+
};
|
|
59
|
+
}
|
|
60
|
+
const explicitSender = firstString(record, ['sender_id', 'sender_user_id', 'member_id', 'from_member_id']);
|
|
61
|
+
return {
|
|
62
|
+
conversationType: 'group',
|
|
63
|
+
conversationId: groupId,
|
|
64
|
+
senderId: explicitSender || (fromUserId !== groupId ? fromUserId : ''),
|
|
65
|
+
mentionedBot: isBotMentioned(record, botId),
|
|
66
|
+
};
|
|
67
|
+
}
|
|
68
|
+
//# sourceMappingURL=assistantRouting.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"assistantRouting.js","sourceRoot":"","sources":["../../../src/services/assistantRouting.ts"],"names":[],"mappings":"AAqBA,SAAS,SAAS,CAAC,KAAa;IAC9B,OAAO,IAAI,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAA;AACjF,CAAC;AAED,MAAM,UAAU,uBAAuB,CACrC,OAAwG,EACxG,MAA6B;IAE7B,MAAM,eAAe,GAAG,SAAS,CAAC,MAAM,CAAC,eAAe,CAAC,CAAA;IACzD,IAAI,OAAO,CAAC,gBAAgB,KAAK,OAAO,EAAE,CAAC;QACzC,OAAO,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC;YAC1C,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE;YACtC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,wBAAwB,EAAE,CAAA;IAC1D,CAAC;IAED,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,CAAC;QAClE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,uBAAuB,EAAE,CAAA;IAC5D,CAAC;IACD,IAAI,CAAC,OAAO,CAAC,QAAQ;QAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,sBAAsB,EAAE,CAAA;IAChF,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC3C,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,8BAA8B,EAAE,CAAA;IACnE,CAAC;IACD,IAAI,MAAM,CAAC,mBAAmB,IAAI,CAAC,OAAO,CAAC,YAAY,EAAE,CAAC;QACxD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,wBAAwB,EAAE,CAAA;IAC7D,CAAC;IACD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,SAAS,EAAE,CAAA;AAC7C,CAAC;AAED,SAAS,WAAW,CAAC,MAA2B,EAAE,IAAc;IAC9D,KAAK,MAAM,GAAG,IAAI,IAAI,EAAE,CAAC;QACvB,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,CAAA;QACzB,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,IAAI,EAAE;YAAE,OAAO,KAAK,CAAC,IAAI,EAAE,CAAA;IACpE,CAAC;IACD,OAAO,EAAE,CAAA;AACX,CAAC;AAED,SAAS,cAAc,CAAC,MAA2B,EAAE,KAAa;IAChE,IAAI,MAAM,CAAC,SAAS,KAAK,IAAI,IAAI,MAAM,CAAC,YAAY,KAAK,IAAI,IAAI,MAAM,CAAC,aAAa,KAAK,IAAI;QAAE,OAAO,IAAI,CAAA;IAC3G,IAAI,CAAC,KAAK;QAAE,OAAO,KAAK,CAAA;IACxB,KAAK,MAAM,GAAG,IAAI,CAAC,aAAa,EAAE,oBAAoB,EAAE,cAAc,CAAC,EAAE,CAAC;QACxE,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,CAAA;QACzB,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,KAAK,CAAC;YAAE,OAAO,IAAI,CAAA;IACrF,CAAC;IACD,OAAO,KAAK,CAAA;AACd,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,qBAAqB,CAAC,MAA2B,EAAE,KAAK,GAAG,EAAE;IAC3E,MAAM,UAAU,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC,cAAc,EAAE,YAAY,CAAC,CAAC,CAAA;IACtE,MAAM,OAAO,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC,aAAa,EAAE,cAAc,EAAE,UAAU,EAAE,SAAS,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;WAC3G,CAAC,UAAU,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC,CAAA;IACzD,IAAI,CAAC,OAAO,EAAE,CAAC;QACb,OAAO;YACL,gBAAgB,EAAE,QAAQ;YAC1B,cAAc,EAAE,UAAU;YAC1B,QAAQ,EAAE,UAAU;YACpB,YAAY,EAAE,KAAK;SACpB,CAAA;IACH,CAAC;IAED,MAAM,cAAc,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC,WAAW,EAAE,gBAAgB,EAAE,WAAW,EAAE,gBAAgB,CAAC,CAAC,CAAA;IAC1G,OAAO;QACL,gBAAgB,EAAE,OAAO;QACzB,cAAc,EAAE,OAAO;QACvB,QAAQ,EAAE,cAAc,IAAI,CAAC,UAAU,KAAK,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,CAAC;QACtE,YAAY,EAAE,cAAc,CAAC,MAAM,EAAE,KAAK,CAAC;KAC5C,CAAA;AACH,CAAC"}
|
|
@@ -0,0 +1,37 @@
|
|
|
1
|
+
import type { Message } from '../types.js';
|
|
2
|
+
export declare class AssistantService {
|
|
3
|
+
private svc;
|
|
4
|
+
private memory;
|
|
5
|
+
private running;
|
|
6
|
+
/** 消息串行队列: 保证 handleMessage 不并发交错 (记忆窗口一致性) */
|
|
7
|
+
private queue;
|
|
8
|
+
/** 每日用量计数 (内存态, 重启重置 — 配额护栏防烧钱, 无需持久精确) */
|
|
9
|
+
private dailyCount;
|
|
10
|
+
private dailyDate;
|
|
11
|
+
private engineConfig;
|
|
12
|
+
/**
|
|
13
|
+
* Direct messages require an explicit sender allowlist. Groups are disabled
|
|
14
|
+
* until the upstream provides an explicit group ID, and then require all
|
|
15
|
+
* three controls: group allowlist, sender allowlist, and an @ mention.
|
|
16
|
+
*/
|
|
17
|
+
private accessDecision;
|
|
18
|
+
private dailyQuotaLeft;
|
|
19
|
+
/** 底层 LLM 调用 (含出境审计) */
|
|
20
|
+
private callLLM;
|
|
21
|
+
/** 在明确授权后分析本地聊天,输出证据线索而非法律结论。 */
|
|
22
|
+
reviewEvidence(talker: string, messages: Message[], allowCloud?: boolean): Promise<{
|
|
23
|
+
text: string;
|
|
24
|
+
localInference: boolean;
|
|
25
|
+
redactions: number;
|
|
26
|
+
}>;
|
|
27
|
+
/** 组装系统提示: 基础人格 + L2 摘要 + L3 事实 */
|
|
28
|
+
private buildSystemPrompt;
|
|
29
|
+
/** 单条消息处理: 指令路由 → ReAct 循环 → 记忆更新 */
|
|
30
|
+
handleMessage(userId: string, text: string, kind: string): Promise<string>;
|
|
31
|
+
/** 启动常驻监听 */
|
|
32
|
+
start(onLog?: (line: string) => void): Promise<void>;
|
|
33
|
+
stop(): void;
|
|
34
|
+
isRunning(): boolean;
|
|
35
|
+
memoryUserCount(): number;
|
|
36
|
+
}
|
|
37
|
+
//# sourceMappingURL=assistantService.d.ts.map
|