@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.
- package/CHANGELOG.md +8 -0
- package/README.md +226 -0
- package/lib/agent.d.ts +9 -7
- package/lib/agent.d.ts.map +1 -1
- package/lib/agent.js +63 -29
- package/lib/agent.js.map +1 -1
- package/lib/bootstrap.js +1 -1
- package/lib/bootstrap.js.map +1 -1
- package/lib/builtin-tools.d.ts.map +1 -1
- package/lib/builtin-tools.js +34 -31
- package/lib/builtin-tools.js.map +1 -1
- package/lib/compaction.js +1 -1
- package/lib/compaction.js.map +1 -1
- package/lib/context-manager.js +1 -1
- package/lib/context-manager.js.map +1 -1
- package/lib/conversation-memory.d.ts +11 -0
- package/lib/conversation-memory.d.ts.map +1 -1
- package/lib/conversation-memory.js +39 -1
- package/lib/conversation-memory.js.map +1 -1
- package/lib/cron-engine.d.ts.map +1 -1
- package/lib/cron-engine.js +5 -3
- package/lib/cron-engine.js.map +1 -1
- package/lib/follow-up.js +1 -1
- package/lib/follow-up.js.map +1 -1
- package/lib/hooks.js +1 -1
- package/lib/hooks.js.map +1 -1
- package/lib/index.d.ts +2 -0
- package/lib/index.d.ts.map +1 -1
- package/lib/index.js +1 -0
- package/lib/index.js.map +1 -1
- package/lib/init/create-zhin-agent.d.ts +3 -0
- package/lib/init/create-zhin-agent.d.ts.map +1 -0
- package/lib/init/create-zhin-agent.js +127 -0
- package/lib/init/create-zhin-agent.js.map +1 -0
- package/lib/init/register-ai-service.d.ts +7 -0
- package/lib/init/register-ai-service.d.ts.map +1 -0
- package/lib/init/register-ai-service.js +42 -0
- package/lib/init/register-ai-service.js.map +1 -0
- package/lib/init/register-ai-trigger.d.ts +7 -0
- package/lib/init/register-ai-trigger.d.ts.map +1 -0
- package/lib/init/register-ai-trigger.js +163 -0
- package/lib/init/register-ai-trigger.js.map +1 -0
- package/lib/init/register-builtin-tools.d.ts +3 -0
- package/lib/init/register-builtin-tools.d.ts.map +1 -0
- package/lib/init/register-builtin-tools.js +179 -0
- package/lib/init/register-builtin-tools.js.map +1 -0
- package/lib/init/register-db-models.d.ts +2 -0
- package/lib/init/register-db-models.d.ts.map +1 -0
- package/lib/init/register-db-models.js +28 -0
- package/lib/init/register-db-models.js.map +1 -0
- package/lib/init/register-db-upgrade.d.ts +10 -0
- package/lib/init/register-db-upgrade.d.ts.map +1 -0
- package/lib/init/register-db-upgrade.js +67 -0
- package/lib/init/register-db-upgrade.js.map +1 -0
- package/lib/init/register-management-tools.d.ts +6 -0
- package/lib/init/register-management-tools.d.ts.map +1 -0
- package/lib/init/register-management-tools.js +68 -0
- package/lib/init/register-management-tools.js.map +1 -0
- package/lib/init/register-message-recorder.d.ts +3 -0
- package/lib/init/register-message-recorder.d.ts.map +1 -0
- package/lib/init/register-message-recorder.js +27 -0
- package/lib/init/register-message-recorder.js.map +1 -0
- package/lib/init/register-tool-service.d.ts +2 -0
- package/lib/init/register-tool-service.d.ts.map +1 -0
- package/lib/init/register-tool-service.js +9 -0
- package/lib/init/register-tool-service.js.map +1 -0
- package/lib/init/shared-refs.d.ts +14 -0
- package/lib/init/shared-refs.d.ts.map +1 -0
- package/lib/init/shared-refs.js +7 -0
- package/lib/init/shared-refs.js.map +1 -0
- package/lib/init/types.d.ts +15 -0
- package/lib/init/types.d.ts.map +1 -0
- package/lib/init/types.js +7 -0
- package/lib/init/types.js.map +1 -0
- package/lib/init.d.ts +14 -17
- package/lib/init.d.ts.map +1 -1
- package/lib/init.js +34 -670
- package/lib/init.js.map +1 -1
- package/lib/rate-limiter.js +1 -1
- package/lib/rate-limiter.js.map +1 -1
- package/lib/service.js +2 -2
- package/lib/service.js.map +1 -1
- package/lib/session.d.ts +7 -0
- package/lib/session.d.ts.map +1 -1
- package/lib/session.js +26 -4
- package/lib/session.js.map +1 -1
- package/lib/storage.d.ts +68 -0
- package/lib/storage.d.ts.map +1 -0
- package/lib/storage.js +105 -0
- package/lib/storage.js.map +1 -0
- package/lib/subagent.js +1 -1
- package/lib/subagent.js.map +1 -1
- package/lib/tools.d.ts.map +1 -1
- package/lib/tools.js +8 -9
- package/lib/tools.js.map +1 -1
- package/lib/user-profile.js +1 -1
- package/lib/user-profile.js.map +1 -1
- package/lib/zhin-agent/builtin-tools.js.map +1 -1
- package/lib/zhin-agent/config.d.ts +1 -1
- package/lib/zhin-agent/config.d.ts.map +1 -1
- package/lib/zhin-agent/config.js.map +1 -1
- package/lib/zhin-agent/index.js +1 -1
- package/lib/zhin-agent/index.js.map +1 -1
- package/lib/zhin-agent/tool-collector.js +3 -3
- package/lib/zhin-agent/tool-collector.js.map +1 -1
- package/package.json +4 -3
- package/src/agent.ts +53 -30
- package/src/bootstrap.ts +1 -1
- package/src/builtin-tools.ts +44 -40
- package/src/compaction.ts +1 -1
- package/src/context-manager.ts +1 -1
- package/src/conversation-memory.ts +43 -1
- package/src/cron-engine.ts +5 -3
- package/src/follow-up.ts +1 -1
- package/src/hooks.ts +1 -1
- package/src/index.ts +10 -0
- package/src/init/create-zhin-agent.ts +133 -0
- package/src/init/register-ai-service.ts +53 -0
- package/src/init/register-ai-trigger.ts +179 -0
- package/src/init/register-builtin-tools.ts +177 -0
- package/src/init/register-db-models.ts +31 -0
- package/src/init/register-db-upgrade.ts +77 -0
- package/src/init/register-management-tools.ts +71 -0
- package/src/init/register-message-recorder.ts +31 -0
- package/src/init/register-tool-service.ts +9 -0
- package/src/init/shared-refs.ts +20 -0
- package/src/init/types.ts +18 -0
- package/src/init.ts +35 -735
- package/src/rate-limiter.ts +1 -1
- package/src/service.ts +4 -4
- package/src/session.ts +26 -4
- package/src/storage.ts +135 -0
- package/src/subagent.ts +1 -1
- package/src/tools.ts +7 -10
- package/src/user-profile.ts +1 -1
- package/src/zhin-agent/builtin-tools.ts +2 -2
- package/src/zhin-agent/config.ts +3 -3
- package/src/zhin-agent/index.ts +1 -1
- 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 @@
|
|
|
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 @@
|
|
|
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 @@
|
|
|
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 @@
|
|
|
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
|