@zhin.js/ai 1.0.0 → 1.0.1

package/src/tools.ts

@@ -1,205 +0,0 @@
-/**
- * @zhin.js/ai - Built-in Tools
- * 内置工具集合 - 使用 ZhinTool 类定义
- */
-
-import { ZhinTool, type Tool } from '@zhin.js/core';
-
-/**
- * 计算器工具
- * 支持基本运算和数学函数
- */
-export const calculatorTool = new ZhinTool('calculator')
-  .desc('执行数学计算。支持基本运算（+、-、*、/、^）和数学函数（sin、cos、sqrt 等）')
-  .param('expression', { type: 'string', description: '数学表达式，例如 "2 + 3 * 4" 或 "sqrt(16)"' }, true)
-  .execute(async ({ expression }) => {
-    try {
-      // 安全的数学表达式求值
-      const sanitized = (expression as string)
-        .replace(/[^0-9+\-*/().^eEsincosqrtabspowlogxMathPI\s]/g, '')
-        .replace(/\bsqrt\b/g, 'Math.sqrt')
-        .replace(/\bsin\b/g, 'Math.sin')
-        .replace(/\bcos\b/g, 'Math.cos')
-        .replace(/\btan\b/g, 'Math.tan')
-        .replace(/\blog\b/g, 'Math.log')
-        .replace(/\babs\b/g, 'Math.abs')
-        .replace(/\bpow\b/g, 'Math.pow')
-        .replace(/\bPI\b/g, 'Math.PI')
-        .replace(/\bE\b(?![0-9])/g, 'Math.E')
-        .replace(/\^/g, '**');
-
-      const result = new Function(`return ${sanitized}`)();
-      return { result, expression };
-    } catch (error) {
-      return { error: '无法计算表达式', expression };
-    }
-  });
-
-/**
- * 时间工具
- * 获取当前时间和日期
- */
-export const timeTool = new ZhinTool('get_time')
-  .desc('获取当前时间和日期信息')
-  .param('timezone', { type: 'string', description: '时区，例如 "Asia/Shanghai" 或 "UTC"' })
-  .param('format', { type: 'string', description: '输出格式: full(完整), date(日期), time(时间), timestamp(时间戳)' })
-  .execute(async ({ timezone, format = 'full' }) => {
-    const now = new Date();
-    const options: Intl.DateTimeFormatOptions = {
-      timeZone: (timezone as string) || 'Asia/Shanghai',
-    };
-
-    switch (format) {
-      case 'date':
-        options.dateStyle = 'full';
-        break;
-      case 'time':
-        options.timeStyle = 'long';
-        break;
-      case 'timestamp':
-        return { timestamp: now.getTime(), iso: now.toISOString() };
-      default:
-        options.dateStyle = 'full';
-        options.timeStyle = 'long';
-    }
-
-    return {
-      formatted: now.toLocaleString('zh-CN', options),
-      timestamp: now.getTime(),
-      iso: now.toISOString(),
-    };
-  });
-
-/**
- * 网页搜索工具
- * 需要配置搜索 API
- */
-export const searchTool = new ZhinTool('web_search')
-  .desc('在互联网上搜索信息（需要配置搜索 API）')
-  .param('query', { type: 'string', description: '搜索关键词' }, true)
-  .param('limit', { type: 'number', description: '返回结果数量（默认 5）' })
-  .execute(async ({ query, limit = 5 }) => {
-    // 默认实现提示需要配置
-    return {
-      query,
-      error: '搜索功能未配置，请提供搜索 API',
-      hint: '可以集成 SerpAPI、Bing Search API 或 Google Custom Search',
-    };
-  });
-
-/**
- * 代码执行工具
- * 在安全沙箱中执行 JavaScript
- */
-export const codeRunnerTool = new ZhinTool('run_code')
-  .desc('执行 JavaScript 代码（在安全沙箱中）')
-  .param('code', { type: 'string', description: 'JavaScript 代码' }, true)
-  .execute(async ({ code }) => {
-    try {
-      // 简单的沙箱执行（生产环境应使用 vm2 或 isolated-vm）
-      const result = new Function(`
-        'use strict';
-        const console = { log: (...args) => args.join(' ') };
-        return (function() { ${code} })();
-      `)();
-      
-      return {
-        success: true,
-        result: result !== undefined ? String(result) : 'undefined',
-      };
-    } catch (error) {
-      return {
-        success: false,
-        error: error instanceof Error ? error.message : String(error),
-      };
-    }
-  });
-
-/**
- * HTTP 请求工具
- * 发送 HTTP 请求获取数据
- */
-export const httpTool = new ZhinTool('http_request')
-  .desc('发送 HTTP 请求获取数据')
-  .param('url', { type: 'string', description: '请求 URL' }, true)
-  .param('method', { type: 'string', description: 'HTTP 方法: GET, POST, PUT, DELETE（默认 GET）' })
-  .param('headers', { type: 'object', description: '请求头（JSON 对象）' })
-  .param('body', { type: 'string', description: '请求体（JSON 字符串）' })
-  .execute(async ({ url, method = 'GET', headers = {}, body }) => {
-    try {
-      const response = await fetch(url as string, {
-        method: method as string,
-        headers: {
-          'Content-Type': 'application/json',
-          ...(headers as Record<string, string>),
-        },
-        body: body ? (body as string) : undefined,
-      });
-
-      const contentType = response.headers.get('content-type') || '';
-      let data: any;
-
-      if (contentType.includes('application/json')) {
-        data = await response.json();
-      } else {
-        data = await response.text();
-        // 限制文本长度
-        if (data.length > 5000) {
-          data = data.substring(0, 5000) + '... (truncated)';
-        }
-      }
-
-      return {
-        status: response.status,
-        statusText: response.statusText,
-        data,
-      };
-    } catch (error) {
-      return {
-        error: error instanceof Error ? error.message : String(error),
-      };
-    }
-  });
-
-/**
- * 记忆工具
- * 让 AI 记住重要信息
- */
-export const memoryTool = new ZhinTool('remember')
-  .desc('记住用户告诉你的重要信息，以便后续对话中使用')
-  .param('key', { type: 'string', description: '记忆的标识符，如 "user_name", "preference"' }, true)
-  .param('value', { type: 'string', description: '要记住的内容' }, true)
-  .execute(async ({ key, value }, context) => {
-    // 这里需要与 session/context manager 集成
-    return {
-      success: true,
-      message: `已记住 ${key}: ${value}`,
-      key,
-      value,
-    };
-  });
-
-/**
- * 获取所有内置工具（ZhinTool 实例）
- * 注意：天气工具已移除，请使用 weather-tool 插件，支持多平台配置
- */
-export function getBuiltinTools(): ZhinTool[] {
-  return [
-    calculatorTool,
-    timeTool,
-  ];
-}
-
-/**
- * 获取所有可用的内置工具（包括可选工具）
- */
-export function getAllBuiltinTools(): ZhinTool[] {
-  return [
-    calculatorTool,
-    timeTool,
-    searchTool,
-    codeRunnerTool,
-    httpTool,
-    memoryTool,
-  ];
-}

package/tests/ai-trigger.test.ts

@@ -1,369 +0,0 @@
-/**
- * AI Trigger 工具函数测试
- * 
- * 测试内容：
- * 1. 前缀触发检测
- * 2. @机器人触发检测
- * 3. 私聊直接对话检测
- * 4. 关键词触发检测
- * 5. 忽略前缀检测
- * 6. 权限推断
- */
-import { describe, it, expect, vi } from 'vitest';
-import {
-  shouldTriggerAI,
-  mergeAITriggerConfig,
-  inferSenderPermissions,
-  parseRichMediaContent,
-  extractTextContent,
-  DEFAULT_AI_TRIGGER_CONFIG,
-  type AITriggerConfig,
-} from '@zhin.js/core';
-
-// 创建模拟消息
-function createMockMessage(options: {
-  content: string | any[];
-  bot?: string;
-  channelType?: 'private' | 'group' | 'channel';
-  senderId?: string;
-  senderPermissions?: string[];
-  senderRole?: 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 || [],
-      role: options.senderRole,
-    },
-    $adapter: 'test',
-    $reply: vi.fn(),
-  };
-}
-
-describe('AI Trigger 工具函数', () => {
-  describe('shouldTriggerAI - 前缀触发', () => {
-    it('应该检测 # 前缀', () => {
-      const message = createMockMessage({ content: '# 你好' });
-      const result = shouldTriggerAI(message as any, { prefixes: ['#'] });
-      
-      expect(result.triggered).toBe(true);
-      expect(result.content).toBe('你好');
-    });
-
-    it('应该检测 AI: 前缀', () => {
-      const message = createMockMessage({ content: 'AI:帮我计算' });
-      const result = shouldTriggerAI(message as any, { prefixes: ['AI:', 'ai:'] });
-      
-      expect(result.triggered).toBe(true);
-      expect(result.content).toBe('帮我计算');
-    });
-
-    it('应该检测小写 ai: 前缀', () => {
-      const message = createMockMessage({ content: 'ai:今天天气' });
-      const result = shouldTriggerAI(message as any, { prefixes: ['AI:', 'ai:'] });
-      
-      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);
-    });
-  });
-
-  describe('shouldTriggerAI - 忽略前缀', () => {
-    it('应该忽略命令前缀 /', () => {
-      const message = createMockMessage({ 
-        content: '/help',
-        channelType: 'private',
-      });
-      const result = shouldTriggerAI(message as any, { 
-        prefixes: ['#'],
-        ignorePrefixes: ['/'],
-        respondToPrivate: true,
-      });
-      
-      expect(result.triggered).toBe(false);
-    });
-
-    it('应该忽略命令前缀 !', () => {
-      const message = createMockMessage({ 
-        content: '!command',
-        channelType: 'private',
-      });
-      const result = shouldTriggerAI(message as any, { 
-        prefixes: ['#'],
-        ignorePrefixes: ['!', '！'],
-        respondToPrivate: true,
-      });
-      
-      expect(result.triggered).toBe(false);
-    });
-  });
-
-  describe('shouldTriggerAI - @机器人触发', () => {
-    it('应该检测 @机器人', () => {
-      const message = createMockMessage({
-        content: [
-          { type: 'at', data: { user_id: 'bot123' } },
-          { type: 'text', data: { text: ' 你好呀' } },
-        ],
-        bot: 'bot123',
-      });
-      const result = shouldTriggerAI(message as any, { respondToAt: true });
-      
-      expect(result.triggered).toBe(true);
-      expect(result.content).toBe(' 你好呀');
-    });
-
-    it('关闭 respondToAt 时不应触发', () => {
-      const message = createMockMessage({
-        content: [
-          { type: 'at', data: { user_id: 'bot123' } },
-          { type: 'text', data: { text: ' 你好' } },
-        ],
-        bot: 'bot123',
-      });
-      const result = shouldTriggerAI(message as any, { respondToAt: false });
-      
-      expect(result.triggered).toBe(false);
-    });
-
-    it('@其他人时不应触发', () => {
-      const message = createMockMessage({
-        content: [
-          { type: 'at', data: { user_id: 'other_user' } },
-          { type: 'text', data: { text: ' 你好' } },
-        ],
-        bot: 'bot123',
-      });
-      const result = shouldTriggerAI(message as any, { respondToAt: true });
-      
-      expect(result.triggered).toBe(false);
-    });
-  });
-
-  describe('shouldTriggerAI - 私聊直接对话', () => {
-    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('关闭 respondToPrivate 时私聊不应触发', () => {
-      const message = createMockMessage({
-        content: '你好',
-        channelType: 'private',
-      });
-      const result = shouldTriggerAI(message as any, { respondToPrivate: false });
-      
-      expect(result.triggered).toBe(false);
-    });
-
-    it('群聊时不应直接触发（需要前缀或@）', () => {
-      const message = createMockMessage({
-        content: '你好',
-        channelType: 'group',
-      });
-      const result = shouldTriggerAI(message as any, { respondToPrivate: true });
-      
-      expect(result.triggered).toBe(false);
-    });
-  });
-
-  describe('shouldTriggerAI - 关键词触发', () => {
-    it('应该检测关键词', () => {
-      const message = createMockMessage({ content: '今天天气怎么样' });
-      const result = shouldTriggerAI(message as any, { keywords: ['天气', '新闻'] });
-      
-      expect(result.triggered).toBe(true);
-      expect(result.content).toBe('今天天气怎么样');
-    });
-
-    it('关键词不区分大小写', () => {
-      const message = createMockMessage({ content: '说 hello 世界' });
-      const result = shouldTriggerAI(message as any, { keywords: ['HELLO'] });
-      
-      expect(result.triggered).toBe(true);
-    });
-
-    it('没有关键词配置时不应触发', () => {
-      const message = createMockMessage({ content: '天气真好' });
-      const result = shouldTriggerAI(message as any, { keywords: [] });
-      
-      expect(result.triggered).toBe(false);
-    });
-  });
-
-  describe('shouldTriggerAI - 禁用状态', () => {
-    it('enabled 为 false 时不应触发', () => {
-      const message = createMockMessage({ content: '# 你好' });
-      const result = shouldTriggerAI(message as any, { 
-        enabled: false,
-        prefixes: ['#'],
-      });
-      
-      expect(result.triggered).toBe(false);
-    });
-  });
-
-  describe('触发优先级', () => {
-    it('前缀触发应该优先于私聊触发', () => {
-      const message = createMockMessage({ 
-        content: '# 命令内容',
-        channelType: 'private',
-      });
-      const result = shouldTriggerAI(message as any, { 
-        prefixes: ['#'],
-        respondToPrivate: true,
-      });
-      
-      expect(result.triggered).toBe(true);
-      expect(result.content).toBe('命令内容'); // 应该去掉前缀
-    });
-  });
-
-  describe('边界情况', () => {
-    it('空消息不应触发', () => {
-      const message = createMockMessage({ 
-        content: '',
-        channelType: 'private',
-      });
-      const result = shouldTriggerAI(message as any, { respondToPrivate: true });
-      
-      expect(result.triggered).toBe(false);
-    });
-
-    it('只有空格的消息不应触发', () => {
-      const message = createMockMessage({ 
-        content: '   ',
-        channelType: 'private',
-      });
-      const result = shouldTriggerAI(message as any, { respondToPrivate: true });
-      
-      expect(result.triggered).toBe(false);
-    });
-
-    it('前缀后没有内容时应该触发但内容为空', () => {
-      const message = createMockMessage({ content: '#' });
-      const result = shouldTriggerAI(message as any, { prefixes: ['#'] });
-      
-      expect(result.triggered).toBe(true);
-      expect(result.content).toBe('');
-    });
-  });
-
-  describe('mergeAITriggerConfig', () => {
-    it('应该合并配置', () => {
-      const config = mergeAITriggerConfig({ prefixes: ['##'] });
-      
-      expect(config.prefixes).toEqual(['##']);
-      expect(config.enabled).toBe(true);
-      expect(config.respondToAt).toBe(true);
-    });
-
-    it('应该使用默认值', () => {
-      const config = mergeAITriggerConfig({});
-      
-      expect(config).toEqual(DEFAULT_AI_TRIGGER_CONFIG);
-    });
-  });
-
-  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('应该推断 bot_admin 权限', () => {
-      const message = createMockMessage({
-        content: 'test',
-        senderId: 'admin1',
-      });
-      const result = inferSenderPermissions(message as any, { botAdmins: ['admin1'] });
-      
-      expect(result.isBotAdmin).toBe(true);
-      expect(result.permissionLevel).toBe('bot_admin');
-    });
-
-    it('应该推断 group_owner 权限', () => {
-      const message = createMockMessage({
-        content: 'test',
-        senderPermissions: ['owner'],
-      });
-      const result = inferSenderPermissions(message as any, {});
-      
-      expect(result.isGroupOwner).toBe(true);
-      expect(result.permissionLevel).toBe('group_owner');
-    });
-
-    it('应该推断 group_admin 权限', () => {
-      const message = createMockMessage({
-        content: 'test',
-        senderPermissions: ['admin'],
-      });
-      const result = inferSenderPermissions(message as any, {});
-      
-      expect(result.isGroupAdmin).toBe(true);
-      expect(result.permissionLevel).toBe('group_admin');
-    });
-
-    it('默认应该是 user 权限', () => {
-      const message = createMockMessage({ content: 'test' });
-      const result = inferSenderPermissions(message as any, {});
-      
-      expect(result.permissionLevel).toBe('user');
-    });
-
-    it('应该推断 scope', () => {
-      const privateMsg = createMockMessage({ content: 'test', channelType: 'private' });
-      const groupMsg = createMockMessage({ content: 'test', channelType: 'group' });
-      
-      expect(inferSenderPermissions(privateMsg as any, {}).scope).toBe('private');
-      expect(inferSenderPermissions(groupMsg as any, {}).scope).toBe('group');
-    });
-  });
-
-  describe('parseRichMediaContent', () => {
-    it('应该解析纯文本', () => {
-      const result = parseRichMediaContent('Hello World');
-      
-      expect(result.length).toBeGreaterThan(0);
-      const textElement = result.find(el => el.type === 'text');
-      expect(textElement?.data?.text).toContain('Hello');
-    });
-
-    it('应该处理 XML 标签', () => {
-      const result = parseRichMediaContent('文本<image url="test.jpg"/>');
-      
-      expect(result.length).toBeGreaterThan(0);
-    });
-
-    it('应该处理空字符串', () => {
-      const result = parseRichMediaContent('');
-      
-      expect(Array.isArray(result)).toBe(true);
-    });
-  });
-});