@zhin.js/ai 1.0.0 → 1.0.1

package/tests/session.test.ts

@@ -1,15 +1,9 @@
 /**
  * Session Manager 测试
- * 
- * 测试会话管理功能：
- * - 会话创建和获取
- * - 消息历史管理
- * - 会话超时清理
  */
 import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
 
-// Mock Logger
-vi.mock('@zhin.js/core', async (importOriginal) => {
+vi.mock('@zhin.js/logger', async (importOriginal) => {
   const original = await importOriginal() as any;
   return {
     ...original,
@@ -22,11 +16,12 @@ vi.mock('@zhin.js/core', async (importOriginal) => {
   };
 });
 
-import { 
-  MemorySessionManager, 
-  SessionManager, 
-  createMemorySessionManager 
-} from '../src/session.js';
+import {
+  MemorySessionManager,
+  DatabaseSessionManager,
+  SessionManager,
+  createMemorySessionManager,
+} from '@zhin.js/ai';
 
 describe('MemorySessionManager', () => {
   let manager: MemorySessionManager;
@@ -35,7 +30,7 @@ describe('MemorySessionManager', () => {
     vi.useFakeTimers();
     manager = new MemorySessionManager({
       maxHistory: 10,
-      expireMs: 60000, // 1 分钟
+      expireMs: 60000,
     });
   });
 
@@ -47,7 +42,6 @@ describe('MemorySessionManager', () => {
   describe('会话创建', () => {
     it('应该创建新会话', () => {
       const session = manager.get('user-1');
-      
       expect(session).toBeDefined();
       expect(session.id).toBe('user-1');
       expect(session.messages).toEqual([]);
@@ -56,16 +50,13 @@ describe('MemorySessionManager', () => {
     it('应该返回已存在的会话', () => {
       const session1 = manager.get('user-1');
       manager.addMessage('user-1', { role: 'user', content: 'test' });
-      
       const session2 = manager.get('user-1');
-      
       expect(session2.messages).toHaveLength(1);
     });
 
     it('不同用户应该有不同的会话', () => {
       const session1 = manager.get('user-1');
       const session2 = manager.get('user-2');
-      
       expect(session1.id).not.toBe(session2.id);
     });
   });
@@ -74,7 +65,6 @@ describe('MemorySessionManager', () => {
     it('应该添加消息到历史', () => {
       manager.addMessage('user-1', { role: 'user', content: '你好' });
       manager.addMessage('user-1', { role: 'assistant', content: '你好！有什么可以帮你的？' });
-      
       const messages = manager.getMessages('user-1');
       expect(messages).toHaveLength(2);
       expect(messages[0].role).toBe('user');
@@ -82,24 +72,20 @@ describe('MemorySessionManager', () => {
     });
 
     it('应该限制历史记录数量', () => {
-      // 添加超过限制的消息
       for (let i = 0; i < 15; i++) {
         manager.addMessage('user-1', { role: 'user', content: `消息 ${i}` });
       }
-      
       const messages = manager.getMessages('user-1');
       expect(messages.length).toBeLessThanOrEqual(10);
     });
 
     it('系统消息应该保留', () => {
       manager.setSystemPrompt('user-1', '你是一个助手');
-      
       for (let i = 0; i < 15; i++) {
         manager.addMessage('user-1', { role: 'user', content: `消息 ${i}` });
       }
-      
       const messages = manager.getMessages('user-1');
-      const systemMessages = messages.filter(m => m.role === 'system');
+      const systemMessages = messages.filter((m: { role: string }) => m.role === 'system');
       expect(systemMessages.length).toBe(1);
     });
   });
@@ -107,7 +93,6 @@ describe('MemorySessionManager', () => {
   describe('系统提示', () => {
     it('应该设置系统提示', () => {
       manager.setSystemPrompt('user-1', '你是一个助手');
-      
       const messages = manager.getMessages('user-1');
       expect(messages[0].role).toBe('system');
       expect(messages[0].content).toBe('你是一个助手');
@@ -116,9 +101,8 @@ describe('MemorySessionManager', () => {
     it('应该替换旧的系统提示', () => {
       manager.setSystemPrompt('user-1', '旧提示');
       manager.setSystemPrompt('user-1', '新提示');
-      
       const messages = manager.getMessages('user-1');
-      const systemMessages = messages.filter(m => m.role === 'system');
+      const systemMessages = messages.filter((m: { role: string }) => m.role === 'system');
       expect(systemMessages.length).toBe(1);
       expect(systemMessages[0].content).toBe('新提示');
     });
@@ -128,7 +112,6 @@ describe('MemorySessionManager', () => {
     it('应该清除指定会话', () => {
       manager.get('user-1');
       manager.addMessage('user-1', { role: 'user', content: 'test' });
-      
       expect(manager.clear('user-1')).toBe(true);
       expect(manager.has('user-1')).toBe(false);
     });
@@ -140,9 +123,7 @@ describe('MemorySessionManager', () => {
     it('reset 应该保留系统消息', () => {
       manager.setSystemPrompt('user-1', '系统提示');
       manager.addMessage('user-1', { role: 'user', content: 'test' });
-      
       manager.reset('user-1');
-      
       const messages = manager.getMessages('user-1');
       expect(messages.length).toBe(1);
       expect(messages[0].role).toBe('system');
@@ -152,33 +133,18 @@ describe('MemorySessionManager', () => {
   describe('会话超时', () => {
     it('应该在超时后清理会话', () => {
       manager.get('user-1');
-      
       expect(manager.has('user-1')).toBe(true);
-      
-      // 前进时间超过超时时间
       vi.advanceTimersByTime(61000);
-      
-      // 触发清理
       manager.cleanup();
-      
       expect(manager.has('user-1')).toBe(false);
     });
 
     it('活动会话不应该被清理', () => {
       manager.get('user-1');
-      
-      // 前进一半时间
       vi.advanceTimersByTime(30000);
-      
-      // 添加新消息（刷新活动时间）
       manager.addMessage('user-1', { role: 'user', content: '新消息' });
-      
-      // 再前进一半时间
       vi.advanceTimersByTime(30000);
-      
       manager.cleanup();
-      
-      // 会话应该还在
       expect(manager.has('user-1')).toBe(true);
     });
   });
@@ -187,7 +153,6 @@ describe('MemorySessionManager', () => {
     it('应该返回正确的统计信息', () => {
       manager.get('user-1');
       manager.get('user-2');
-      
       const stats = manager.getStats();
       expect(stats.total).toBe(2);
       expect(stats.active).toBe(2);
@@ -198,7 +163,6 @@ describe('MemorySessionManager', () => {
       manager.get('user-1');
       manager.get('user-2');
       manager.get('user-3');
-      
       const sessions = manager.listSessions();
       expect(sessions).toContain('user-1');
       expect(sessions).toContain('user-2');
@@ -210,9 +174,7 @@ describe('MemorySessionManager', () => {
     it('应该清理所有会话', () => {
       manager.get('user-1');
       manager.get('user-2');
-      
       manager.dispose();
-      
       expect(manager.has('user-1')).toBe(false);
       expect(manager.has('user-2')).toBe(false);
     });
@@ -241,3 +203,132 @@ describe('createMemorySessionManager', () => {
     manager.dispose();
   });
 });
+
+describe('DatabaseSessionManager', () => {
+  function createMockModel(records: any[] = []) {
+    return {
+      select: vi.fn().mockReturnValue({
+        where: vi.fn().mockResolvedValue(records),
+      }),
+      create: vi.fn().mockResolvedValue(undefined),
+      update: vi.fn().mockReturnValue({
+        where: vi.fn().mockResolvedValue(undefined),
+      }),
+      delete: vi.fn().mockReturnValue({
+        where: vi.fn().mockResolvedValue(undefined),
+      }),
+    };
+  }
+
+  afterEach(() => {
+    vi.useRealTimers();
+  });
+
+  describe('loadSession JSON 解析', () => {
+    it('应该解析字符串形式的 messages (JSON)', async () => {
+      const messagesJson = JSON.stringify([
+        { role: 'user', content: 'hello' },
+        { role: 'assistant', content: 'hi there' },
+      ]);
+      const model = createMockModel([{
+        session_id: 'test-1',
+        messages: messagesJson,
+        config: JSON.stringify({ provider: 'ollama' }),
+        created_at: 1000,
+        updated_at: 2000,
+      }]);
+
+      const manager = new DatabaseSessionManager(model);
+      const session = await manager.get('test-1');
+
+      expect(Array.isArray(session.messages)).toBe(true);
+      expect(session.messages).toHaveLength(2);
+      expect(session.messages[0]).toEqual({ role: 'user', content: 'hello' });
+      manager.dispose();
+    });
+
+    it('应该解析字符串形式的 config (JSON)', async () => {
+      const model = createMockModel([{
+        session_id: 'test-2',
+        messages: '[]',
+        config: '{"provider":"ollama","maxHistory":100}',
+        created_at: 1000,
+        updated_at: 2000,
+      }]);
+
+      const manager = new DatabaseSessionManager(model);
+      const session = await manager.get('test-2');
+
+      expect(typeof session.config).toBe('object');
+      expect(session.config).toEqual({ provider: 'ollama', maxHistory: 100 });
+      manager.dispose();
+    });
+
+    it('messages 已是数组时不应二次解析', async () => {
+      const messagesArray = [{ role: 'system', content: 'You are a bot' }];
+      const model = createMockModel([{
+        session_id: 'test-3',
+        messages: messagesArray,
+        config: { provider: 'openai' },
+        created_at: 1000,
+        updated_at: 2000,
+      }]);
+
+      const manager = new DatabaseSessionManager(model);
+      const session = await manager.get('test-3');
+
+      expect(session.messages).toBe(messagesArray);
+      expect(session.config).toEqual({ provider: 'openai' });
+      manager.dispose();
+    });
+
+    it('加载后的 messages 应支持 push 操作', async () => {
+      const model = createMockModel([{
+        session_id: 'test-4',
+        messages: '[{"role":"user","content":"hi"}]',
+        config: '{"provider":"openai"}',
+        created_at: 1000,
+        updated_at: 2000,
+      }]);
+
+      const manager = new DatabaseSessionManager(model);
+      await manager.addMessage('test-4', { role: 'assistant', content: 'hello!' });
+
+      const session = await manager.get('test-4');
+      expect(session.messages).toHaveLength(2);
+      expect(session.messages[1]).toEqual({ role: 'assistant', content: 'hello!' });
+      manager.dispose();
+    });
+
+    it('messages 或 config 为 null/undefined 时应使用默认值', async () => {
+      const model = createMockModel([{
+        session_id: 'test-5',
+        messages: null,
+        config: null,
+        created_at: 1000,
+        updated_at: 2000,
+      }]);
+
+      const manager = new DatabaseSessionManager(model);
+      const session = await manager.get('test-5');
+
+      expect(Array.isArray(session.messages)).toBe(true);
+      expect(session.messages).toHaveLength(0);
+      expect(session.config).toEqual({ provider: 'openai' });
+      manager.dispose();
+    });
+  });
+
+  describe('新会话创建', () => {
+    it('数据库无记录时应创建新会话', async () => {
+      const model = createMockModel([]);
+      const manager = new DatabaseSessionManager(model);
+      const session = await manager.get('new-session');
+
+      expect(session.id).toBe('new-session');
+      expect(Array.isArray(session.messages)).toBe(true);
+      expect(session.messages).toHaveLength(0);
+      manager.dispose();
+    });
+  });
+});

package/tests/setup.ts

@@ -1,18 +1,9 @@
 /**
- * AI 模块测试环境设置
- * 
- * 提供：
- * - Mock 对象工厂
- * - 测试辅助函数
- * - 公共测试配置
+ * @zhin.js/ai 测试环境设置
+ * 提供通用 AI Mock 与测试辅助
  */
 import { vi } from 'vitest';
-import type { Message, MessageElement, Tool, ToolContext } from '@zhin.js/core';
-import type { AIConfig, AIProviderConfig, ChatMessage } from '../src/types.js';
-
-// ============================================================================
-// Logger Mock
-// ============================================================================
+import type { ChatMessage, AgentTool } from '@zhin.js/ai';
 
 export const createMockLogger = () => ({
   debug: vi.fn(),
@@ -21,139 +12,6 @@ export const createMockLogger = () => ({
   error: vi.fn(),
 });
 
-// ============================================================================
-// Plugin Mock
-// ============================================================================
-
-export const createMockPlugin = (name = 'test-plugin') => ({
-  name,
-  root: {
-    inject: vi.fn(),
-    contexts: new Map(),
-    middleware: vi.fn(),
-  },
-  logger: createMockLogger(),
-  onDispose: vi.fn(),
-  collectAllTools: vi.fn(() => []),
-  provide: vi.fn(),
-  useContext: vi.fn(),
-  addMiddleware: vi.fn(),
-});
-
-// ============================================================================
-// Message Mock
-// ============================================================================
-
-export interface MockMessageOptions {
-  content?: string;
-  elements?: MessageElement[];
-  platform?: string;
-  channelType?: 'group' | 'private' | 'guild';
-  channelId?: string;
-  senderId?: string;
-  senderPermissions?: string[];
-  senderRole?: string;
-  botId?: string;
-}
-
-export const createMockMessage = (options: MockMessageOptions = {}): Partial<Message> => {
-  const {
-    content = '测试消息',
-    elements,
-    platform = 'test',
-    channelType = 'group',
-    channelId = 'channel-1',
-    senderId = 'user-1',
-    senderPermissions = [],
-    senderRole,
-    botId = 'bot-1',
-  } = options;
-
-  const $content: MessageElement[] = elements || [
-    { type: 'text', data: { text: content } },
-  ];
-
-  return {
-    $content,
-    $bot: botId,
-    $adapter: platform,
-    $channel: {
-      type: channelType,
-      id: channelId,
-      name: 'Test Channel',
-    },
-    $sender: {
-      id: senderId,
-      name: 'Test User',
-      permissions: senderPermissions,
-      role: senderRole,
-    },
-    $reply: vi.fn().mockResolvedValue(undefined),
-    $quote: vi.fn().mockResolvedValue(undefined),
-  };
-};
-
-// ============================================================================
-// Tool Mock
-// ============================================================================
-
-export interface MockToolOptions {
-  name: string;
-  description?: string;
-  parameters?: Record<string, { type: string; description?: string }>;
-  required?: string[];
-  platforms?: string[];
-  scopes?: ('group' | 'private' | 'guild')[];
-  permissionLevel?: 'user' | 'group_admin' | 'group_owner' | 'bot_admin' | 'owner';
-  tags?: string[];
-  executeResult?: any;
-  executeError?: Error;
-}
-
-export const createMockTool = (options: MockToolOptions): Tool => {
-  const {
-    name,
-    description = `${name} 工具`,
-    parameters = {},
-    required = [],
-    platforms,
-    scopes,
-    permissionLevel,
-    tags = [],
-    executeResult = 'success',
-    executeError,
-  } = options;
-
-  const properties: Record<string, any> = {};
-  for (const [key, value] of Object.entries(parameters)) {
-    properties[key] = {
-      type: value.type,
-      description: value.description,
-    };
-  }
-
-  return {
-    name,
-    description,
-    parameters: {
-      type: 'object',
-      properties,
-      required,
-    },
-    platforms,
-    scopes,
-    permissionLevel,
-    tags,
-    execute: executeError
-      ? vi.fn().mockRejectedValue(executeError)
-      : vi.fn().mockResolvedValue(executeResult),
-  };
-};
-
-// ============================================================================
-// AI Provider Mock
-// ============================================================================
-
 export interface MockProviderOptions {
   name?: string;
   response?: string | AsyncGenerator<string>;
@@ -172,83 +30,54 @@ export const createMockProvider = (options: MockProviderOptions = {}) => {
   if (error) {
     return {
       name,
+      models: ['mock-model'],
       chat: vi.fn().mockRejectedValue(error),
       healthCheck: vi.fn().mockResolvedValue(false),
     };
   }
 
-  const generateResponse = async function* (): AsyncGenerator<{
-    content?: string;
-    toolCalls?: Array<{ name: string; arguments: Record<string, any> }>;
-    done: boolean;
-  }> {
-    if (toolCalls.length > 0) {
-      yield { toolCalls, done: false };
-    }
-    
-    if (typeof response === 'string') {
-      yield { content: response, done: true };
-    } else {
-      for await (const chunk of response) {
-        yield { content: chunk, done: false };
-      }
-      yield { done: true };
-    }
-  };
-
   return {
     name,
-    chat: vi.fn().mockImplementation(() => generateResponse()),
+    models: ['mock-model'],
+    chat: vi.fn(),
     healthCheck: vi.fn().mockResolvedValue(true),
   };
 };
 
-// ============================================================================
-// Context Factory
-// ============================================================================
-
-export const createToolContext = (options: Partial<ToolContext> = {}): ToolContext => ({
-  platform: 'test',
-  scope: 'group',
-  permissionLevel: 'user',
-  ...options,
+/** 构建 ChatCompletionResponse 用于测试 */
+export const createChatResponse = (content: string, toolCalls?: any[]) => ({
+  id: 'test-id',
+  object: 'chat.completion',
+  created: Date.now(),
+  model: 'mock-model',
+  choices: [{
+    index: 0,
+    message: {
+      role: 'assistant',
+      content,
+      tool_calls: toolCalls,
+    },
+    finish_reason: toolCalls ? 'tool_calls' : 'stop',
+  }],
+  usage: {
+    prompt_tokens: 10,
+    completion_tokens: 10,
+    total_tokens: 20,
+  },
 });
 
-// ============================================================================
-// AI Config Factory
-// ============================================================================
-
-export const createMockAIConfig = (overrides: Partial<AIConfig> = {}): AIConfig => ({
-  defaultProvider: 'mock',
-  sessions: {
-    maxHistory: 10,
-    timeout: 300000,
-  },
-  context: {
-    enabled: false,
-    maxSize: 100,
-  },
-  trigger: {
-    enabled: true,
-    prefixes: ['#'],
-    ignorePrefixes: ['/'],
-    allowAtBot: true,
-    allowPrivateChat: true,
-  },
-  ...overrides,
+export const createChatMessage = (
+  role: 'user' | 'assistant' | 'system',
+  content: string
+): ChatMessage => ({
+  role,
+  content,
 });
 
-// ============================================================================
-// Test Utilities
-// ============================================================================
+export const delay = (ms: number): Promise<void> =>
+  new Promise((resolve) => setTimeout(resolve, ms));
 
-/**
- * 等待 Promise 解决或超时
- */
-export const waitFor = async <T>(
-  promise: Promise<T>,
-  timeout = 5000
-): Promise<T> => {
+export const waitFor = async <T>(promise: Promise<T>, timeout = 5000): Promise<T> => {
   return Promise.race([
     promise,
     new Promise<T>((_, reject) =>
@@ -257,12 +86,7 @@ export const waitFor = async <T>(
   ]);
 };
 
-/**
- * 收集 AsyncGenerator 的所有值
- */
-export const collectAsyncGenerator = async <T>(
-  generator: AsyncGenerator<T>
-): Promise<T[]> => {
+export const collectAsyncGenerator = async <T>(generator: AsyncGenerator<T>): Promise<T[]> => {
   const results: T[] = [];
   for await (const item of generator) {
     results.push(item);
@@ -270,35 +94,53 @@ export const collectAsyncGenerator = async <T>(
   return results;
 };
 
-/**
- * 创建延迟 Promise
- */
-export const delay = (ms: number): Promise<void> =>
-  new Promise((resolve) => setTimeout(resolve, ms));
+export interface MockToolOptions {
+  name: string;
+  description?: string;
+  parameters?: Record<string, { type: string; description?: string }>;
+  required?: string[];
+  tags?: string[];
+  keywords?: string[];
+  permissionLevel?: number;
+  executeResult?: any;
+  executeError?: Error;
+}
 
-/**
- * 创建带有所有字段的 ChatMessage
- */
-export const createChatMessage = (
-  role: 'user' | 'assistant' | 'system',
-  content: string
-): ChatMessage => ({
-  role,
-  content,
-});
+export const createMockTool = (options: MockToolOptions): AgentTool => {
+  const {
+    name,
+    description = `${name} 工具`,
+    parameters = {},
+    required = [],
+    tags = [],
+    keywords = [],
+    permissionLevel,
+    executeResult = 'success',
+    executeError,
+  } = options;
 
-/**
- * 验证工具参数结构
- */
-export const assertToolParameters = (
-  tool: Tool,
-  expectedProperties: string[],
-  expectedRequired: string[] = []
-) => {
-  const props = Object.keys(tool.parameters.properties || {});
-  expect(props).toEqual(expect.arrayContaining(expectedProperties));
-  
-  if (expectedRequired.length > 0) {
-    expect(tool.parameters.required).toEqual(expect.arrayContaining(expectedRequired));
+  const properties: Record<string, any> = {};
+  for (const [key, value] of Object.entries(parameters)) {
+    properties[key] = {
+      type: value.type,
+      description: value.description,
+    };
   }
+
+  const tool: any = {
+    name,
+    description,
+    parameters: {
+      type: 'object',
+      properties,
+      required,
+    },
+    tags,
+    keywords,
+    permissionLevel,
+    execute: executeError
+      ? vi.fn().mockRejectedValue(executeError)
+      : vi.fn().mockResolvedValue(executeResult),
+  };
+  return tool;
 };