@zhin.js/core 1.0.37 → 1.0.38
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 +9 -0
- package/README.md +57 -3
- package/lib/adapter.d.ts +11 -0
- package/lib/adapter.d.ts.map +1 -1
- package/lib/adapter.js +61 -0
- package/lib/adapter.js.map +1 -1
- package/lib/ai/index.d.ts +3 -39
- package/lib/ai/index.d.ts.map +1 -1
- package/lib/ai/index.js +2 -44
- package/lib/ai/index.js.map +1 -1
- package/lib/ai/types.d.ts +4 -3
- package/lib/ai/types.d.ts.map +1 -1
- package/lib/built/ai-trigger.js.map +1 -1
- package/lib/built/common-adapter-tools.d.ts +55 -0
- package/lib/built/common-adapter-tools.d.ts.map +1 -0
- package/lib/built/common-adapter-tools.js +158 -0
- package/lib/built/common-adapter-tools.js.map +1 -0
- package/lib/built/dispatcher.d.ts.map +1 -1
- package/lib/built/dispatcher.js +50 -46
- package/lib/built/dispatcher.js.map +1 -1
- package/lib/built/skill.d.ts.map +1 -1
- package/lib/built/skill.js +0 -1
- package/lib/built/skill.js.map +1 -1
- package/lib/built/tool.d.ts +3 -3
- package/lib/built/tool.d.ts.map +1 -1
- package/lib/built/tool.js.map +1 -1
- package/lib/feature.d.ts +16 -1
- package/lib/feature.d.ts.map +1 -1
- package/lib/feature.js +41 -2
- package/lib/feature.js.map +1 -1
- package/lib/index.d.ts +1 -0
- package/lib/index.d.ts.map +1 -1
- package/lib/index.js +2 -0
- package/lib/index.js.map +1 -1
- package/lib/plugin.d.ts +38 -1
- package/lib/plugin.d.ts.map +1 -1
- package/lib/plugin.js +73 -22
- package/lib/plugin.js.map +1 -1
- package/lib/scheduler/scheduler.js +1 -1
- package/lib/scheduler/scheduler.js.map +1 -1
- package/lib/types.d.ts +43 -28
- package/lib/types.d.ts.map +1 -1
- package/lib/utils.d.ts +12 -3
- package/lib/utils.d.ts.map +1 -1
- package/lib/utils.js +64 -54
- package/lib/utils.js.map +1 -1
- package/package.json +5 -5
- package/src/adapter.ts +85 -5
- package/src/ai/index.ts +8 -186
- package/src/ai/types.ts +5 -4
- package/src/built/ai-trigger.ts +2 -2
- package/src/built/common-adapter-tools.ts +207 -0
- package/src/built/dispatcher.ts +51 -52
- package/src/built/skill.ts +3 -4
- package/src/built/tool.ts +3 -3
- package/src/feature.ts +45 -2
- package/src/index.ts +2 -0
- package/src/plugin.ts +92 -31
- package/src/scheduler/scheduler.ts +1 -1
- package/src/types.ts +39 -28
- package/src/utils.ts +63 -52
- package/tests/ai/setup.ts +2 -2
- package/tests/utils.test.ts +1 -3
- package/lib/ai/agent.d.ts +0 -130
- package/lib/ai/agent.d.ts.map +0 -1
- package/lib/ai/agent.js +0 -702
- package/lib/ai/agent.js.map +0 -1
- package/lib/ai/bootstrap.d.ts +0 -91
- package/lib/ai/bootstrap.d.ts.map +0 -1
- package/lib/ai/bootstrap.js +0 -243
- package/lib/ai/bootstrap.js.map +0 -1
- package/lib/ai/builtin-tools.d.ts +0 -59
- package/lib/ai/builtin-tools.d.ts.map +0 -1
- package/lib/ai/builtin-tools.js +0 -777
- package/lib/ai/builtin-tools.js.map +0 -1
- package/lib/ai/compaction.d.ts +0 -132
- package/lib/ai/compaction.d.ts.map +0 -1
- package/lib/ai/compaction.js +0 -370
- package/lib/ai/compaction.js.map +0 -1
- package/lib/ai/context-manager.d.ts +0 -213
- package/lib/ai/context-manager.d.ts.map +0 -1
- package/lib/ai/context-manager.js +0 -313
- package/lib/ai/context-manager.js.map +0 -1
- package/lib/ai/conversation-memory.d.ts +0 -181
- package/lib/ai/conversation-memory.d.ts.map +0 -1
- package/lib/ai/conversation-memory.js +0 -581
- package/lib/ai/conversation-memory.js.map +0 -1
- package/lib/ai/cron-engine.d.ts +0 -92
- package/lib/ai/cron-engine.d.ts.map +0 -1
- package/lib/ai/cron-engine.js +0 -278
- package/lib/ai/cron-engine.js.map +0 -1
- package/lib/ai/follow-up.d.ts +0 -131
- package/lib/ai/follow-up.d.ts.map +0 -1
- package/lib/ai/follow-up.js +0 -265
- package/lib/ai/follow-up.js.map +0 -1
- package/lib/ai/hooks.d.ts +0 -143
- package/lib/ai/hooks.d.ts.map +0 -1
- package/lib/ai/hooks.js +0 -108
- package/lib/ai/hooks.js.map +0 -1
- package/lib/ai/init.d.ts +0 -30
- package/lib/ai/init.d.ts.map +0 -1
- package/lib/ai/init.js +0 -686
- package/lib/ai/init.js.map +0 -1
- package/lib/ai/output.d.ts +0 -93
- package/lib/ai/output.d.ts.map +0 -1
- package/lib/ai/output.js +0 -176
- package/lib/ai/output.js.map +0 -1
- package/lib/ai/rate-limiter.d.ts +0 -38
- package/lib/ai/rate-limiter.d.ts.map +0 -1
- package/lib/ai/rate-limiter.js +0 -86
- package/lib/ai/rate-limiter.js.map +0 -1
- package/lib/ai/service.d.ts +0 -88
- package/lib/ai/service.d.ts.map +0 -1
- package/lib/ai/service.js +0 -285
- package/lib/ai/service.js.map +0 -1
- package/lib/ai/session.d.ts +0 -186
- package/lib/ai/session.d.ts.map +0 -1
- package/lib/ai/session.js +0 -443
- package/lib/ai/session.js.map +0 -1
- package/lib/ai/subagent.d.ts +0 -50
- package/lib/ai/subagent.d.ts.map +0 -1
- package/lib/ai/subagent.js +0 -144
- package/lib/ai/subagent.js.map +0 -1
- package/lib/ai/tone-detector.d.ts +0 -19
- package/lib/ai/tone-detector.d.ts.map +0 -1
- package/lib/ai/tone-detector.js +0 -72
- package/lib/ai/tone-detector.js.map +0 -1
- package/lib/ai/tools.d.ts +0 -45
- package/lib/ai/tools.d.ts.map +0 -1
- package/lib/ai/tools.js +0 -206
- package/lib/ai/tools.js.map +0 -1
- package/lib/ai/user-profile.d.ts +0 -56
- package/lib/ai/user-profile.d.ts.map +0 -1
- package/lib/ai/user-profile.js +0 -130
- package/lib/ai/user-profile.js.map +0 -1
- package/lib/ai/zhin-agent/builtin-tools.d.ts +0 -17
- package/lib/ai/zhin-agent/builtin-tools.d.ts.map +0 -1
- package/lib/ai/zhin-agent/builtin-tools.js +0 -220
- package/lib/ai/zhin-agent/builtin-tools.js.map +0 -1
- package/lib/ai/zhin-agent/config.d.ts +0 -54
- package/lib/ai/zhin-agent/config.d.ts.map +0 -1
- package/lib/ai/zhin-agent/config.js +0 -76
- package/lib/ai/zhin-agent/config.js.map +0 -1
- package/lib/ai/zhin-agent/exec-policy.d.ts +0 -20
- package/lib/ai/zhin-agent/exec-policy.d.ts.map +0 -1
- package/lib/ai/zhin-agent/exec-policy.js +0 -71
- package/lib/ai/zhin-agent/exec-policy.js.map +0 -1
- package/lib/ai/zhin-agent/index.d.ts +0 -70
- package/lib/ai/zhin-agent/index.d.ts.map +0 -1
- package/lib/ai/zhin-agent/index.js +0 -404
- package/lib/ai/zhin-agent/index.js.map +0 -1
- package/lib/ai/zhin-agent/prompt.d.ts +0 -21
- package/lib/ai/zhin-agent/prompt.d.ts.map +0 -1
- package/lib/ai/zhin-agent/prompt.js +0 -111
- package/lib/ai/zhin-agent/prompt.js.map +0 -1
- package/lib/ai/zhin-agent/tool-collector.d.ts +0 -22
- package/lib/ai/zhin-agent/tool-collector.d.ts.map +0 -1
- package/lib/ai/zhin-agent/tool-collector.js +0 -218
- package/lib/ai/zhin-agent/tool-collector.js.map +0 -1
- package/src/ai/agent.ts +0 -831
- package/src/ai/bootstrap.ts +0 -309
- package/src/ai/builtin-tools.ts +0 -849
- package/src/ai/compaction.ts +0 -529
- package/src/ai/context-manager.ts +0 -440
- package/src/ai/conversation-memory.ts +0 -774
- package/src/ai/cron-engine.ts +0 -337
- package/src/ai/follow-up.ts +0 -357
- package/src/ai/hooks.ts +0 -223
- package/src/ai/init.ts +0 -762
- package/src/ai/output.ts +0 -261
- package/src/ai/rate-limiter.ts +0 -129
- package/src/ai/service.ts +0 -331
- package/src/ai/session.ts +0 -544
- package/src/ai/subagent.ts +0 -209
- package/src/ai/tone-detector.ts +0 -89
- package/src/ai/tools.ts +0 -218
- package/src/ai/user-profile.ts +0 -181
- package/src/ai/zhin-agent/builtin-tools.ts +0 -247
- package/src/ai/zhin-agent/config.ts +0 -113
- package/src/ai/zhin-agent/exec-policy.ts +0 -78
- package/src/ai/zhin-agent/index.ts +0 -512
- package/src/ai/zhin-agent/prompt.ts +0 -131
- package/src/ai/zhin-agent/tool-collector.ts +0 -243
- package/tests/ai/agent.test.ts +0 -614
- package/tests/ai/context-manager.test.ts +0 -413
- package/tests/ai/conversation-memory.test.ts +0 -128
- package/tests/ai/follow-up.test.ts +0 -175
- package/tests/ai/integration.test.ts +0 -584
- package/tests/ai/output.test.ts +0 -128
- package/tests/ai/rate-limiter.test.ts +0 -108
- package/tests/ai/session.test.ts +0 -375
- package/tests/ai/subagent.test.ts +0 -270
- package/tests/ai/tone-detector.test.ts +0 -80
- package/tests/ai/tools-builtin.test.ts +0 -346
- package/tests/ai/user-profile.test.ts +0 -73
- package/tests/ai/zhin-agent.test.ts +0 -177
|
@@ -1,73 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* UserProfileStore 测试
|
|
3
|
-
*/
|
|
4
|
-
import { describe, it, expect, beforeEach } from 'vitest';
|
|
5
|
-
import { UserProfileStore } from '../../src/ai/user-profile.js';
|
|
6
|
-
|
|
7
|
-
describe('UserProfileStore(内存模式)', () => {
|
|
8
|
-
let store: UserProfileStore;
|
|
9
|
-
|
|
10
|
-
beforeEach(() => {
|
|
11
|
-
store = new UserProfileStore();
|
|
12
|
-
});
|
|
13
|
-
|
|
14
|
-
it('初始 get 应返回 null', async () => {
|
|
15
|
-
expect(await store.get('u1', 'name')).toBeNull();
|
|
16
|
-
});
|
|
17
|
-
|
|
18
|
-
it('set 后应能 get', async () => {
|
|
19
|
-
await store.set('u1', 'name', '小明');
|
|
20
|
-
expect(await store.get('u1', 'name')).toBe('小明');
|
|
21
|
-
});
|
|
22
|
-
|
|
23
|
-
it('getAll 应返回所有键值', async () => {
|
|
24
|
-
await store.set('u1', 'name', '小明');
|
|
25
|
-
await store.set('u1', 'style', '简洁');
|
|
26
|
-
const all = await store.getAll('u1');
|
|
27
|
-
expect(all).toEqual({ name: '小明', style: '简洁' });
|
|
28
|
-
});
|
|
29
|
-
|
|
30
|
-
it('getAll 不存在的用户应返回空对象', async () => {
|
|
31
|
-
expect(await store.getAll('nonexistent')).toEqual({});
|
|
32
|
-
});
|
|
33
|
-
|
|
34
|
-
it('delete 已有键应返回 true', async () => {
|
|
35
|
-
await store.set('u1', 'name', '小明');
|
|
36
|
-
const result = await store.delete('u1', 'name');
|
|
37
|
-
expect(result).toBe(true);
|
|
38
|
-
expect(await store.get('u1', 'name')).toBeNull();
|
|
39
|
-
});
|
|
40
|
-
|
|
41
|
-
it('delete 不存在的键应返回 false', async () => {
|
|
42
|
-
expect(await store.delete('u1', 'nonexistent')).toBe(false);
|
|
43
|
-
});
|
|
44
|
-
|
|
45
|
-
it('不同用户数据应隔离', async () => {
|
|
46
|
-
await store.set('u1', 'name', '小明');
|
|
47
|
-
await store.set('u2', 'name', '小红');
|
|
48
|
-
expect(await store.get('u1', 'name')).toBe('小明');
|
|
49
|
-
expect(await store.get('u2', 'name')).toBe('小红');
|
|
50
|
-
});
|
|
51
|
-
|
|
52
|
-
describe('buildProfileSummary', () => {
|
|
53
|
-
it('无数据应返回空字符串', async () => {
|
|
54
|
-
expect(await store.buildProfileSummary('u1')).toBe('');
|
|
55
|
-
});
|
|
56
|
-
|
|
57
|
-
it('有数据应生成摘要', async () => {
|
|
58
|
-
await store.set('u1', 'name', '小明');
|
|
59
|
-
await store.set('u1', 'interests', '编程');
|
|
60
|
-
const summary = await store.buildProfileSummary('u1');
|
|
61
|
-
expect(summary).toContain('[用户画像]');
|
|
62
|
-
expect(summary).toContain('name: 小明');
|
|
63
|
-
expect(summary).toContain('interests: 编程');
|
|
64
|
-
});
|
|
65
|
-
});
|
|
66
|
-
|
|
67
|
-
it('dispose 应清理数据', async () => {
|
|
68
|
-
await store.set('u1', 'name', '小明');
|
|
69
|
-
store.dispose();
|
|
70
|
-
// dispose 后重新获取,内存实现应清空
|
|
71
|
-
expect(await store.get('u1', 'name')).toBeNull();
|
|
72
|
-
});
|
|
73
|
-
});
|
|
@@ -1,177 +0,0 @@
|
|
|
1
|
-
/**
|
|
2
|
-
* ZhinAgent 补全测试
|
|
3
|
-
*
|
|
4
|
-
* 测试 collectTools 逻辑、handleMessage 端到端流程、会话管理等
|
|
5
|
-
*/
|
|
6
|
-
import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
|
|
7
|
-
import { ZhinAgent } from '../../src/ai/zhin-agent/index.js';
|
|
8
|
-
import { SkillFeature } from '../../src/built/skill.js';
|
|
9
|
-
import type { AIProvider, ChatResponse, AgentTool } from '../../src/ai/types.js';
|
|
10
|
-
import type { Tool, ToolContext } from '../../src/types.js';
|
|
11
|
-
|
|
12
|
-
// Mock AIProvider
|
|
13
|
-
function createMockProvider(response: string = '你好!'): AIProvider {
|
|
14
|
-
return {
|
|
15
|
-
name: 'mock',
|
|
16
|
-
models: ['mock-model'],
|
|
17
|
-
chat: vi.fn(async () => ({
|
|
18
|
-
choices: [{ message: { role: 'assistant' as const, content: response }, finish_reason: 'stop' }],
|
|
19
|
-
} as ChatResponse)),
|
|
20
|
-
listModels: vi.fn(async () => ['mock-model']),
|
|
21
|
-
};
|
|
22
|
-
}
|
|
23
|
-
|
|
24
|
-
function makeToolContext(overrides: Partial<ToolContext> = {}): ToolContext {
|
|
25
|
-
return {
|
|
26
|
-
platform: 'test',
|
|
27
|
-
botId: 'bot1',
|
|
28
|
-
sceneId: 'scene1',
|
|
29
|
-
senderId: 'user1',
|
|
30
|
-
...overrides,
|
|
31
|
-
};
|
|
32
|
-
}
|
|
33
|
-
|
|
34
|
-
function makeTool(name: string, desc: string = '', opts: Partial<Tool> = {}): Tool {
|
|
35
|
-
return {
|
|
36
|
-
name,
|
|
37
|
-
description: desc,
|
|
38
|
-
parameters: { type: 'object', properties: {} },
|
|
39
|
-
execute: vi.fn(async () => `result of ${name}`),
|
|
40
|
-
...opts,
|
|
41
|
-
};
|
|
42
|
-
}
|
|
43
|
-
|
|
44
|
-
describe('ZhinAgent', () => {
|
|
45
|
-
let agent: ZhinAgent;
|
|
46
|
-
let provider: AIProvider;
|
|
47
|
-
|
|
48
|
-
beforeEach(() => {
|
|
49
|
-
provider = createMockProvider();
|
|
50
|
-
agent = new ZhinAgent(provider, {
|
|
51
|
-
persona: '测试助手',
|
|
52
|
-
maxIterations: 3,
|
|
53
|
-
});
|
|
54
|
-
});
|
|
55
|
-
|
|
56
|
-
afterEach(() => {
|
|
57
|
-
agent.dispose();
|
|
58
|
-
});
|
|
59
|
-
|
|
60
|
-
describe('构造', () => {
|
|
61
|
-
it('应正确初始化', () => {
|
|
62
|
-
expect(agent).toBeDefined();
|
|
63
|
-
});
|
|
64
|
-
});
|
|
65
|
-
|
|
66
|
-
describe('依赖注入', () => {
|
|
67
|
-
it('setSkillRegistry 应正常工作', () => {
|
|
68
|
-
const registry = new SkillFeature();
|
|
69
|
-
expect(() => agent.setSkillRegistry(registry)).not.toThrow();
|
|
70
|
-
});
|
|
71
|
-
|
|
72
|
-
it('registerTool 应添加和移除工具', () => {
|
|
73
|
-
const tool: AgentTool = {
|
|
74
|
-
name: 'test_tool',
|
|
75
|
-
description: '测试',
|
|
76
|
-
parameters: { type: 'object', properties: {} },
|
|
77
|
-
execute: async () => 'ok',
|
|
78
|
-
};
|
|
79
|
-
|
|
80
|
-
const dispose = agent.registerTool(tool);
|
|
81
|
-
expect(typeof dispose).toBe('function');
|
|
82
|
-
|
|
83
|
-
// 移除
|
|
84
|
-
dispose();
|
|
85
|
-
});
|
|
86
|
-
});
|
|
87
|
-
|
|
88
|
-
describe('process', () => {
|
|
89
|
-
it('应处理简单文本消息并返回 OutputElement[]', async () => {
|
|
90
|
-
const context = makeToolContext();
|
|
91
|
-
|
|
92
|
-
const result = await agent.process(
|
|
93
|
-
'你好',
|
|
94
|
-
context,
|
|
95
|
-
[],
|
|
96
|
-
);
|
|
97
|
-
|
|
98
|
-
expect(result).toBeDefined();
|
|
99
|
-
expect(Array.isArray(result)).toBe(true);
|
|
100
|
-
expect(result.length).toBeGreaterThan(0);
|
|
101
|
-
expect(provider.chat).toHaveBeenCalled();
|
|
102
|
-
});
|
|
103
|
-
|
|
104
|
-
it('应传递工具列表', async () => {
|
|
105
|
-
const tools: Tool[] = [makeTool('get_time', '获取时间')];
|
|
106
|
-
const context = makeToolContext();
|
|
107
|
-
|
|
108
|
-
await agent.process('现在几点', context, tools);
|
|
109
|
-
|
|
110
|
-
// provider.chat 应被调用
|
|
111
|
-
expect(provider.chat).toHaveBeenCalled();
|
|
112
|
-
});
|
|
113
|
-
|
|
114
|
-
it('速率限制应生效', async () => {
|
|
115
|
-
// 创建一个严格限制的 agent
|
|
116
|
-
const strictAgent = new ZhinAgent(provider, {
|
|
117
|
-
rateLimit: { maxRequestsPerMinute: 1, cooldownSeconds: 5 },
|
|
118
|
-
});
|
|
119
|
-
|
|
120
|
-
const context = makeToolContext();
|
|
121
|
-
|
|
122
|
-
// 第一次请求
|
|
123
|
-
await strictAgent.process('hello', context, []);
|
|
124
|
-
|
|
125
|
-
// 第二次应被限制
|
|
126
|
-
const result = await strictAgent.process('hello again', context, []);
|
|
127
|
-
|
|
128
|
-
// 被限制时应返回友好提示(OutputElement[])
|
|
129
|
-
expect(result).toBeDefined();
|
|
130
|
-
expect(Array.isArray(result)).toBe(true);
|
|
131
|
-
|
|
132
|
-
strictAgent.dispose();
|
|
133
|
-
});
|
|
134
|
-
});
|
|
135
|
-
|
|
136
|
-
describe('collectTools 去重', () => {
|
|
137
|
-
it('应优先使用 Skill 中的工具', async () => {
|
|
138
|
-
const registry = new SkillFeature();
|
|
139
|
-
|
|
140
|
-
// 注册一个 Skill 包含 tool_a
|
|
141
|
-
registry.add({
|
|
142
|
-
name: 'skill1',
|
|
143
|
-
description: '技能1',
|
|
144
|
-
tools: [makeTool('tool_a', '来自 skill 的工具', { keywords: ['天气'] })],
|
|
145
|
-
keywords: ['天气'],
|
|
146
|
-
pluginName: 'p1',
|
|
147
|
-
}, 'p1');
|
|
148
|
-
|
|
149
|
-
agent.setSkillRegistry(registry);
|
|
150
|
-
|
|
151
|
-
// 外部也传入 tool_a(同名)
|
|
152
|
-
const externalTools = [makeTool('tool_a', '来自外部的工具')];
|
|
153
|
-
const context = makeToolContext();
|
|
154
|
-
|
|
155
|
-
// 调用 process,两个同名工具应该只保留一个
|
|
156
|
-
await agent.process('查看天气', context, externalTools);
|
|
157
|
-
|
|
158
|
-
// provider.chat 应被调用(正常处理)
|
|
159
|
-
expect(provider.chat).toHaveBeenCalled();
|
|
160
|
-
});
|
|
161
|
-
});
|
|
162
|
-
|
|
163
|
-
describe('getUserProfiles', () => {
|
|
164
|
-
it('应返回 UserProfileStore 实例', () => {
|
|
165
|
-
const profiles = agent.getUserProfiles();
|
|
166
|
-
expect(profiles).toBeDefined();
|
|
167
|
-
expect(typeof profiles.get).toBe('function');
|
|
168
|
-
expect(typeof profiles.set).toBe('function');
|
|
169
|
-
});
|
|
170
|
-
});
|
|
171
|
-
|
|
172
|
-
describe('dispose', () => {
|
|
173
|
-
it('应正常清理资源', () => {
|
|
174
|
-
expect(() => agent.dispose()).not.toThrow();
|
|
175
|
-
});
|
|
176
|
-
});
|
|
177
|
-
});
|