@zhin.js/ai 0.0.1

package/tests/providers.integration.test.ts

@@ -0,0 +1,227 @@
+/**
+ * AI Providers 集成测试（真实 API 调用）
+ * 
+ * 使用 test-bot 的环境配置进行真实 API 测试
+ * 
+ * 运行方式（需要网络权限）：
+ * pnpm test packages/ai/tests/providers.integration.test.ts
+ * 
+ * 注意：此测试需要网络访问，在 sandbox 中会自动跳过
+ */
+import { describe, it, expect, beforeAll } from 'vitest';
+import { OllamaProvider } from '../src/providers/ollama.js';
+import type { ChatMessage } from '@zhin.js/core';
+
+// Ollama 配置
+const OLLAMA_CONFIG = {
+  baseUrl: 'https://ollama.l2cl.link',
+  models: ['qwen2.5:7b', 'qwen3:8b'],
+};
+
+// 全局状态
+let canRun = false;
+let provider: OllamaProvider | null = null;
+
+// 跳过检查
+const skipIfNoNetwork = (ctx: any) => {
+  if (!canRun) {
+    ctx.skip();
+    return true;
+  }
+  return false;
+};
+
+beforeAll(async () => {
+  try {
+    const controller = new AbortController();
+    const timeout = setTimeout(() => controller.abort(), 5000);
+    const response = await globalThis.fetch(`${OLLAMA_CONFIG.baseUrl}/api/tags`, {
+      signal: controller.signal,
+    });
+    clearTimeout(timeout);
+    canRun = response.ok;
+    
+    if (canRun) {
+      provider = new OllamaProvider({
+        baseUrl: OLLAMA_CONFIG.baseUrl,
+        models: OLLAMA_CONFIG.models,
+      });
+      console.log('✅ Ollama 服务可用');
+    }
+  } catch {
+    canRun = false;
+    console.log('⚠️ 网络不可用，跳过集成测试');
+  }
+});
+
+describe('Ollama Provider 集成测试', () => {
+  describe('基本聊天', () => {
+    it('应该能进行简单对话', async (ctx) => {
+      if (skipIfNoNetwork(ctx)) return;
+      
+      const response = await provider!.chat({
+        model: OLLAMA_CONFIG.models[0],
+        messages: [{ role: 'user', content: '你好，一句话介绍自己' }],
+        max_tokens: 100,
+      });
+
+      expect(response.choices[0].message.content).toBeTruthy();
+      console.log('📝 AI 回复:', response.choices[0].message.content);
+    }, 30000);
+
+    it('应该能处理多轮对话', async (ctx) => {
+      if (skipIfNoNetwork(ctx)) return;
+      
+      const messages: ChatMessage[] = [
+        { role: 'user', content: '记住：我叫小明。简短回复。' },
+      ];
+
+      const r1 = await provider!.chat({
+        model: OLLAMA_CONFIG.models[0],
+        messages,
+        max_tokens: 30,
+      });
+
+      messages.push({ role: 'assistant', content: r1.choices[0].message.content as string });
+      messages.push({ role: 'user', content: '我叫什么？' });
+
+      const r2 = await provider!.chat({
+        model: OLLAMA_CONFIG.models[0],
+        messages,
+        max_tokens: 30,
+      });
+
+      const reply = r2.choices[0].message.content as string;
+      console.log('📝 多轮对话:', reply);
+      expect(reply.toLowerCase()).toContain('小明');
+    }, 120000);
+  });
+
+  describe('工具调用', () => {
+    it('应该能调用计算器工具', async (ctx) => {
+      if (skipIfNoNetwork(ctx)) return;
+      
+      const response = await provider!.chat({
+        model: OLLAMA_CONFIG.models[0],
+        messages: [
+          { role: 'system', content: '使用 calculator 工具计算。' },
+          { role: 'user', content: '计算 15 * 8' },
+        ],
+        tools: [{
+          type: 'function',
+          function: {
+            name: 'calculator',
+            description: '计算数学表达式',
+            parameters: {
+              type: 'object',
+              properties: { expression: { type: 'string' } },
+              required: ['expression'],
+            },
+          },
+        }],
+        tool_choice: 'auto',
+        max_tokens: 200,
+      });
+
+      console.log('📝 工具调用:', JSON.stringify(response.choices[0].message, null, 2));
+      expect(response.choices[0].message).toBeDefined();
+    }, 30000);
+
+    it('应该能调用天气工具', async (ctx) => {
+      if (skipIfNoNetwork(ctx)) return;
+      
+      const response = await provider!.chat({
+        model: OLLAMA_CONFIG.models[0],
+        messages: [
+          { role: 'system', content: '使用 get_weather 工具查天气。' },
+          { role: 'user', content: '上海天气？' },
+        ],
+        tools: [{
+          type: 'function',
+          function: {
+            name: 'get_weather',
+            description: '查询天气',
+            parameters: {
+              type: 'object',
+              properties: { city: { type: 'string' } },
+              required: ['city'],
+            },
+          },
+        }],
+        tool_choice: 'auto',
+        max_tokens: 200,
+      });
+
+      console.log('📝 天气工具测试');
+      expect(response.choices[0].message).toBeDefined();
+    }, 30000);
+  });
+
+  describe('流式输出', () => {
+    it('应该能进行流式对话', async (ctx) => {
+      if (skipIfNoNetwork(ctx)) return;
+      
+      const stream = await provider!.chatStream({
+        model: OLLAMA_CONFIG.models[0],
+        messages: [{ role: 'user', content: '写4行诗' }],
+        max_tokens: 100,
+        stream: true,
+      });
+
+      let content = '';
+      let chunks = 0;
+
+      console.log('📝 流式输出:');
+      for await (const chunk of stream) {
+        const c = chunk.choices?.[0]?.delta?.content;
+        if (c) {
+          content += c;
+          chunks++;
+          process.stdout.write(c);
+        }
+      }
+      console.log(`\n  (${chunks} chunks)`);
+
+      expect(content.length).toBeGreaterThan(0);
+    }, 60000);
+  });
+
+  describe('错误处理', () => {
+    it('应该处理无效模型', async (ctx) => {
+      if (skipIfNoNetwork(ctx)) return;
+      
+      await expect(
+        provider!.chat({
+          model: 'invalid-model-xyz',
+          messages: [{ role: 'user', content: 'test' }],
+        })
+      ).rejects.toThrow();
+    }, 60000); // Ollama 拉取模型可能需要较长时间才会失败
+  });
+
+  describe('健康检查', () => {
+    it('应该返回健康状态', async (ctx) => {
+      if (skipIfNoNetwork(ctx)) return;
+      
+      const healthy = await provider!.healthCheck();
+      console.log('📝 健康:', healthy ? '✅' : '❌');
+      expect(typeof healthy).toBe('boolean');
+    }, 10000);
+  });
+});
+
+describe('性能测试', () => {
+  it('应该快速响应', async (ctx) => {
+    if (skipIfNoNetwork(ctx)) return;
+    
+    const start = Date.now();
+    await provider!.chat({
+      model: OLLAMA_CONFIG.models[0],
+      messages: [{ role: 'user', content: '1+1' }],
+      max_tokens: 10,
+    });
+    const ms = Date.now() - start;
+    console.log(`📝 响应: ${ms}ms`);
+    expect(ms).toBeLessThan(30000);
+  }, 35000);
+});

package/tests/session.test.ts

@@ -0,0 +1,243 @@
+/**
+ * Session Manager 测试
+ * 
+ * 测试会话管理功能：
+ * - 会话创建和获取
+ * - 消息历史管理
+ * - 会话超时清理
+ */
+import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';
+
+// Mock Logger
+vi.mock('@zhin.js/core', async (importOriginal) => {
+  const original = await importOriginal() as any;
+  return {
+    ...original,
+    Logger: class {
+      debug = vi.fn();
+      info = vi.fn();
+      warn = vi.fn();
+      error = vi.fn();
+    },
+  };
+});
+
+import { 
+  MemorySessionManager, 
+  SessionManager, 
+  createMemorySessionManager 
+} from '../src/session.js';
+
+describe('MemorySessionManager', () => {
+  let manager: MemorySessionManager;
+
+  beforeEach(() => {
+    vi.useFakeTimers();
+    manager = new MemorySessionManager({
+      maxHistory: 10,
+      expireMs: 60000, // 1 分钟
+    });
+  });
+
+  afterEach(() => {
+    manager.dispose();
+    vi.useRealTimers();
+  });
+
+  describe('会话创建', () => {
+    it('应该创建新会话', () => {
+      const session = manager.get('user-1');
+      
+      expect(session).toBeDefined();
+      expect(session.id).toBe('user-1');
+      expect(session.messages).toEqual([]);
+    });
+
+    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);
+    });
+  });
+
+  describe('消息历史', () => {
+    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');
+      expect(messages[1].role).toBe('assistant');
+    });
+
+    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');
+      expect(systemMessages.length).toBe(1);
+    });
+  });
+
+  describe('系统提示', () => {
+    it('应该设置系统提示', () => {
+      manager.setSystemPrompt('user-1', '你是一个助手');
+      
+      const messages = manager.getMessages('user-1');
+      expect(messages[0].role).toBe('system');
+      expect(messages[0].content).toBe('你是一个助手');
+    });
+
+    it('应该替换旧的系统提示', () => {
+      manager.setSystemPrompt('user-1', '旧提示');
+      manager.setSystemPrompt('user-1', '新提示');
+      
+      const messages = manager.getMessages('user-1');
+      const systemMessages = messages.filter(m => m.role === 'system');
+      expect(systemMessages.length).toBe(1);
+      expect(systemMessages[0].content).toBe('新提示');
+    });
+  });
+
+  describe('会话清理', () => {
+    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);
+    });
+
+    it('清除不存在的会话应返回 false', () => {
+      expect(manager.clear('nonexistent')).toBe(false);
+    });
+
+    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');
+    });
+  });
+
+  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);
+    });
+  });
+
+  describe('统计信息', () => {
+    it('应该返回正确的统计信息', () => {
+      manager.get('user-1');
+      manager.get('user-2');
+      
+      const stats = manager.getStats();
+      expect(stats.total).toBe(2);
+      expect(stats.active).toBe(2);
+      expect(stats.expired).toBe(0);
+    });
+
+    it('应该返回会话列表', () => {
+      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');
+      expect(sessions).toContain('user-3');
+    });
+  });
+
+  describe('dispose', () => {
+    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);
+    });
+  });
+});
+
+describe('SessionManager', () => {
+  describe('generateId', () => {
+    it('应该生成正确的会话 ID（带频道）', () => {
+      const id = SessionManager.generateId('qq', 'user123', 'channel456');
+      expect(id).toBe('qq:channel456:user123');
+    });
+
+    it('应该生成正确的会话 ID（无频道）', () => {
+      const id = SessionManager.generateId('telegram', 'user123');
+      expect(id).toBe('telegram:user123');
+    });
+  });
+});
+
+describe('createMemorySessionManager', () => {
+  it('应该创建 SessionManager 实例', () => {
+    const manager = createMemorySessionManager({ maxHistory: 50 });
+    expect(manager).toBeDefined();
+    expect(manager.get).toBeDefined();
+    manager.dispose();
+  });
+});