@zhin.js/agent 0.0.0 → 0.0.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.
Files changed (139) hide show
  1. package/CHANGELOG.md +8 -0
  2. package/README.md +226 -0
  3. package/lib/agent.d.ts +9 -7
  4. package/lib/agent.d.ts.map +1 -1
  5. package/lib/agent.js +63 -29
  6. package/lib/agent.js.map +1 -1
  7. package/lib/bootstrap.js +1 -1
  8. package/lib/bootstrap.js.map +1 -1
  9. package/lib/builtin-tools.d.ts.map +1 -1
  10. package/lib/builtin-tools.js +34 -31
  11. package/lib/builtin-tools.js.map +1 -1
  12. package/lib/compaction.js +1 -1
  13. package/lib/compaction.js.map +1 -1
  14. package/lib/context-manager.js +1 -1
  15. package/lib/context-manager.js.map +1 -1
  16. package/lib/conversation-memory.d.ts +11 -0
  17. package/lib/conversation-memory.d.ts.map +1 -1
  18. package/lib/conversation-memory.js +39 -1
  19. package/lib/conversation-memory.js.map +1 -1
  20. package/lib/cron-engine.d.ts.map +1 -1
  21. package/lib/cron-engine.js +5 -3
  22. package/lib/cron-engine.js.map +1 -1
  23. package/lib/follow-up.js +1 -1
  24. package/lib/follow-up.js.map +1 -1
  25. package/lib/hooks.js +1 -1
  26. package/lib/hooks.js.map +1 -1
  27. package/lib/index.d.ts +2 -0
  28. package/lib/index.d.ts.map +1 -1
  29. package/lib/index.js +1 -0
  30. package/lib/index.js.map +1 -1
  31. package/lib/init/create-zhin-agent.d.ts +3 -0
  32. package/lib/init/create-zhin-agent.d.ts.map +1 -0
  33. package/lib/init/create-zhin-agent.js +127 -0
  34. package/lib/init/create-zhin-agent.js.map +1 -0
  35. package/lib/init/register-ai-service.d.ts +7 -0
  36. package/lib/init/register-ai-service.d.ts.map +1 -0
  37. package/lib/init/register-ai-service.js +42 -0
  38. package/lib/init/register-ai-service.js.map +1 -0
  39. package/lib/init/register-ai-trigger.d.ts +7 -0
  40. package/lib/init/register-ai-trigger.d.ts.map +1 -0
  41. package/lib/init/register-ai-trigger.js +163 -0
  42. package/lib/init/register-ai-trigger.js.map +1 -0
  43. package/lib/init/register-builtin-tools.d.ts +3 -0
  44. package/lib/init/register-builtin-tools.d.ts.map +1 -0
  45. package/lib/init/register-builtin-tools.js +179 -0
  46. package/lib/init/register-builtin-tools.js.map +1 -0
  47. package/lib/init/register-db-models.d.ts +2 -0
  48. package/lib/init/register-db-models.d.ts.map +1 -0
  49. package/lib/init/register-db-models.js +28 -0
  50. package/lib/init/register-db-models.js.map +1 -0
  51. package/lib/init/register-db-upgrade.d.ts +10 -0
  52. package/lib/init/register-db-upgrade.d.ts.map +1 -0
  53. package/lib/init/register-db-upgrade.js +67 -0
  54. package/lib/init/register-db-upgrade.js.map +1 -0
  55. package/lib/init/register-management-tools.d.ts +6 -0
  56. package/lib/init/register-management-tools.d.ts.map +1 -0
  57. package/lib/init/register-management-tools.js +68 -0
  58. package/lib/init/register-management-tools.js.map +1 -0
  59. package/lib/init/register-message-recorder.d.ts +3 -0
  60. package/lib/init/register-message-recorder.d.ts.map +1 -0
  61. package/lib/init/register-message-recorder.js +27 -0
  62. package/lib/init/register-message-recorder.js.map +1 -0
  63. package/lib/init/register-tool-service.d.ts +2 -0
  64. package/lib/init/register-tool-service.d.ts.map +1 -0
  65. package/lib/init/register-tool-service.js +9 -0
  66. package/lib/init/register-tool-service.js.map +1 -0
  67. package/lib/init/shared-refs.d.ts +14 -0
  68. package/lib/init/shared-refs.d.ts.map +1 -0
  69. package/lib/init/shared-refs.js +7 -0
  70. package/lib/init/shared-refs.js.map +1 -0
  71. package/lib/init/types.d.ts +15 -0
  72. package/lib/init/types.d.ts.map +1 -0
  73. package/lib/init/types.js +7 -0
  74. package/lib/init/types.js.map +1 -0
  75. package/lib/init.d.ts +14 -17
  76. package/lib/init.d.ts.map +1 -1
  77. package/lib/init.js +34 -670
  78. package/lib/init.js.map +1 -1
  79. package/lib/rate-limiter.js +1 -1
  80. package/lib/rate-limiter.js.map +1 -1
  81. package/lib/service.js +2 -2
  82. package/lib/service.js.map +1 -1
  83. package/lib/session.d.ts +7 -0
  84. package/lib/session.d.ts.map +1 -1
  85. package/lib/session.js +26 -4
  86. package/lib/session.js.map +1 -1
  87. package/lib/storage.d.ts +68 -0
  88. package/lib/storage.d.ts.map +1 -0
  89. package/lib/storage.js +105 -0
  90. package/lib/storage.js.map +1 -0
  91. package/lib/subagent.js +1 -1
  92. package/lib/subagent.js.map +1 -1
  93. package/lib/tools.d.ts.map +1 -1
  94. package/lib/tools.js +8 -9
  95. package/lib/tools.js.map +1 -1
  96. package/lib/user-profile.js +1 -1
  97. package/lib/user-profile.js.map +1 -1
  98. package/lib/zhin-agent/builtin-tools.js.map +1 -1
  99. package/lib/zhin-agent/config.d.ts +1 -1
  100. package/lib/zhin-agent/config.d.ts.map +1 -1
  101. package/lib/zhin-agent/config.js.map +1 -1
  102. package/lib/zhin-agent/index.js +1 -1
  103. package/lib/zhin-agent/index.js.map +1 -1
  104. package/lib/zhin-agent/tool-collector.js +3 -3
  105. package/lib/zhin-agent/tool-collector.js.map +1 -1
  106. package/package.json +4 -3
  107. package/src/agent.ts +53 -30
  108. package/src/bootstrap.ts +1 -1
  109. package/src/builtin-tools.ts +44 -40
  110. package/src/compaction.ts +1 -1
  111. package/src/context-manager.ts +1 -1
  112. package/src/conversation-memory.ts +43 -1
  113. package/src/cron-engine.ts +5 -3
  114. package/src/follow-up.ts +1 -1
  115. package/src/hooks.ts +1 -1
  116. package/src/index.ts +10 -0
  117. package/src/init/create-zhin-agent.ts +133 -0
  118. package/src/init/register-ai-service.ts +53 -0
  119. package/src/init/register-ai-trigger.ts +179 -0
  120. package/src/init/register-builtin-tools.ts +177 -0
  121. package/src/init/register-db-models.ts +31 -0
  122. package/src/init/register-db-upgrade.ts +77 -0
  123. package/src/init/register-management-tools.ts +71 -0
  124. package/src/init/register-message-recorder.ts +31 -0
  125. package/src/init/register-tool-service.ts +9 -0
  126. package/src/init/shared-refs.ts +20 -0
  127. package/src/init/types.ts +18 -0
  128. package/src/init.ts +35 -735
  129. package/src/rate-limiter.ts +1 -1
  130. package/src/service.ts +4 -4
  131. package/src/session.ts +26 -4
  132. package/src/storage.ts +135 -0
  133. package/src/subagent.ts +1 -1
  134. package/src/tools.ts +7 -10
  135. package/src/user-profile.ts +1 -1
  136. package/src/zhin-agent/builtin-tools.ts +2 -2
  137. package/src/zhin-agent/config.ts +3 -3
  138. package/src/zhin-agent/index.ts +1 -1
  139. package/src/zhin-agent/tool-collector.ts +8 -8
@@ -0,0 +1,163 @@
1
+ /**
2
+ * Register AI trigger handler via MessageDispatcher or fallback middleware.
3
+ */
4
+ import './types.js';
5
+ import { getPlugin, shouldTriggerAI, inferSenderPermissions, parseRichMediaContent, mergeAITriggerConfig } from '@zhin.js/core';
6
+ function extractImageUrls(message) {
7
+ const urls = [];
8
+ const raw = typeof message.$raw === 'string' ? message.$raw : JSON.stringify(message.$raw || '');
9
+ const xmlMatches = raw.match(/<image[^>]+url="([^"]+)"/g);
10
+ if (xmlMatches) {
11
+ for (const m of xmlMatches) {
12
+ const urlMatch = m.match(/url="([^"]+)"/);
13
+ if (urlMatch)
14
+ urls.push(urlMatch[1]);
15
+ }
16
+ }
17
+ const cqMatches = raw.match(/\[CQ:image[^\]]*url=([^\],]+)/g);
18
+ if (cqMatches) {
19
+ for (const m of cqMatches) {
20
+ const urlMatch = m.match(/url=([^\],]+)/);
21
+ if (urlMatch)
22
+ urls.push(urlMatch[1]);
23
+ }
24
+ }
25
+ return urls;
26
+ }
27
+ function renderOutput(elements) {
28
+ const parts = [];
29
+ for (const el of elements) {
30
+ switch (el.type) {
31
+ case 'text':
32
+ if (el.content)
33
+ parts.push(el.content);
34
+ break;
35
+ case 'image':
36
+ parts.push(`<image url="${el.url}"/>`);
37
+ break;
38
+ case 'audio':
39
+ parts.push(`<audio url="${el.url}"/>`);
40
+ break;
41
+ case 'video':
42
+ parts.push(`<video url="${el.url}"/>`);
43
+ break;
44
+ case 'card': {
45
+ const cp = [`📋 ${el.title}`];
46
+ if (el.description)
47
+ cp.push(el.description);
48
+ if (el.fields?.length)
49
+ for (const f of el.fields)
50
+ cp.push(` ${f.label}: ${f.value}`);
51
+ if (el.imageUrl)
52
+ cp.push(`<image url="${el.imageUrl}"/>`);
53
+ parts.push(cp.join('\n'));
54
+ break;
55
+ }
56
+ case 'file':
57
+ parts.push(`📎 ${el.name}: ${el.url}`);
58
+ break;
59
+ }
60
+ }
61
+ return parts.join('\n') || '';
62
+ }
63
+ export function registerAITrigger(refs) {
64
+ const plugin = getPlugin();
65
+ const { useContext, root, logger } = plugin;
66
+ useContext('ai', (ai) => {
67
+ const rawConfig = ai.getTriggerConfig();
68
+ const triggerConfig = mergeAITriggerConfig(rawConfig);
69
+ if (!triggerConfig.enabled) {
70
+ logger.info('AI Trigger is disabled');
71
+ return;
72
+ }
73
+ const handleAIMessage = async (message, content) => {
74
+ const t0 = performance.now();
75
+ if (!ai.isReady())
76
+ return;
77
+ if (triggerConfig.thinkingMessage)
78
+ await message.$reply(triggerConfig.thinkingMessage);
79
+ const permissions = inferSenderPermissions(message, triggerConfig);
80
+ const toolContext = {
81
+ platform: message.$adapter,
82
+ botId: message.$bot,
83
+ sceneId: message.$channel?.id || message.$sender.id,
84
+ senderId: message.$sender.id,
85
+ message,
86
+ scope: permissions.scope,
87
+ senderPermissionLevel: permissions.permissionLevel,
88
+ isGroupAdmin: permissions.isGroupAdmin,
89
+ isGroupOwner: permissions.isGroupOwner,
90
+ isBotAdmin: permissions.isBotAdmin,
91
+ isOwner: permissions.isOwner,
92
+ };
93
+ const tCollect = performance.now();
94
+ const toolService = root.inject('tool');
95
+ let externalTools = [];
96
+ if (toolService) {
97
+ externalTools = toolService.collectAll(root);
98
+ externalTools = toolService.filterByContext(externalTools, toolContext);
99
+ }
100
+ logger.debug(`[AI Handler] 工具收集: ${externalTools.length} 个, ${(performance.now() - tCollect).toFixed(0)}ms`);
101
+ try {
102
+ const timeout = new Promise((_, rej) => setTimeout(() => rej(new Error('AI 响应超时')), triggerConfig.timeout));
103
+ let responseText;
104
+ if (refs.zhinAgent) {
105
+ const imageUrls = extractImageUrls(message);
106
+ let elements;
107
+ if (imageUrls.length > 0) {
108
+ const parts = [];
109
+ if (content)
110
+ parts.push({ type: 'text', text: content });
111
+ for (const url of imageUrls) {
112
+ parts.push({ type: 'image_url', image_url: { url } });
113
+ }
114
+ elements = await Promise.race([
115
+ refs.zhinAgent.processMultimodal(parts, toolContext),
116
+ timeout,
117
+ ]);
118
+ }
119
+ else {
120
+ elements = await Promise.race([
121
+ refs.zhinAgent.process(content, toolContext, externalTools),
122
+ timeout,
123
+ ]);
124
+ }
125
+ responseText = renderOutput(elements);
126
+ }
127
+ else {
128
+ const response = await Promise.race([
129
+ ai.process(content, toolContext, externalTools),
130
+ timeout,
131
+ ]);
132
+ responseText = typeof response === 'string' ? response : '';
133
+ }
134
+ if (responseText)
135
+ await message.$reply(parseRichMediaContent(responseText));
136
+ logger.info(`[AI Handler] 总耗时: ${(performance.now() - t0).toFixed(0)}ms`);
137
+ }
138
+ catch (error) {
139
+ const msg = error instanceof Error ? error.message : String(error);
140
+ logger.warn(`[AI Handler] 失败 (${(performance.now() - t0).toFixed(0)}ms): ${msg}`);
141
+ await message.$reply(triggerConfig.errorTemplate.replace('{error}', msg));
142
+ }
143
+ };
144
+ const dispatcher = root.inject('dispatcher');
145
+ if (dispatcher && typeof dispatcher.setAIHandler === 'function') {
146
+ dispatcher.setAITriggerMatcher((message) => shouldTriggerAI(message, triggerConfig));
147
+ dispatcher.setAIHandler(handleAIMessage);
148
+ logger.debug('AI Handler registered via MessageDispatcher');
149
+ return () => { logger.info('AI Handler unregistered'); };
150
+ }
151
+ const aiMw = async (message, next) => {
152
+ const { triggered, content } = shouldTriggerAI(message, triggerConfig);
153
+ if (!triggered)
154
+ return await next();
155
+ await handleAIMessage(message, content);
156
+ await next();
157
+ };
158
+ const dispose = root.addMiddleware(aiMw);
159
+ logger.debug('AI Trigger middleware registered (fallback mode)');
160
+ return () => { dispose(); };
161
+ });
162
+ }
163
+ //# sourceMappingURL=register-ai-trigger.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"register-ai-trigger.js","sourceRoot":"","sources":["../../src/init/register-ai-trigger.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,YAAY,CAAC;AACpB,OAAO,EAAE,SAAS,EAAW,eAAe,EAAE,sBAAsB,EAAE,qBAAqB,EAAE,oBAAoB,EAAE,MAAM,eAAe,CAAC;AAMzI,SAAS,gBAAgB,CAAC,OAAqB;IAC7C,MAAM,IAAI,GAAa,EAAE,CAAC;IAC1B,MAAM,GAAG,GAAG,OAAO,OAAO,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;IAEjG,MAAM,UAAU,GAAG,GAAG,CAAC,KAAK,CAAC,2BAA2B,CAAC,CAAC;IAC1D,IAAI,UAAU,EAAE,CAAC;QACf,KAAK,MAAM,CAAC,IAAI,UAAU,EAAE,CAAC;YAC3B,MAAM,QAAQ,GAAG,CAAC,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;YAC1C,IAAI,QAAQ;gBAAE,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;QACvC,CAAC;IACH,CAAC;IAED,MAAM,SAAS,GAAG,GAAG,CAAC,KAAK,CAAC,gCAAgC,CAAC,CAAC;IAC9D,IAAI,SAAS,EAAE,CAAC;QACd,KAAK,MAAM,CAAC,IAAI,SAAS,EAAE,CAAC;YAC1B,MAAM,QAAQ,GAAG,CAAC,CAAC,KAAK,CAAC,eAAe,CAAC,CAAC;YAC1C,IAAI,QAAQ;gBAAE,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;QACvC,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,YAAY,CAAC,QAAyB;IAC7C,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,KAAK,MAAM,EAAE,IAAI,QAAQ,EAAE,CAAC;QAC1B,QAAQ,EAAE,CAAC,IAAI,EAAE,CAAC;YAChB,KAAK,MAAM;gBACT,IAAI,EAAE,CAAC,OAAO;oBAAE,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC;gBACvC,MAAM;YACR,KAAK,OAAO;gBACV,KAAK,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC,GAAG,KAAK,CAAC,CAAC;gBACvC,MAAM;YACR,KAAK,OAAO;gBACV,KAAK,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC,GAAG,KAAK,CAAC,CAAC;gBACvC,MAAM;YACR,KAAK,OAAO;gBACV,KAAK,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC,GAAG,KAAK,CAAC,CAAC;gBACvC,MAAM;YACR,KAAK,MAAM,CAAC,CAAC,CAAC;gBACZ,MAAM,EAAE,GAAG,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC;gBAC9B,IAAI,EAAE,CAAC,WAAW;oBAAE,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC;gBAC5C,IAAI,EAAE,CAAC,MAAM,EAAE,MAAM;oBACnB,KAAK,MAAM,CAAC,IAAI,EAAE,CAAC,MAAM;wBAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,KAAK,KAAK,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC;gBACjE,IAAI,EAAE,CAAC,QAAQ;oBAAE,EAAE,CAAC,IAAI,CAAC,eAAe,EAAE,CAAC,QAAQ,KAAK,CAAC,CAAC;gBAC1D,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;gBAC1B,MAAM;YACR,CAAC;YACD,KAAK,MAAM;gBACT,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,IAAI,KAAK,EAAE,CAAC,GAAG,EAAE,CAAC,CAAC;gBACvC,MAAM;QACV,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;AAChC,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,IAAmB;IACnD,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAC3B,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC;IAE5C,UAAU,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,EAAE;QACtB,MAAM,SAAS,GAAG,EAAE,CAAC,gBAAgB,EAAE,CAAC;QACxC,MAAM,aAAa,GAAG,oBAAoB,CAAC,SAAS,CAAC,CAAC;QACtD,IAAI,CAAC,aAAa,CAAC,OAAO,EAAE,CAAC;YAC3B,MAAM,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;YACtC,OAAO;QACT,CAAC;QAED,MAAM,eAAe,GAAG,KAAK,EAC3B,OAAqB,EACrB,OAAe,EACf,EAAE;YACF,MAAM,EAAE,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;YAC7B,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE;gBAAE,OAAO;YAC1B,IAAI,aAAa,CAAC,eAAe;gBAC/B,MAAM,OAAO,CAAC,MAAM,CAAC,aAAa,CAAC,eAAe,CAAC,CAAC;YAEtD,MAAM,WAAW,GAAG,sBAAsB,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;YACnE,MAAM,WAAW,GAAgB;gBAC/B,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,KAAK,EAAE,OAAO,CAAC,IAAI;gBACnB,OAAO,EAAE,OAAO,CAAC,QAAQ,EAAE,EAAE,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE;gBACnD,QAAQ,EAAE,OAAO,CAAC,OAAO,CAAC,EAAE;gBAC5B,OAAO;gBACP,KAAK,EAAE,WAAW,CAAC,KAAK;gBACxB,qBAAqB,EAAE,WAAW,CAAC,eAAe;gBAClD,YAAY,EAAE,WAAW,CAAC,YAAY;gBACtC,YAAY,EAAE,WAAW,CAAC,YAAY;gBACtC,UAAU,EAAE,WAAW,CAAC,UAAU;gBAClC,OAAO,EAAE,WAAW,CAAC,OAAO;aAC7B,CAAC;YAEF,MAAM,QAAQ,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;YACnC,MAAM,WAAW,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YACxC,IAAI,aAAa,GAAW,EAAE,CAAC;YAC/B,IAAI,WAAW,EAAE,CAAC;gBAChB,aAAa,GAAG,WAAW,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;gBAC7C,aAAa,GAAG,WAAW,CAAC,eAAe,CAAC,aAAa,EAAE,WAAW,CAAC,CAAC;YAC1E,CAAC;YACD,MAAM,CAAC,KAAK,CAAC,sBAAsB,aAAa,CAAC,MAAM,OAAO,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,QAAQ,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YAE7G,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,IAAI,OAAO,CAAQ,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,CAC5C,UAAU,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,IAAI,KAAK,CAAC,SAAS,CAAC,CAAC,EAAE,aAAa,CAAC,OAAO,CAAC,CACnE,CAAC;gBAEF,IAAI,YAAoB,CAAC;gBACzB,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;oBACnB,MAAM,SAAS,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;oBAC5C,IAAI,QAAyB,CAAC;oBAC9B,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACzB,MAAM,KAAK,GAAkB,EAAE,CAAC;wBAChC,IAAI,OAAO;4BAAE,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;wBACzD,KAAK,MAAM,GAAG,IAAI,SAAS,EAAE,CAAC;4BAC5B,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,SAAS,EAAE,EAAE,GAAG,EAAE,EAAE,CAAC,CAAC;wBACxD,CAAC;wBACD,QAAQ,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC;4BAC5B,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC,KAAK,EAAE,WAAW,CAAC;4BACpD,OAAO;yBACR,CAAC,CAAC;oBACL,CAAC;yBAAM,CAAC;wBACN,QAAQ,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC;4BAC5B,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,OAAO,EAAE,WAAW,EAAE,aAAa,CAAC;4BAC3D,OAAO;yBACR,CAAC,CAAC;oBACL,CAAC;oBACD,YAAY,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;gBACxC,CAAC;qBAAM,CAAC;oBACN,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC;wBAClC,EAAE,CAAC,OAAO,CAAC,OAAO,EAAE,WAAW,EAAE,aAAa,CAAC;wBAC/C,OAAO;qBACR,CAAC,CAAC;oBACH,YAAY,GAAG,OAAO,QAAQ,KAAK,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC9D,CAAC;gBAED,IAAI,YAAY;oBAAE,MAAM,OAAO,CAAC,MAAM,CAAC,qBAAqB,CAAC,YAAY,CAAC,CAAC,CAAC;gBAC5E,MAAM,CAAC,IAAI,CAAC,qBAAqB,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;YAC5E,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,GAAG,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;gBACnE,MAAM,CAAC,IAAI,CAAC,oBAAoB,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,GAAG,EAAE,CAAC,CAAC;gBAClF,MAAM,OAAO,CAAC,MAAM,CAAC,aAAa,CAAC,aAAa,CAAC,OAAO,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC,CAAC;YAC5E,CAAC;QACH,CAAC,CAAC;QAEF,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAE7C,IAAI,UAAU,IAAI,OAAO,UAAU,CAAC,YAAY,KAAK,UAAU,EAAE,CAAC;YAChE,UAAU,CAAC,mBAAmB,CAAC,CAAC,OAAqB,EAAE,EAAE,CACvD,eAAe,CAAC,OAAO,EAAE,aAAa,CAAC,CACxC,CAAC;YACF,UAAU,CAAC,YAAY,CAAC,eAAe,CAAC,CAAC;YACzC,MAAM,CAAC,KAAK,CAAC,6CAA6C,CAAC,CAAC;YAC5D,OAAO,GAAG,EAAE,GAAG,MAAM,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC,CAAC,CAAC,CAAC;QAC3D,CAAC;QAED,MAAM,IAAI,GAAG,KAAK,EAChB,OAAqB,EACrB,IAAyB,EACzB,EAAE;YACF,MAAM,EAAE,SAAS,EAAE,OAAO,EAAE,GAAG,eAAe,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC;YACvE,IAAI,CAAC,SAAS;gBAAE,OAAO,MAAM,IAAI,EAAE,CAAC;YACpC,MAAM,eAAe,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YACxC,MAAM,IAAI,EAAE,CAAC;QACf,CAAC,CAAC;QACF,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,kDAAkD,CAAC,CAAC;QACjE,OAAO,GAAG,EAAE,GAAG,OAAO,EAAE,CAAC,CAAC,CAAC,CAAC;IAC9B,CAAC,CAAC,CAAC;AACL,CAAC"}
@@ -0,0 +1,3 @@
1
+ import type { AIServiceRefs } from './shared-refs.js';
2
+ export declare function registerBuiltinTools(refs: AIServiceRefs): void;
3
+ //# sourceMappingURL=register-builtin-tools.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"register-builtin-tools.d.ts","sourceRoot":"","sources":["../../src/init/register-builtin-tools.ts"],"names":[],"mappings":"AAaA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AAEtD,wBAAgB,oBAAoB,CAAC,IAAI,EAAE,aAAa,GAAG,IAAI,CAiK9D"}
@@ -0,0 +1,179 @@
1
+ /**
2
+ * Register built-in system tools (file/shell/web/schedule/memory/skill)
3
+ * and workspace skills with hot-reload support.
4
+ */
5
+ import * as fs from 'fs';
6
+ import * as os from 'os';
7
+ import * as path from 'path';
8
+ import { getPlugin } from '@zhin.js/core';
9
+ import { createBuiltinTools, discoverWorkspaceSkills, loadAlwaysSkillsContent, buildSkillsSummaryXML } from '../builtin-tools.js';
10
+ import { resolveSkillInstructionMaxChars, DEFAULT_CONFIG } from '../zhin-agent/config.js';
11
+ import { loadBootstrapFiles, buildContextFiles, buildBootstrapContextSection } from '../bootstrap.js';
12
+ import { triggerAIHook, createAIHookEvent } from '../hooks.js';
13
+ import { createCronTools } from '../cron-engine.js';
14
+ export function registerBuiltinTools(refs) {
15
+ const plugin = getPlugin();
16
+ const { useContext, root, logger } = plugin;
17
+ useContext('ai', 'tool', (ai, toolService) => {
18
+ if (!ai || !toolService)
19
+ return;
20
+ const provider = ai.getProvider();
21
+ const agentCfg = ai.getAgentConfig();
22
+ const fullCfg = { ...DEFAULT_CONFIG, ...agentCfg };
23
+ const modelName = provider.models[0] || '';
24
+ const builtinTools = createBuiltinTools({ skillInstructionMaxChars: resolveSkillInstructionMaxChars(fullCfg, modelName) });
25
+ const disposers = [];
26
+ for (const tool of builtinTools)
27
+ disposers.push(toolService.addTool(tool, root.name));
28
+ const cronTools = createCronTools();
29
+ for (const tool of cronTools)
30
+ disposers.push(toolService.addTool(tool, root.name));
31
+ logger.info(`Registered ${builtinTools.length} built-in + ${cronTools.length} cron tools`);
32
+ let skillWatchers = [];
33
+ let skillReloadDebounce = null;
34
+ async function syncWorkspaceSkills() {
35
+ const skillFeature = root.inject?.('skill');
36
+ if (!skillFeature)
37
+ return 0;
38
+ const existing = skillFeature.getByPlugin(root.name);
39
+ for (const s of existing)
40
+ skillFeature.remove(s);
41
+ const skills = await discoverWorkspaceSkills();
42
+ if (skills.length === 0)
43
+ return 0;
44
+ const allRegisteredTools = toolService.getAll();
45
+ const toolNameIndex = new Map();
46
+ for (const t of allRegisteredTools) {
47
+ toolNameIndex.set(t.name, t);
48
+ const parts = t.name.split('_');
49
+ if (parts.length === 2)
50
+ toolNameIndex.set(`${parts[1]}_${parts[0]}`, t);
51
+ }
52
+ for (const s of skills) {
53
+ const associatedTools = [];
54
+ const toolNames = s.toolNames || [];
55
+ for (const toolName of toolNames) {
56
+ let tool = toolService.get(toolName) || toolNameIndex.get(toolName);
57
+ if (tool)
58
+ associatedTools.push(tool);
59
+ }
60
+ skillFeature.add({
61
+ name: s.name,
62
+ description: s.description,
63
+ tools: associatedTools,
64
+ keywords: s.keywords || [],
65
+ tags: s.tags || [],
66
+ pluginName: root.name,
67
+ }, root.name);
68
+ }
69
+ return skills.length;
70
+ }
71
+ (async () => {
72
+ // Step 1: discover workspace skills
73
+ try {
74
+ const count = await syncWorkspaceSkills();
75
+ const skillFeature = root.inject?.('skill');
76
+ if (count > 0 && skillFeature) {
77
+ logger.info(`Registered ${count} workspace skills`);
78
+ }
79
+ }
80
+ catch (e) {
81
+ logger.warn(`Failed to discover workspace skills: ${e.message}`);
82
+ }
83
+ // Step 2: load bootstrap files
84
+ const loadedFiles = [];
85
+ try {
86
+ const workspaceDir = process.cwd();
87
+ const bootstrapFiles = await loadBootstrapFiles(workspaceDir);
88
+ const contextFiles = buildContextFiles(bootstrapFiles);
89
+ logger.debug(`Bootstrap files loaded (cwd: ${workspaceDir}): ${bootstrapFiles.map(f => f.name + (f.missing ? ' (missing)' : '')).join(', ')}`);
90
+ const soulFile = contextFiles.find(f => f.path === 'SOUL.md');
91
+ if (soulFile && refs.zhinAgent)
92
+ loadedFiles.push('SOUL.md');
93
+ const toolsFile = contextFiles.find(f => f.path === 'TOOLS.md');
94
+ if (toolsFile)
95
+ loadedFiles.push('TOOLS.md');
96
+ const agentsFile = contextFiles.find(f => f.path === 'AGENTS.md');
97
+ if (agentsFile)
98
+ loadedFiles.push('AGENTS.md');
99
+ if (loadedFiles.length > 0) {
100
+ logger.info(`Loaded bootstrap: ${loadedFiles.join(', ')} → agent prompt`);
101
+ }
102
+ if (refs.zhinAgent && contextFiles.length > 0) {
103
+ const contextSection = buildBootstrapContextSection(contextFiles);
104
+ refs.zhinAgent.setBootstrapContext(contextSection);
105
+ }
106
+ }
107
+ catch (e) {
108
+ logger.debug(`Bootstrap files not loaded: ${e.message}`);
109
+ }
110
+ // Step 3: inject always-on skills content + XML summary
111
+ try {
112
+ const skillsForContext = await discoverWorkspaceSkills();
113
+ const alwaysContent = await loadAlwaysSkillsContent(skillsForContext);
114
+ const skillsXml = buildSkillsSummaryXML(skillsForContext);
115
+ if (refs.zhinAgent) {
116
+ refs.zhinAgent.setActiveSkillsContext(alwaysContent);
117
+ refs.zhinAgent.setSkillsSummaryXML(skillsXml);
118
+ }
119
+ }
120
+ catch (e) {
121
+ logger.debug(`Skills context not set: ${e instanceof Error ? e.message : String(e)}`);
122
+ }
123
+ // Trigger agent:bootstrap hook
124
+ const skillFeature2 = root.inject('skill');
125
+ await triggerAIHook(createAIHookEvent('agent', 'bootstrap', undefined, {
126
+ workspaceDir: process.cwd(),
127
+ toolCount: builtinTools.length,
128
+ skillCount: skillFeature2?.size ?? 0,
129
+ bootstrapFiles: loadedFiles,
130
+ }));
131
+ // Hot-reload skill directories
132
+ const workspaceSkillDir = path.join(process.cwd(), 'skills');
133
+ const localSkillDir = path.join(os.homedir(), '.zhin', 'skills');
134
+ const onSkillDirChange = () => {
135
+ if (skillReloadDebounce)
136
+ clearTimeout(skillReloadDebounce);
137
+ skillReloadDebounce = setTimeout(async () => {
138
+ skillReloadDebounce = null;
139
+ try {
140
+ const count = await syncWorkspaceSkills();
141
+ const skillsForContext = await discoverWorkspaceSkills();
142
+ const alwaysContent = await loadAlwaysSkillsContent(skillsForContext);
143
+ const skillsXml = buildSkillsSummaryXML(skillsForContext);
144
+ if (refs.zhinAgent) {
145
+ refs.zhinAgent.setActiveSkillsContext(alwaysContent);
146
+ refs.zhinAgent.setSkillsSummaryXML(skillsXml);
147
+ }
148
+ await triggerAIHook(createAIHookEvent('agent', 'skills-reloaded', undefined, { skillCount: count }));
149
+ if (count >= 0)
150
+ logger.info(`[技能热重载] 已更新,当前工作区技能数: ${count}`);
151
+ }
152
+ catch (e) {
153
+ logger.warn(`[技能热重载] 失败: ${e.message}`);
154
+ }
155
+ }, 400);
156
+ };
157
+ for (const dir of [workspaceSkillDir, localSkillDir]) {
158
+ if (fs.existsSync(dir)) {
159
+ try {
160
+ const w = fs.watch(dir, { recursive: true }, onSkillDirChange);
161
+ skillWatchers.push(w);
162
+ logger.debug(`[技能热重载] 监听目录: ${dir}`);
163
+ }
164
+ catch (e) {
165
+ logger.debug(`[技能热重载] 无法监听 ${dir}: ${e.message}`);
166
+ }
167
+ }
168
+ }
169
+ })();
170
+ return () => {
171
+ disposers.forEach(d => d());
172
+ skillWatchers.forEach(w => w.close());
173
+ skillWatchers = [];
174
+ if (skillReloadDebounce)
175
+ clearTimeout(skillReloadDebounce);
176
+ };
177
+ });
178
+ }
179
+ //# sourceMappingURL=register-builtin-tools.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"register-builtin-tools.js","sourceRoot":"","sources":["../../src/init/register-builtin-tools.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,EAAE,SAAS,EAAgC,MAAM,eAAe,CAAC;AACxE,OAAO,EAAE,kBAAkB,EAAE,uBAAuB,EAAE,uBAAuB,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAClI,OAAO,EAAE,+BAA+B,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AAC1F,OAAO,EAAE,kBAAkB,EAAE,iBAAiB,EAAE,4BAA4B,EAAE,MAAM,iBAAiB,CAAC;AACtG,OAAO,EAAE,aAAa,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAC/D,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAGpD,MAAM,UAAU,oBAAoB,CAAC,IAAmB;IACtD,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAC3B,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC;IAE5C,UAAU,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,EAAE,WAAW,EAAE,EAAE;QAC3C,IAAI,CAAC,EAAE,IAAI,CAAC,WAAW;YAAE,OAAO;QAEhC,MAAM,QAAQ,GAAG,EAAE,CAAC,WAAW,EAAE,CAAC;QAClC,MAAM,QAAQ,GAAG,EAAE,CAAC,cAAc,EAAE,CAAC;QACrC,MAAM,OAAO,GAAG,EAAE,GAAG,cAAc,EAAE,GAAG,QAAQ,EAAiE,CAAC;QAClH,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAC3C,MAAM,YAAY,GAAG,kBAAkB,CAAC,EAAE,wBAAwB,EAAE,+BAA+B,CAAC,OAAO,EAAE,SAAS,CAAC,EAAE,CAAC,CAAC;QAC3H,MAAM,SAAS,GAAmB,EAAE,CAAC;QACrC,KAAK,MAAM,IAAI,IAAI,YAAY;YAAE,SAAS,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QACtF,MAAM,SAAS,GAAG,eAAe,EAAE,CAAC;QACpC,KAAK,MAAM,IAAI,IAAI,SAAS;YAAE,SAAS,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QACnF,MAAM,CAAC,IAAI,CAAC,cAAc,YAAY,CAAC,MAAM,eAAe,SAAS,CAAC,MAAM,aAAa,CAAC,CAAC;QAE3F,IAAI,aAAa,GAAmB,EAAE,CAAC;QACvC,IAAI,mBAAmB,GAAyC,IAAI,CAAC;QAErE,KAAK,UAAU,mBAAmB;YAChC,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,OAAO,CAA6B,CAAC;YACxE,IAAI,CAAC,YAAY;gBAAE,OAAO,CAAC,CAAC;YAC5B,MAAM,QAAQ,GAAG,YAAY,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACrD,KAAK,MAAM,CAAC,IAAI,QAAQ;gBAAE,YAAY,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YACjD,MAAM,MAAM,GAAG,MAAM,uBAAuB,EAAE,CAAC;YAC/C,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;gBAAE,OAAO,CAAC,CAAC;YAClC,MAAM,kBAAkB,GAAG,WAAW,CAAC,MAAM,EAAE,CAAC;YAChD,MAAM,aAAa,GAAG,IAAI,GAAG,EAAgB,CAAC;YAC9C,KAAK,MAAM,CAAC,IAAI,kBAAkB,EAAE,CAAC;gBACnC,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;gBAC7B,MAAM,KAAK,GAAG,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;gBAChC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;oBAAE,aAAa,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;YAC1E,CAAC;YACD,KAAK,MAAM,CAAC,IAAI,MAAM,EAAE,CAAC;gBACvB,MAAM,eAAe,GAAW,EAAE,CAAC;gBACnC,MAAM,SAAS,GAAG,CAAC,CAAC,SAAS,IAAI,EAAE,CAAC;gBACpC,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;oBACjC,IAAI,IAAI,GAAG,WAAW,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,aAAa,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;oBACpE,IAAI,IAAI;wBAAE,eAAe,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACvC,CAAC;gBACD,YAAY,CAAC,GAAG,CAAC;oBACf,IAAI,EAAE,CAAC,CAAC,IAAI;oBACZ,WAAW,EAAE,CAAC,CAAC,WAAW;oBAC1B,KAAK,EAAE,eAAe;oBACtB,QAAQ,EAAE,CAAC,CAAC,QAAQ,IAAI,EAAE;oBAC1B,IAAI,EAAE,CAAC,CAAC,IAAI,IAAI,EAAE;oBAClB,UAAU,EAAE,IAAI,CAAC,IAAI;iBACtB,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;YAChB,CAAC;YACD,OAAO,MAAM,CAAC,MAAM,CAAC;QACvB,CAAC;QAED,CAAC,KAAK,IAAI,EAAE;YACV,oCAAoC;YACpC,IAAI,CAAC;gBACH,MAAM,KAAK,GAAG,MAAM,mBAAmB,EAAE,CAAC;gBAC1C,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC,OAAO,CAA6B,CAAC;gBACxE,IAAI,KAAK,GAAG,CAAC,IAAI,YAAY,EAAE,CAAC;oBAC9B,MAAM,CAAC,IAAI,CAAC,cAAc,KAAK,mBAAmB,CAAC,CAAC;gBACtD,CAAC;YACH,CAAC;YAAC,OAAO,CAAM,EAAE,CAAC;gBAChB,MAAM,CAAC,IAAI,CAAC,wCAAwC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;YACnE,CAAC;YAED,+BAA+B;YAC/B,MAAM,WAAW,GAAa,EAAE,CAAC;YACjC,IAAI,CAAC;gBACH,MAAM,YAAY,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC;gBACnC,MAAM,cAAc,GAAG,MAAM,kBAAkB,CAAC,YAAY,CAAC,CAAC;gBAC9D,MAAM,YAAY,GAAG,iBAAiB,CAAC,cAAc,CAAC,CAAC;gBAEvD,MAAM,CAAC,KAAK,CAAC,gCAAgC,YAAY,MAAM,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAE/I,MAAM,QAAQ,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC;gBAC9D,IAAI,QAAQ,IAAI,IAAI,CAAC,SAAS;oBAAE,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBAE5D,MAAM,SAAS,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;gBAChE,IAAI,SAAS;oBAAE,WAAW,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;gBAE5C,MAAM,UAAU,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC;gBAClE,IAAI,UAAU;oBAAE,WAAW,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;gBAE9C,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC3B,MAAM,CAAC,IAAI,CAAC,qBAAqB,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;gBAC5E,CAAC;gBAED,IAAI,IAAI,CAAC,SAAS,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC9C,MAAM,cAAc,GAAG,4BAA4B,CAAC,YAAY,CAAC,CAAC;oBAClE,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC,cAAc,CAAC,CAAC;gBACrD,CAAC;YACH,CAAC;YAAC,OAAO,CAAM,EAAE,CAAC;gBAChB,MAAM,CAAC,KAAK,CAAC,+BAA+B,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;YAC3D,CAAC;YAED,wDAAwD;YACxD,IAAI,CAAC;gBACH,MAAM,gBAAgB,GAAG,MAAM,uBAAuB,EAAE,CAAC;gBACzD,MAAM,aAAa,GAAG,MAAM,uBAAuB,CAAC,gBAAgB,CAAC,CAAC;gBACtE,MAAM,SAAS,GAAG,qBAAqB,CAAC,gBAAgB,CAAC,CAAC;gBAC1D,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;oBACnB,IAAI,CAAC,SAAS,CAAC,sBAAsB,CAAC,aAAa,CAAC,CAAC;oBACrD,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC;gBAChD,CAAC;YACH,CAAC;YAAC,OAAO,CAAU,EAAE,CAAC;gBACpB,MAAM,CAAC,KAAK,CAAC,2BAA2B,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;YACxF,CAAC;YAED,+BAA+B;YAC/B,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,CAA6B,CAAC;YACvE,MAAM,aAAa,CAAC,iBAAiB,CAAC,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE;gBACrE,YAAY,EAAE,OAAO,CAAC,GAAG,EAAE;gBAC3B,SAAS,EAAE,YAAY,CAAC,MAAM;gBAC9B,UAAU,EAAE,aAAa,EAAE,IAAI,IAAI,CAAC;gBACpC,cAAc,EAAE,WAAW;aAC5B,CAAC,CAAC,CAAC;YAEJ,+BAA+B;YAC/B,MAAM,iBAAiB,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,QAAQ,CAAC,CAAC;YAC7D,MAAM,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,OAAO,EAAE,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC;YACjE,MAAM,gBAAgB,GAAG,GAAG,EAAE;gBAC5B,IAAI,mBAAmB;oBAAE,YAAY,CAAC,mBAAmB,CAAC,CAAC;gBAC3D,mBAAmB,GAAG,UAAU,CAAC,KAAK,IAAI,EAAE;oBAC1C,mBAAmB,GAAG,IAAI,CAAC;oBAC3B,IAAI,CAAC;wBACH,MAAM,KAAK,GAAG,MAAM,mBAAmB,EAAE,CAAC;wBAC1C,MAAM,gBAAgB,GAAG,MAAM,uBAAuB,EAAE,CAAC;wBACzD,MAAM,aAAa,GAAG,MAAM,uBAAuB,CAAC,gBAAgB,CAAC,CAAC;wBACtE,MAAM,SAAS,GAAG,qBAAqB,CAAC,gBAAgB,CAAC,CAAC;wBAC1D,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;4BACnB,IAAI,CAAC,SAAS,CAAC,sBAAsB,CAAC,aAAa,CAAC,CAAC;4BACrD,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC;wBAChD,CAAC;wBACD,MAAM,aAAa,CAAC,iBAAiB,CAAC,OAAO,EAAE,iBAAiB,EAAE,SAAS,EAAE,EAAE,UAAU,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;wBACrG,IAAI,KAAK,IAAI,CAAC;4BAAE,MAAM,CAAC,IAAI,CAAC,yBAAyB,KAAK,EAAE,CAAC,CAAC;oBAChE,CAAC;oBAAC,OAAO,CAAM,EAAE,CAAC;wBAChB,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;oBAC1C,CAAC;gBACH,CAAC,EAAE,GAAG,CAAC,CAAC;YACV,CAAC,CAAC;YACF,KAAK,MAAM,GAAG,IAAI,CAAC,iBAAiB,EAAE,aAAa,CAAC,EAAE,CAAC;gBACrD,IAAI,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;oBACvB,IAAI,CAAC;wBACH,MAAM,CAAC,GAAG,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,EAAE,gBAAgB,CAAC,CAAC;wBAC/D,aAAa,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;wBACtB,MAAM,CAAC,KAAK,CAAC,iBAAiB,GAAG,EAAE,CAAC,CAAC;oBACvC,CAAC;oBAAC,OAAO,CAAM,EAAE,CAAC;wBAChB,MAAM,CAAC,KAAK,CAAC,gBAAgB,GAAG,KAAK,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;oBACpD,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC,CAAC,EAAE,CAAC;QAEL,OAAO,GAAG,EAAE;YACV,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;YAC5B,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC;YACtC,aAAa,GAAG,EAAE,CAAC;YACnB,IAAI,mBAAmB;gBAAE,YAAY,CAAC,mBAAmB,CAAC,CAAC;QAC7D,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;AACL,CAAC"}
@@ -0,0 +1,2 @@
1
+ export declare function registerDbModels(): void;
2
+ //# sourceMappingURL=register-db-models.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"register-db-models.d.ts","sourceRoot":"","sources":["../../src/init/register-db-models.ts"],"names":[],"mappings":"AAUA,wBAAgB,gBAAgB,IAAI,IAAI,CAoBvC"}
@@ -0,0 +1,28 @@
1
+ /**
2
+ * Define AI-related database models (7 tables).
3
+ */
4
+ import { getPlugin } from '@zhin.js/core';
5
+ import { AI_SESSION_MODEL } from '../session.js';
6
+ import { CHAT_MESSAGE_MODEL, CONTEXT_SUMMARY_MODEL } from '../context-manager.js';
7
+ import { AI_MESSAGE_MODEL, AI_SUMMARY_MODEL } from '../conversation-memory.js';
8
+ import { AI_USER_PROFILE_MODEL } from '../user-profile.js';
9
+ import { AI_FOLLOWUP_MODEL } from '../follow-up.js';
10
+ export function registerDbModels() {
11
+ const plugin = getPlugin();
12
+ const { logger } = plugin;
13
+ const defineModel = plugin.defineModel;
14
+ if (typeof defineModel === 'function') {
15
+ defineModel('chat_messages', CHAT_MESSAGE_MODEL);
16
+ defineModel('context_summaries', CONTEXT_SUMMARY_MODEL);
17
+ defineModel('ai_sessions', AI_SESSION_MODEL);
18
+ defineModel('ai_messages', AI_MESSAGE_MODEL);
19
+ defineModel('ai_summaries', AI_SUMMARY_MODEL);
20
+ defineModel('ai_user_profiles', AI_USER_PROFILE_MODEL);
21
+ defineModel('ai_followups', AI_FOLLOWUP_MODEL);
22
+ logger.debug('AI database models registered (7 tables)');
23
+ }
24
+ else {
25
+ logger.debug('defineModel not available, AI will use in-memory storage');
26
+ }
27
+ }
28
+ //# sourceMappingURL=register-db-models.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"register-db-models.js","sourceRoot":"","sources":["../../src/init/register-db-models.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAC1C,OAAO,EAAE,gBAAgB,EAAE,MAAM,eAAe,CAAC;AACjD,OAAO,EAAE,kBAAkB,EAAE,qBAAqB,EAAE,MAAM,uBAAuB,CAAC;AAClF,OAAO,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAC/E,OAAO,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,EAAE,iBAAiB,EAAE,MAAM,iBAAiB,CAAC;AAEpD,MAAM,UAAU,gBAAgB;IAC9B,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAC3B,MAAM,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC;IAE1B,MAAM,WAAW,GAAI,MAA6C,CAAC,WAEtD,CAAC;IAEd,IAAI,OAAO,WAAW,KAAK,UAAU,EAAE,CAAC;QACtC,WAAW,CAAC,eAAe,EAAE,kBAAkB,CAAC,CAAC;QACjD,WAAW,CAAC,mBAAmB,EAAE,qBAAqB,CAAC,CAAC;QACxD,WAAW,CAAC,aAAa,EAAE,gBAAgB,CAAC,CAAC;QAC7C,WAAW,CAAC,aAAa,EAAE,gBAAgB,CAAC,CAAC;QAC7C,WAAW,CAAC,cAAc,EAAE,gBAAgB,CAAC,CAAC;QAC9C,WAAW,CAAC,kBAAkB,EAAE,qBAAqB,CAAC,CAAC;QACvD,WAAW,CAAC,cAAc,EAAE,iBAAiB,CAAC,CAAC;QAC/C,MAAM,CAAC,KAAK,CAAC,0CAA0C,CAAC,CAAC;IAC3D,CAAC;SAAM,CAAC;QACN,MAAM,CAAC,KAAK,CAAC,0DAA0D,CAAC,CAAC;IAC3E,CAAC;AACH,CAAC"}
@@ -0,0 +1,10 @@
1
+ /**
2
+ * Upgrade AI components from in-memory to database storage when DB becomes ready.
3
+ *
4
+ * FIX: Replaced the previous `setTimeout(100)` race condition with
5
+ * a proper `useContext('database', 'ai', ...)` dual-dependency wait.
6
+ */
7
+ import './types.js';
8
+ import type { AIServiceRefs } from './shared-refs.js';
9
+ export declare function registerDbUpgrade(refs: AIServiceRefs): void;
10
+ //# sourceMappingURL=register-db-upgrade.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"register-db-upgrade.d.ts","sourceRoot":"","sources":["../../src/init/register-db-upgrade.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AACH,OAAO,YAAY,CAAC;AAKpB,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AAEtD,wBAAgB,iBAAiB,CAAC,IAAI,EAAE,aAAa,GAAG,IAAI,CA+D3D"}
@@ -0,0 +1,67 @@
1
+ /**
2
+ * Upgrade AI components from in-memory to database storage when DB becomes ready.
3
+ *
4
+ * FIX: Replaced the previous `setTimeout(100)` race condition with
5
+ * a proper `useContext('database', 'ai', ...)` dual-dependency wait.
6
+ */
7
+ import './types.js';
8
+ import { getPlugin } from '@zhin.js/core';
9
+ import { createDatabaseSessionManager } from '../session.js';
10
+ import { createContextManager } from '../context-manager.js';
11
+ export function registerDbUpgrade(refs) {
12
+ const plugin = getPlugin();
13
+ const { useContext, root, logger } = plugin;
14
+ useContext('database', 'ai', async (db, _ai) => {
15
+ const aiService = refs.aiService;
16
+ if (!aiService)
17
+ return;
18
+ const configService = root.inject('config');
19
+ const appConfig = configService?.getPrimary() || {};
20
+ const config = appConfig.ai || {};
21
+ if (config.sessions?.useDatabase === false)
22
+ return;
23
+ try {
24
+ const model = db.models?.get('ai_sessions');
25
+ if (!model)
26
+ return;
27
+ const dbSession = createDatabaseSessionManager(model, aiService.getSessionConfig());
28
+ aiService.setSessionManager(dbSession);
29
+ if (refs.zhinAgent)
30
+ refs.zhinAgent.setSessionManager(dbSession);
31
+ const ctxCfg = aiService.getContextConfig();
32
+ if (ctxCfg.enabled !== false) {
33
+ const msgModel = db.models.get('chat_messages');
34
+ const sumModel = db.models.get('context_summaries');
35
+ if (msgModel && sumModel) {
36
+ const ctxMgr = createContextManager(msgModel, sumModel, ctxCfg);
37
+ aiService.setContextManager(ctxMgr);
38
+ if (refs.zhinAgent)
39
+ refs.zhinAgent.setContextManager(ctxMgr);
40
+ }
41
+ }
42
+ if (refs.zhinAgent) {
43
+ const aiMsgModel = db.models.get('ai_messages');
44
+ const aiSumModel = db.models.get('ai_summaries');
45
+ if (aiMsgModel && aiSumModel) {
46
+ refs.zhinAgent.upgradeMemoryToDatabase(aiMsgModel, aiSumModel);
47
+ }
48
+ const profileModel = db.models.get('ai_user_profiles');
49
+ if (profileModel) {
50
+ refs.zhinAgent.upgradeProfilesToDatabase(profileModel);
51
+ }
52
+ const followUpModel = db.models.get('ai_followups');
53
+ if (followUpModel) {
54
+ refs.zhinAgent.upgradeFollowUpsToDatabase(followUpModel);
55
+ refs.zhinAgent.restoreFollowUps().catch(e => {
56
+ logger.warn('FollowUp restore failed:', e);
57
+ });
58
+ }
59
+ }
60
+ logger.debug('AI database storage activated (session, memory, profile, followup)');
61
+ }
62
+ catch (e) {
63
+ logger.error('AI Session: database setup failed:', e);
64
+ }
65
+ });
66
+ }
67
+ //# sourceMappingURL=register-db-upgrade.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"register-db-upgrade.js","sourceRoot":"","sources":["../../src/init/register-db-upgrade.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AACH,OAAO,YAAY,CAAC;AACpB,OAAO,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAE1C,OAAO,EAAE,4BAA4B,EAAE,MAAM,eAAe,CAAC;AAC7D,OAAO,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAG7D,MAAM,UAAU,iBAAiB,CAAC,IAAmB;IACnD,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAC3B,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC;IAE5C,UAAU,CAAC,UAAU,EAAE,IAAI,EAAE,KAAK,EAAE,EAAO,EAAE,GAAG,EAAE,EAAE;QAClD,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC;QACjC,IAAI,CAAC,SAAS;YAAE,OAAO;QAEvB,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC5C,MAAM,SAAS,GACb,aAAa,EAAE,UAAU,EAAqB,IAAI,EAAE,CAAC;QACvD,MAAM,MAAM,GAAG,SAAS,CAAC,EAAE,IAAI,EAAE,CAAC;QAElC,IAAI,MAAM,CAAC,QAAQ,EAAE,WAAW,KAAK,KAAK;YAAE,OAAO;QAEnD,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,EAAE,CAAC,MAAM,EAAE,GAAG,CAAC,aAAa,CAAC,CAAC;YAC5C,IAAI,CAAC,KAAK;gBAAE,OAAO;YAEnB,MAAM,SAAS,GAAG,4BAA4B,CAC5C,KAAK,EACL,SAAS,CAAC,gBAAgB,EAAE,CAC7B,CAAC;YACF,SAAS,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YACvC,IAAI,IAAI,CAAC,SAAS;gBAAE,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;YAEhE,MAAM,MAAM,GAAG,SAAS,CAAC,gBAAgB,EAAE,CAAC;YAC5C,IAAI,MAAM,CAAC,OAAO,KAAK,KAAK,EAAE,CAAC;gBAC7B,MAAM,QAAQ,GAAG,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,eAAe,CAAC,CAAC;gBAChD,MAAM,QAAQ,GAAG,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC;gBACpD,IAAI,QAAQ,IAAI,QAAQ,EAAE,CAAC;oBACzB,MAAM,MAAM,GAAG,oBAAoB,CAAC,QAAQ,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;oBAChE,SAAS,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;oBACpC,IAAI,IAAI,CAAC,SAAS;wBAAE,IAAI,CAAC,SAAS,CAAC,iBAAiB,CAAC,MAAM,CAAC,CAAC;gBAC/D,CAAC;YACH,CAAC;YAED,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;gBACnB,MAAM,UAAU,GAAG,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;gBAChD,MAAM,UAAU,GAAG,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;gBACjD,IAAI,UAAU,IAAI,UAAU,EAAE,CAAC;oBAC7B,IAAI,CAAC,SAAS,CAAC,uBAAuB,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;gBACjE,CAAC;gBAED,MAAM,YAAY,GAAG,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;gBACvD,IAAI,YAAY,EAAE,CAAC;oBACjB,IAAI,CAAC,SAAS,CAAC,yBAAyB,CAAC,YAAY,CAAC,CAAC;gBACzD,CAAC;gBAED,MAAM,aAAa,GAAG,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC;gBACpD,IAAI,aAAa,EAAE,CAAC;oBAClB,IAAI,CAAC,SAAS,CAAC,0BAA0B,CAAC,aAAa,CAAC,CAAC;oBACzD,IAAI,CAAC,SAAS,CAAC,gBAAgB,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE;wBAC1C,MAAM,CAAC,IAAI,CAAC,0BAA0B,EAAE,CAAC,CAAC,CAAC;oBAC7C,CAAC,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YAED,MAAM,CAAC,KAAK,CAAC,oEAAoE,CAAC,CAAC;QACrF,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,MAAM,CAAC,KAAK,CAAC,oCAAoC,EAAE,CAAC,CAAC,CAAC;QACxD,CAAC;IACH,CAAC,CAAC,CAAC;AACL,CAAC"}
@@ -0,0 +1,6 @@
1
+ /**
2
+ * Register AI management tools (ai.models, ai.clear, ai.health).
3
+ */
4
+ import './types.js';
5
+ export declare function registerManagementTools(): void;
6
+ //# sourceMappingURL=register-management-tools.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"register-management-tools.d.ts","sourceRoot":"","sources":["../../src/init/register-management-tools.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,YAAY,CAAC;AAIpB,wBAAgB,uBAAuB,IAAI,IAAI,CA+D9C"}
@@ -0,0 +1,68 @@
1
+ /**
2
+ * Register AI management tools (ai.models, ai.clear, ai.health).
3
+ */
4
+ import './types.js';
5
+ import { getPlugin, ZhinTool } from '@zhin.js/core';
6
+ import { SessionManager } from '../session.js';
7
+ export function registerManagementTools() {
8
+ const plugin = getPlugin();
9
+ const { useContext, root, logger } = plugin;
10
+ useContext('ai', 'tool', (ai, toolService) => {
11
+ if (!ai || !toolService)
12
+ return;
13
+ const listModelsTool = new ZhinTool('ai.models')
14
+ .desc('列出所有可用的 AI 模型')
15
+ .keyword('模型', '可用模型', 'ai模型', 'model', 'models')
16
+ .tag('ai', 'management')
17
+ .execute(async () => {
18
+ const models = await ai.listModels();
19
+ return { providers: models.map(({ provider, models: ml }) => ({ name: provider, models: ml.slice(0, 10), total: ml.length })) };
20
+ })
21
+ .action(async () => {
22
+ const models = await ai.listModels();
23
+ let r = '🤖 可用模型:\n';
24
+ for (const { provider, models: ml } of models) {
25
+ r += `\n【${provider}】\n` + ml.slice(0, 5).map(m => ` • ${m}`).join('\n');
26
+ if (ml.length > 5)
27
+ r += `\n ... 还有 ${ml.length - 5} 个`;
28
+ }
29
+ return r;
30
+ });
31
+ const clearSessionTool = new ZhinTool('ai.clear')
32
+ .desc('清除当前对话的历史记录')
33
+ .keyword('清除', '清空', '重置', 'clear', 'reset')
34
+ .tag('ai', 'session')
35
+ .execute(async (_args, context) => {
36
+ if (!context?.message)
37
+ return { success: false, error: '无消息上下文' };
38
+ const msg = context.message;
39
+ const sid = SessionManager.generateId(msg.$adapter, msg.$sender.id, msg.$channel?.id);
40
+ await ai.sessions.reset(sid);
41
+ return { success: true, message: '对话历史已清除' };
42
+ })
43
+ .action(async (message) => {
44
+ const sid = SessionManager.generateId(message.$adapter, message.$sender.id, message.$channel?.id);
45
+ await ai.sessions.reset(sid);
46
+ return '✅ 对话历史已清除';
47
+ });
48
+ const healthCheckTool = new ZhinTool('ai.health')
49
+ .desc('检查 AI 服务的健康状态')
50
+ .keyword('健康', '状态', '检查', 'health', 'status')
51
+ .tag('ai', 'management')
52
+ .execute(async () => {
53
+ const h = await ai.healthCheck();
54
+ return { providers: Object.entries(h).map(([n, ok]) => ({ name: n, healthy: ok })) };
55
+ })
56
+ .action(async () => {
57
+ const h = await ai.healthCheck();
58
+ return ['🏥 AI 服务健康状态:'].concat(Object.entries(h).map(([p, ok]) => ` ${ok ? '✅' : '❌'} ${p}`)).join('\n');
59
+ });
60
+ const tools = [listModelsTool, clearSessionTool, healthCheckTool];
61
+ const disposers = [];
62
+ for (const tool of tools)
63
+ disposers.push(toolService.addTool(tool, root.name));
64
+ logger.debug(`Registered ${tools.length} AI management tools`);
65
+ return () => disposers.forEach(d => d());
66
+ });
67
+ }
68
+ //# sourceMappingURL=register-management-tools.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"register-management-tools.js","sourceRoot":"","sources":["../../src/init/register-management-tools.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,YAAY,CAAC;AACpB,OAAO,EAAE,SAAS,EAAW,QAAQ,EAAE,MAAM,eAAe,CAAC;AAC7D,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAE/C,MAAM,UAAU,uBAAuB;IACrC,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAC3B,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,EAAE,GAAG,MAAM,CAAC;IAE5C,UAAU,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,EAAE,WAAW,EAAE,EAAE;QAC3C,IAAI,CAAC,EAAE,IAAI,CAAC,WAAW;YAAE,OAAO;QAEhC,MAAM,cAAc,GAAG,IAAI,QAAQ,CAAC,WAAW,CAAC;aAC7C,IAAI,CAAC,eAAe,CAAC;aACrB,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,QAAQ,CAAC;aAChD,GAAG,CAAC,IAAI,EAAE,YAAY,CAAC;aACvB,OAAO,CAAC,KAAK,IAAI,EAAE;YAClB,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,UAAU,EAAE,CAAC;YACrC,OAAO,EAAE,SAAS,EAAE,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC,EAAE,CAAC;QAClI,CAAC,CAAC;aACD,MAAM,CAAC,KAAK,IAAI,EAAE;YACjB,MAAM,MAAM,GAAG,MAAM,EAAE,CAAC,UAAU,EAAE,CAAC;YACrC,IAAI,CAAC,GAAG,YAAY,CAAC;YACrB,KAAK,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,IAAI,MAAM,EAAE,CAAC;gBAC9C,CAAC,IAAI,MAAM,QAAQ,KAAK,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAC1E,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC;oBAAE,CAAC,IAAI,cAAc,EAAE,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC;YAC1D,CAAC;YACD,OAAO,CAAC,CAAC;QACX,CAAC,CAAC,CAAC;QAEL,MAAM,gBAAgB,GAAG,IAAI,QAAQ,CAAC,UAAU,CAAC;aAC9C,IAAI,CAAC,aAAa,CAAC;aACnB,OAAO,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,CAAC;aAC3C,GAAG,CAAC,IAAI,EAAE,SAAS,CAAC;aACpB,OAAO,CAAC,KAAK,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YAChC,IAAI,CAAC,OAAO,EAAE,OAAO;gBAAE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC;YAClE,MAAM,GAAG,GAAG,OAAO,CAAC,OAAkB,CAAC;YACvC,MAAM,GAAG,GAAG,cAAc,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,OAAO,CAAC,EAAE,EAAE,GAAG,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;YACtF,MAAM,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAC7B,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,CAAC;QAC/C,CAAC,CAAC;aACD,MAAM,CAAC,KAAK,EAAE,OAAgB,EAAE,EAAE;YACjC,MAAM,GAAG,GAAG,cAAc,CAAC,UAAU,CAAC,OAAO,CAAC,QAAQ,EAAE,OAAO,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;YAClG,MAAM,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YAC7B,OAAO,WAAW,CAAC;QACrB,CAAC,CAAC,CAAC;QAEL,MAAM,eAAe,GAAG,IAAI,QAAQ,CAAC,WAAW,CAAC;aAC9C,IAAI,CAAC,eAAe,CAAC;aACrB,OAAO,CAAC,IAAI,EAAE,IAAI,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,CAAC;aAC7C,GAAG,CAAC,IAAI,EAAE,YAAY,CAAC;aACvB,OAAO,CAAC,KAAK,IAAI,EAAE;YAClB,MAAM,CAAC,GAAG,MAAM,EAAE,CAAC,WAAW,EAAE,CAAC;YACjC,OAAO,EAAE,SAAS,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC;QACvF,CAAC,CAAC;aACD,MAAM,CAAC,KAAK,IAAI,EAAE;YACjB,MAAM,CAAC,GAAG,MAAM,EAAE,CAAC,WAAW,EAAE,CAAC;YACjC,OAAO,CAAC,eAAe,CAAC,CAAC,MAAM,CAC7B,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,IAAI,CAAC,EAAE,CAAC,CAC/D,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACf,CAAC,CAAC,CAAC;QAEL,MAAM,KAAK,GAAG,CAAC,cAAc,EAAE,gBAAgB,EAAE,eAAe,CAAC,CAAC;QAClE,MAAM,SAAS,GAAmB,EAAE,CAAC;QACrC,KAAK,MAAM,IAAI,IAAI,KAAK;YAAE,SAAS,CAAC,IAAI,CAAC,WAAW,CAAC,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QAC/E,MAAM,CAAC,KAAK,CAAC,cAAc,KAAK,CAAC,MAAM,sBAAsB,CAAC,CAAC;QAC/D,OAAO,GAAG,EAAE,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;AACL,CAAC"}
@@ -0,0 +1,3 @@
1
+ import type { AIServiceRefs } from './shared-refs.js';
2
+ export declare function registerMessageRecorder(refs: AIServiceRefs): void;
3
+ //# sourceMappingURL=register-message-recorder.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"register-message-recorder.d.ts","sourceRoot":"","sources":["../../src/init/register-message-recorder.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AAEtD,wBAAgB,uBAAuB,CAAC,IAAI,EAAE,aAAa,GAAG,IAAI,CAuBjE"}
@@ -0,0 +1,27 @@
1
+ /**
2
+ * Register middleware that records messages to the context manager.
3
+ */
4
+ import { getPlugin } from '@zhin.js/core';
5
+ export function registerMessageRecorder(refs) {
6
+ const plugin = getPlugin();
7
+ const { root } = plugin;
8
+ root.addMiddleware(async (message, next) => {
9
+ await next();
10
+ if (refs.aiService?.contextManager) {
11
+ const record = {
12
+ platform: message.$adapter,
13
+ scene_id: message.$channel?.id || message.$sender.id,
14
+ scene_type: message.$channel?.type || 'private',
15
+ scene_name: message.$channel?.name || '',
16
+ sender_id: message.$sender.id,
17
+ sender_name: message.$sender.name || message.$sender.id,
18
+ message: typeof message.$raw === 'string'
19
+ ? message.$raw
20
+ : JSON.stringify(message.$raw),
21
+ time: message.$timestamp || Date.now(),
22
+ };
23
+ refs.aiService.contextManager.recordMessage(record).catch(() => { });
24
+ }
25
+ });
26
+ }
27
+ //# sourceMappingURL=register-message-recorder.js.map