@zhin.js/core 1.0.24 → 1.0.26

package/tests/ai/integration.test.ts

@@ -0,0 +1,584 @@
+/**
+ * AI 模块集成测试
+ * 
+ * 完整测试环境，包括：
+ * 1. AI 服务初始化
+ * 2. 工具服务功能
+ * 3. AI 触发中间件
+ * 4. 内置工具
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+
+// Mock Logger first
+vi.mock('@zhin.js/core', async (importOriginal) => {
+  const original = await importOriginal() as any;
+  return {
+    ...original,
+    defineModel: vi.fn(), // Mock defineModel
+    getPlugin: vi.fn(() => ({
+      name: 'test-plugin',
+      root: { inject: vi.fn() },
+      logger: { debug: vi.fn(), info: vi.fn(), warn: vi.fn(), error: vi.fn() },
+      onDispose: vi.fn(),
+      collectAllTools: vi.fn(() => []),
+    })),
+    usePlugin: vi.fn(() => ({
+      name: 'test-plugin',
+      root: { 
+        inject: vi.fn(),
+        addMiddleware: vi.fn(),
+      },
+      provide: vi.fn(),
+      useContext: vi.fn(),
+      addMiddleware: vi.fn(),
+      defineModel: vi.fn(),
+      logger: { debug: vi.fn(), info: vi.fn(), warn: vi.fn(), error: vi.fn() },
+    })),
+    Logger: class {
+      debug = vi.fn();
+      info = vi.fn();
+      warn = vi.fn();
+      error = vi.fn();
+    },
+    segment: {
+      toString: (elements: any[]) => {
+        if (!Array.isArray(elements)) return String(elements);
+        return elements.map(el => {
+          if (typeof el === 'string') return el;
+          if (el.type === 'text') return el.data?.text || '';
+          if (el.type === 'at') return `<at user_id="${el.data?.user_id || el.data?.qq}"/>`;
+          if (el.type === 'image') return `<image url="${el.data?.url}"/>`;
+          return '';
+        }).join('');
+      },
+      from: (str: string) => {
+        if (!str) return [];
+        return [{ type: 'text', data: { text: str } }];
+      },
+    },
+  };
+});
+
+// Import after mocking
+import { AIService } from '../../src/ai/index.js';
+import { ToolFeature, ZhinTool, shouldTriggerAI, inferSenderPermissions } from '@zhin.js/core';
+import { calculatorTool, timeTool, getBuiltinTools } from '../../src/ai/tools.js';
+import type { Tool, ToolContext } from '@zhin.js/core';
+import type { AgentTool } from '../../src/ai/types.js';
+
+// ============================================================================
+// AI Service 测试
+// ============================================================================
+
+describe('AI Service 集成测试', () => {
+  let aiService: AIService;
+
+  beforeEach(() => {
+    vi.clearAllMocks();
+    aiService = new AIService({
+      defaultProvider: 'mock',
+      sessions: { maxHistory: 10 },
+    });
+  });
+
+  afterEach(() => {
+    aiService?.dispose();
+  });
+
+  describe('服务初始化', () => {
+    it('应该创建 AI 服务实例', () => {
+      expect(aiService).toBeDefined();
+      expect(aiService.sessions).toBeDefined();
+    });
+
+    it('没有配置提供商时 isReady 应返回 false', () => {
+      expect(aiService.isReady()).toBe(false);
+    });
+
+    it('应该返回空的提供商列表', () => {
+      expect(aiService.listProviders()).toEqual([]);
+    });
+
+    it('获取不存在的提供商应抛出错误', () => {
+      expect(() => aiService.getProvider('nonexistent')).toThrow();
+    });
+
+    it('应该根据配置初始化所有 Provider', () => {
+      const fullService = new AIService({
+        providers: {
+          openai: { apiKey: 'sk-test' },
+          anthropic: { apiKey: 'sk-ant-test' },
+          deepseek: { apiKey: 'sk-deepseek' },
+          moonshot: { apiKey: 'sk-moonshot' },
+          zhipu: { apiKey: 'sk-zhipu' },
+          ollama: { baseUrl: 'http://localhost:11434' },
+        },
+      });
+
+      const providers = fullService.listProviders();
+      expect(providers).toContain('openai');
+      expect(providers).toContain('anthropic');
+      expect(providers).toContain('deepseek');
+      expect(providers).toContain('moonshot');
+      expect(providers).toContain('zhipu');
+      expect(providers).toContain('ollama');
+      expect(providers).toHaveLength(6);
+
+      fullService.dispose();
+    });
+
+    it('应该只初始化有 apiKey 的 Provider', () => {
+      const partialService = new AIService({
+        providers: {
+          openai: { apiKey: 'sk-test' },
+          deepseek: {}, // 没有 apiKey
+          moonshot: { apiKey: 'sk-moonshot' },
+        },
+      });
+
+      const providers = partialService.listProviders();
+      expect(providers).toContain('openai');
+      expect(providers).toContain('moonshot');
+      expect(providers).not.toContain('deepseek');
+      expect(providers).toHaveLength(2);
+
+      partialService.dispose();
+    });
+  });
+
+  describe('配置管理', () => {
+    it('应该返回会话配置', () => {
+      const config = aiService.getSessionConfig();
+      expect(config.maxHistory).toBe(10);
+    });
+
+    it('应该返回上下文配置', () => {
+      const config = aiService.getContextConfig();
+      expect(config).toBeDefined();
+    });
+
+    it('应该返回触发器配置', () => {
+      const config = aiService.getTriggerConfig();
+      expect(config).toBeDefined();
+    });
+  });
+
+  describe('工具管理', () => {
+    it('应该收集内置工具', () => {
+      const tools = aiService.collectAllTools();
+      expect(Array.isArray(tools)).toBe(true);
+      const names = tools.map(t => t.name);
+      expect(names).toContain('calculator');
+      expect(names).toContain('get_time');
+    });
+
+    it('应该注册自定义工具', () => {
+      const customTool: AgentTool = {
+        name: 'custom_tool',
+        description: '自定义工具',
+        parameters: { type: 'object', properties: {} },
+        execute: async () => 'result',
+      };
+
+      const dispose = aiService.registerTool(customTool);
+      
+      const tools = aiService.collectAllTools();
+      expect(tools.some(t => t.name === 'custom_tool')).toBe(true);
+      
+      dispose();
+      const toolsAfter = aiService.collectAllTools();
+      expect(toolsAfter.some(t => t.name === 'custom_tool')).toBe(false);
+    });
+  });
+
+  describe('dispose', () => {
+    it('应该正确清理资源', () => {
+      aiService.dispose();
+      expect(aiService.listProviders()).toEqual([]);
+    });
+  });
+});
+
+// ============================================================================
+// Tool Service 测试
+// ============================================================================
+
+describe('Tool Service 集成测试', () => {
+  let service: ToolFeature;
+
+  beforeEach(() => {
+    vi.clearAllMocks();
+    service = new ToolFeature();
+  });
+
+  describe('Context 创建', () => {
+    it('应该创建正确的 Context', () => {
+      expect(service.name).toBe('tool');
+      expect(service.desc).toBeDefined();
+      expect(service.icon).toBe('Wrench');
+    });
+  });
+
+  describe('工具注册', () => {
+    it('应该注册 Tool 对象', () => {
+      const tool: Tool = {
+        name: 'test_tool',
+        description: '测试工具',
+        parameters: { type: 'object', properties: {} },
+        execute: async () => 'result',
+      };
+
+      const dispose = service.addTool(tool, 'test-plugin', false);
+      
+      expect(service.get('test_tool')).toBeDefined();
+      expect(service.getAll()).toHaveLength(1);
+      
+      dispose();
+      expect(service.get('test_tool')).toBeUndefined();
+    });
+
+    it('应该注册 ZhinTool 实例', () => {
+      const zhinTool = new ZhinTool('zhin_tool')
+        .desc('ZhinTool 测试')
+        .execute(async () => 'result');
+
+      service.addTool(zhinTool, 'test-plugin', false);
+      
+      const registered = service.get('zhin_tool');
+      expect(registered).toBeDefined();
+      expect(registered?.description).toBe('ZhinTool 测试');
+    });
+
+    it('应该正确移除工具', () => {
+      const tool: Tool = {
+        name: 'removable',
+        description: '',
+        parameters: { type: 'object', properties: {} },
+        execute: async () => '',
+      };
+
+      service.addTool(tool, 'test', false);
+      expect(service.removeTool('removable')).toBe(true);
+      expect(service.removeTool('removable')).toBe(false);
+    });
+
+    it('应该自动添加来源标识', () => {
+      const tool: Tool = {
+        name: 'sourced_tool',
+        description: '',
+        parameters: { type: 'object', properties: {} },
+        execute: async () => '',
+      };
+
+      service.addTool(tool, 'my-plugin', false);
+      
+      const registered = service.get('sourced_tool');
+      expect(registered?.source).toContain('my-plugin');
+      expect(registered?.tags).toContain('my-plugin');
+    });
+  });
+
+  describe('工具执行', () => {
+    it('应该执行已注册的工具', async () => {
+      const tool: Tool = {
+        name: 'executable',
+        description: '',
+        parameters: { type: 'object', properties: {} },
+        execute: async (args) => `received: ${JSON.stringify(args)}`,
+      };
+
+      service.addTool(tool, 'test', false);
+      
+      const result = await service.execute('executable', { key: 'value' });
+      expect(result).toBe('received: {"key":"value"}');
+    });
+
+    it('执行不存在的工具应抛出错误', async () => {
+      await expect(service.execute('nonexistent', {})).rejects.toThrow('not found');
+    });
+  });
+
+  describe('标签过滤', () => {
+    it('应该按标签过滤工具', () => {
+      service.addTool({
+        name: 'tagged1',
+        description: '',
+        parameters: { type: 'object', properties: {} },
+        tags: ['utility'],
+        execute: async () => '',
+      }, 'test', false);
+      
+      service.addTool({
+        name: 'tagged2',
+        description: '',
+        parameters: { type: 'object', properties: {} },
+        tags: ['helper'],
+        execute: async () => '',
+      }, 'test', false);
+
+      const utilityTools = service.getByTags(['utility']);
+      expect(utilityTools).toHaveLength(1);
+      expect(utilityTools[0].name).toBe('tagged1');
+    });
+  });
+
+  describe('上下文过滤', () => {
+    it('应该按平台过滤工具', () => {
+      service.addTool({
+        name: 'qq_only',
+        description: '',
+        parameters: { type: 'object', properties: {} },
+        platforms: ['qq'],
+        execute: async () => '',
+      }, 'test', false);
+      
+      service.addTool({
+        name: 'all_platforms',
+        description: '',
+        parameters: { type: 'object', properties: {} },
+        execute: async () => '',
+      }, 'test', false);
+
+      const qqContext: ToolContext = { platform: 'qq' };
+      const telegramContext: ToolContext = { platform: 'telegram' };
+      
+      const allTools = service.getAll();
+      
+      const qqFiltered = service.filterByContext(allTools, qqContext);
+      expect(qqFiltered.some(t => t.name === 'qq_only')).toBe(true);
+      expect(qqFiltered.some(t => t.name === 'all_platforms')).toBe(true);
+      
+      const telegramFiltered = service.filterByContext(allTools, telegramContext);
+      expect(telegramFiltered.some(t => t.name === 'qq_only')).toBe(false);
+      expect(telegramFiltered.some(t => t.name === 'all_platforms')).toBe(true);
+    });
+
+    it('应该按权限过滤工具', () => {
+      service.addTool({
+        name: 'admin_tool',
+        description: '',
+        parameters: { type: 'object', properties: {} },
+        permissionLevel: 'bot_admin',
+        execute: async () => '',
+      }, 'test', false);
+
+      const allTools = service.getAll();
+      
+      const userContext: ToolContext = {};
+      const adminContext: ToolContext = { isBotAdmin: true };
+      
+      const userFiltered = service.filterByContext(allTools, userContext);
+      expect(userFiltered.some(t => t.name === 'admin_tool')).toBe(false);
+      
+      const adminFiltered = service.filterByContext(allTools, adminContext);
+      expect(adminFiltered.some(t => t.name === 'admin_tool')).toBe(true);
+    });
+  });
+});
+
+// ============================================================================
+// AI Trigger 工具函数测试
+// ============================================================================
+
+describe('AI Trigger 工具函数测试', () => {
+  function createMockMessage(options: {
+    content: string | any[];
+    bot?: string;
+    channelType?: 'private' | 'group' | 'channel';
+    senderId?: string;
+    senderPermissions?: string[];
+  }) {
+    const content = typeof options.content === 'string' 
+      ? [{ type: 'text', data: { text: options.content } }]
+      : options.content;
+    
+    return {
+      $content: content,
+      $bot: options.bot || 'bot123',
+      $channel: options.channelType ? { type: options.channelType, id: 'channel1' } : null,
+      $sender: { 
+        id: options.senderId || 'user1', 
+        permissions: options.senderPermissions || [],
+      },
+      $adapter: 'test',
+    };
+  }
+
+  describe('shouldTriggerAI', () => {
+    it('应该检测前缀触发', () => {
+      const message = createMockMessage({ content: '# 你好' });
+      const result = shouldTriggerAI(message as any, { prefixes: ['#'] });
+      
+      expect(result.triggered).toBe(true);
+      expect(result.content).toBe('你好');
+    });
+
+    it('没有匹配前缀时不应触发', () => {
+      const message = createMockMessage({ content: '普通消息' });
+      const result = shouldTriggerAI(message as any, { prefixes: ['#'] });
+      
+      expect(result.triggered).toBe(false);
+    });
+
+    it('私聊应该直接触发', () => {
+      const message = createMockMessage({ content: '你好', channelType: 'private' });
+      const result = shouldTriggerAI(message as any, { respondToPrivate: true });
+      
+      expect(result.triggered).toBe(true);
+      expect(result.content).toBe('你好');
+    });
+
+    it('禁用时不应触发', () => {
+      const message = createMockMessage({ content: '# 你好' });
+      const result = shouldTriggerAI(message as any, { enabled: false, prefixes: ['#'] });
+      
+      expect(result.triggered).toBe(false);
+    });
+  });
+
+  describe('inferSenderPermissions', () => {
+    it('应该正确推断 owner 权限', () => {
+      const message = createMockMessage({ content: 'test', senderId: 'owner1' });
+      const result = inferSenderPermissions(message as any, { owners: ['owner1'] });
+      
+      expect(result.isOwner).toBe(true);
+      expect(result.permissionLevel).toBe('owner');
+    });
+
+    it('默认应该是 user 权限', () => {
+      const message = createMockMessage({ content: 'test' });
+      const result = inferSenderPermissions(message as any, {});
+      
+      expect(result.permissionLevel).toBe('user');
+    });
+  });
+});
+
+// ============================================================================
+// 内置工具测试
+// ============================================================================
+
+describe('内置工具完整测试', () => {
+  describe('计算器工具', () => {
+    const calculator = calculatorTool.toTool();
+
+    it('应该正确处理嵌套括号', async () => {
+      const result = await calculator.execute({ expression: '((2 + 3) * (4 - 1))' });
+      expect(result.result).toBe(15);
+    });
+
+    it('应该处理负数', async () => {
+      const result = await calculator.execute({ expression: '-5 + 3' });
+      expect(result.result).toBe(-2);
+    });
+
+    it('应该处理小数', async () => {
+      const result = await calculator.execute({ expression: '1.5 + 2.5' });
+      expect(result.result).toBe(4);
+    });
+
+    it('应该处理科学计数法', async () => {
+      const result = await calculator.execute({ expression: '1e2 + 50' });
+      expect(result.result).toBe(150);
+    });
+  });
+
+  describe('时间工具', () => {
+    const timeToolObj = timeTool.toTool();
+
+    it('应该返回 ISO 格式时间', async () => {
+      const result = await timeToolObj.execute({ format: 'timestamp' });
+      expect(result.iso).toMatch(/^\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}/);
+    });
+
+    it('时间戳应该是合理的', async () => {
+      const result = await timeToolObj.execute({});
+      const now = Date.now();
+      expect(Math.abs(result.timestamp - now)).toBeLessThan(1000);
+    });
+  });
+
+  describe('getBuiltinTools', () => {
+    it('应该返回内置工具列表', () => {
+      const tools = getBuiltinTools();
+      expect(Array.isArray(tools)).toBe(true);
+      expect(tools.length).toBeGreaterThan(0);
+      
+      const names = tools.map(t => t.name);
+      expect(names).toContain('calculator');
+      expect(names).toContain('get_time');
+    });
+  });
+});
+
+// ============================================================================
+// ZhinTool 完整流程测试
+// ============================================================================
+
+describe('ZhinTool 完整流程', () => {
+  it('应该支持完整的工具定义流程', async () => {
+    const tool = new ZhinTool('complete_tool')
+      .desc('完整测试工具')
+      .param('required_param', { type: 'string', description: '必填参数' }, true)
+      .param('optional_param', { type: 'number', description: '可选参数' }, false)
+      .platform('qq', 'telegram')
+      .scope('group', 'private')
+      .permission('user')
+      .tag('test', 'example')
+      .usage('这是使用说明')
+      .examples('/complete_tool arg1', '/complete_tool arg1 123')
+      .alias('ct')
+      .execute(async (args, ctx) => {
+        return {
+          received: args,
+          platform: ctx?.platform,
+        };
+      })
+      .action(async (message, result) => {
+        return `Command executed: ${result.params.required_param}`;
+      });
+
+    // 转换为 Tool
+    const toolObj = tool.toTool();
+    
+    // 验证基本属性
+    expect(toolObj.name).toBe('complete_tool');
+    expect(toolObj.description).toBe('完整测试工具');
+    expect(toolObj.platforms).toEqual(['qq', 'telegram']);
+    expect(toolObj.scopes).toEqual(['group', 'private']);
+    expect(toolObj.tags).toContain('test');
+    expect(toolObj.tags).toContain('example');
+    
+    // 验证参数
+    expect(toolObj.parameters.properties).toHaveProperty('required_param');
+    expect(toolObj.parameters.properties).toHaveProperty('optional_param');
+    expect(toolObj.parameters.required).toContain('required_param');
+    
+    // 验证命令配置
+    expect(toolObj.command).not.toBe(false);
+    expect((toolObj.command as any).usage).toContain('这是使用说明');
+    expect((toolObj.command as any).examples).toContain('/complete_tool arg1');
+    expect((toolObj.command as any).alias).toContain('ct');
+    
+    // 验证执行
+    const result = await toolObj.execute(
+      { required_param: 'test', optional_param: 42 },
+      { platform: 'qq' }
+    );
+    
+    expect(result.received.required_param).toBe('test');
+    expect(result.received.optional_param).toBe(42);
+    expect(result.platform).toBe('qq');
+    
+    // 验证 JSON 输出
+    const json = tool.toJSON();
+    expect(json.name).toBe('complete_tool');
+    expect(json).not.toHaveProperty('execute');
+    
+    // 验证帮助信息
+    const help = tool.help;
+    expect(help).toContain('complete_tool');
+    expect(help).toContain('必填参数');
+    expect(help).toContain('可选参数');
+  });
+});