@zhin.js/ai 0.0.2 → 1.0.1

package/tests/context-manager.test.ts

@@ -1,6 +1,6 @@
 /**
  * Context Manager 测试
- * 
+ *
  * 测试上下文管理功能：
  * - 消息记录
  * - 上下文构建
@@ -10,7 +10,7 @@
 import { describe, it, expect, vi, beforeEach } 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,
@@ -23,8 +23,8 @@ vi.mock('@zhin.js/core', async (importOriginal) => {
   };
 });
 
-import { ContextManager, createContextManager, CHAT_MESSAGE_MODEL, CONTEXT_SUMMARY_MODEL } from '../src/context-manager.js';
-import type { MessageRecord } from '../src/context-manager.js';
+import { ContextManager, createContextManager, CHAT_MESSAGE_MODEL, CONTEXT_SUMMARY_MODEL } from '@zhin.js/ai';
+import type { MessageRecord } from '@zhin.js/ai';
 
 describe('ContextManager', () => {
   let manager: ContextManager;
@@ -36,24 +36,60 @@ describe('ContextManager', () => {
     const messageStore: MessageRecord[] = [];
     const summaryStore: any[] = [];
 
+    // 链式查询构建器 mock：select().where({...}).orderBy('field', 'DIR').limit(n)
+    function createChainableMock(store: any[], filterKey?: string) {
+      return vi.fn((..._fields: string[]) => {
+        let whereFilter: any = null;
+        let orderField: string | null = null;
+        let orderDir: string = 'ASC';
+        let limitNum: number | null = null;
+
+        const chain: any = {
+          where(condition: any) {
+            whereFilter = condition;
+            return chain;
+          },
+          orderBy(field: string, dir: string = 'ASC') {
+            orderField = field;
+            orderDir = dir;
+            return chain;
+          },
+          limit(n: number) {
+            limitNum = n;
+            return chain;
+          },
+          then(resolve: (v: any) => any, reject?: (e: any) => any) {
+            try {
+              let results = [...store];
+              if (whereFilter && filterKey) {
+                results = results.filter(r => r[filterKey] === whereFilter[filterKey]);
+              }
+              if (orderField) {
+                results.sort((a, b) =>
+                  orderDir.toUpperCase() === 'DESC'
+                    ? (b[orderField!] ?? 0) - (a[orderField!] ?? 0)
+                    : (a[orderField!] ?? 0) - (b[orderField!] ?? 0)
+                );
+              }
+              if (limitNum !== null) {
+                results = results.slice(0, limitNum);
+              }
+              return resolve(results);
+            } catch (e) {
+              return reject ? reject(e) : Promise.reject(e);
+            }
+          },
+        };
+        return chain;
+      });
+    }
+
     mockMessageModel = {
       create: vi.fn((record: MessageRecord) => {
         messageStore.push({ ...record, id: messageStore.length + 1 });
         return Promise.resolve();
       }),
-      select: vi.fn((where?: any, options?: any) => {
-        let results = [...messageStore];
-        if (where?.scene_id) {
-          results = results.filter(m => m.scene_id === where.scene_id);
-        }
-        if (options?.orderBy?.time === 'desc') {
-          results.sort((a, b) => b.time - a.time);
-        }
-        if (options?.limit) {
-          results = results.slice(0, options.limit);
-        }
-        return Promise.resolve(results);
-      }),
+      select: createChainableMock(messageStore, 'scene_id'),
       delete: vi.fn(() => Promise.resolve(0)),
     };
 
@@ -62,13 +98,7 @@ describe('ContextManager', () => {
         summaryStore.push({ ...record, id: summaryStore.length + 1 });
         return Promise.resolve();
       }),
-      select: vi.fn((where?: any) => {
-        let results = [...summaryStore];
-        if (where?.scene_id) {
-          results = results.filter(s => s.scene_id === where.scene_id);
-        }
-        return Promise.resolve(results);
-      }),
+      select: createChainableMock(summaryStore, 'scene_id'),
     };
 
     manager = new ContextManager(mockMessageModel, mockSummaryModel, {
@@ -142,12 +172,10 @@ describe('ContextManager', () => {
       });
 
       const messages = await manager.getRecentMessages('group-123');
-      
+
       expect(messages.length).toBe(2);
-      expect(mockMessageModel.select).toHaveBeenCalledWith(
-        { scene_id: 'group-123' },
-        expect.objectContaining({ orderBy: { time: 'desc' } })
-      );
+      // 链式 API: select() 无参数调用
+      expect(mockMessageModel.select).toHaveBeenCalled();
     });
 
     it('空场景应返回空数组', async () => {
@@ -157,11 +185,9 @@ describe('ContextManager', () => {
 
     it('应该限制返回数量', async () => {
       const messages = await manager.getRecentMessages('group-123', 10);
-      
-      expect(mockMessageModel.select).toHaveBeenCalledWith(
-        { scene_id: 'group-123' },
-        expect.objectContaining({ limit: 10 })
-      );
+
+      // 链式 API: select() 无参数调用，limit 通过链式传递
+      expect(mockMessageModel.select).toHaveBeenCalled();
     });
   });
 
@@ -179,7 +205,7 @@ describe('ContextManager', () => {
       });
 
       const context = await manager.buildContext('group-123', 'qq');
-      
+
       expect(context.sceneId).toBe('group-123');
       expect(context.platform).toBe('qq');
       expect(context.chatMessages).toBeDefined();
@@ -187,7 +213,7 @@ describe('ContextManager', () => {
 
     it('空场景应返回有效的空上下文', async () => {
       const context = await manager.buildContext('empty-scene', 'qq');
-      
+
       expect(context.sceneId).toBe('empty-scene');
       expect(context.recentMessages).toEqual([]);
       expect(context.summaries).toEqual([]);
@@ -211,7 +237,7 @@ describe('ContextManager', () => {
       ];
 
       const chatMessages = manager.formatToChatMessages([], messages);
-      
+
       expect(chatMessages.length).toBe(1);
       expect(chatMessages[0].role).toBe('user');
       expect(chatMessages[0].content).toContain('张三');
@@ -223,7 +249,7 @@ describe('ContextManager', () => {
       const messages: MessageRecord[] = [];
 
       const chatMessages = manager.formatToChatMessages(summaries, messages);
-      
+
       expect(chatMessages.length).toBe(1);
       expect(chatMessages[0].role).toBe('system');
       expect(chatMessages[0].content).toContain('这是之前的总结');
@@ -245,7 +271,7 @@ describe('ContextManager', () => {
       ];
 
       const chatMessages = manager.formatToChatMessages([], messages);
-      
+
       expect(chatMessages[0].role).toBe('assistant');
     });
   });
@@ -254,7 +280,7 @@ describe('ContextManager', () => {
     it('应该估算中文文本的 token 数量', () => {
       const text = '你好世界'; // 4 个中文字符
       const tokens = manager.estimateTokens(text);
-      
+
       // 中文约 2 tokens/字
       expect(tokens).toBeGreaterThanOrEqual(8);
     });
@@ -262,7 +288,7 @@ describe('ContextManager', () => {
     it('应该估算英文文本的 token 数量', () => {
       const text = 'hello world'; // 11 个字符
       const tokens = manager.estimateTokens(text);
-      
+
       // 英文约 0.25 tokens/字符
       expect(tokens).toBeGreaterThanOrEqual(2);
     });
@@ -270,7 +296,7 @@ describe('ContextManager', () => {
     it('应该估算混合文本的 token 数量', () => {
       const text = 'Hello 世界';
       const tokens = manager.estimateTokens(text);
-      
+
       expect(tokens).toBeGreaterThan(0);
     });
   });
@@ -325,9 +351,9 @@ describe('ContextManager', () => {
       };
 
       manager.setAIProvider(mockProvider as any);
-      
+
       const result = await manager.summarize('group-123');
-      
+
       expect(mockProvider.chat).toHaveBeenCalled();
     });
   });
@@ -346,14 +372,14 @@ describe('ContextManager', () => {
       });
 
       const stats = await manager.getSceneStats('group-123');
-      
+
       expect(stats.messageCount).toBeGreaterThan(0);
       expect(stats.summaryCount).toBe(0);
     });
 
     it('空场景应返回零统计', async () => {
       const stats = await manager.getSceneStats('empty');
-      
+
       expect(stats.messageCount).toBe(0);
       expect(stats.summaryCount).toBe(0);
     });
@@ -381,7 +407,7 @@ describe('createContextManager', () => {
     const manager = createContextManager(mockModel, mockModel, {
       maxRecentMessages: 50,
     });
-    
+
     expect(manager).toBeInstanceOf(ContextManager);
   });
 });

package/tests/conversation-memory.test.ts

@@ -0,0 +1,128 @@
+/**
+ * ConversationMemory 测试
+ */
+import { describe, it, expect, beforeEach, afterEach } from 'vitest';
+import { ConversationMemory } from '@zhin.js/ai';
+
+describe('ConversationMemory（内存模式）', () => {
+  let memory: ConversationMemory;
+
+  beforeEach(() => {
+    memory = new ConversationMemory({
+      minTopicRounds: 5,
+      slidingWindowSize: 3,
+      topicChangeThreshold: 0.15,
+    });
+  });
+
+  afterEach(() => {
+    memory.dispose();
+  });
+
+  describe('saveRound / buildContext', () => {
+    it('应保存对话并构建上下文', async () => {
+      await memory.saveRound('s1', '你好', '你好呀！');
+      await memory.saveRound('s1', '天气怎么样', '今天晴天');
+
+      const context = await memory.buildContext('s1');
+      expect(context.length).toBeGreaterThan(0);
+      expect(context.some(m => m.content === '你好')).toBe(true);
+      expect(context.some(m => m.content === '今天晴天')).toBe(true);
+    });
+
+    it('空会话应返回空上下文', async () => {
+      const context = await memory.buildContext('nonexistent');
+      expect(context).toEqual([]);
+    });
+
+    it('轮次应递增', async () => {
+      await memory.saveRound('s1', '第一轮', '回复1');
+      await memory.saveRound('s1', '第二轮', '回复2');
+      await memory.saveRound('s1', '第三轮', '回复3');
+
+      const round = await memory.getCurrentRound('s1');
+      expect(round).toBe(3);
+    });
+  });
+
+  describe('滑动窗口', () => {
+    it('应限制上下文到窗口大小', async () => {
+      // slidingWindowSize = 3
+      await memory.saveRound('s1', 'm1', 'r1');
+      await memory.saveRound('s1', 'm2', 'r2');
+      await memory.saveRound('s1', 'm3', 'r3');
+      await memory.saveRound('s1', 'm4', 'r4');
+      await memory.saveRound('s1', 'm5', 'r5');
+
+      const context = await memory.buildContext('s1');
+      // 窗口大小 3 → 最多 6 条消息 (3轮 × 2)
+      expect(context.length).toBeLessThanOrEqual(6);
+      // 不应包含最早的消息
+      expect(context.some(m => m.content === 'm1')).toBe(false);
+    });
+  });
+
+  describe('searchMessages', () => {
+    it('应能搜索消息', async () => {
+      await memory.saveRound('s1', '今天去打篮球', '好主意');
+      await memory.saveRound('s1', '明天去游泳', '注意安全');
+
+      const results = await memory.searchMessages('s1', '篮球');
+      expect(results.length).toBeGreaterThan(0);
+      expect(results[0].content).toContain('篮球');
+    });
+
+    it('搜索不到应返回空数组', async () => {
+      await memory.saveRound('s1', '你好', '你好');
+      const results = await memory.searchMessages('s1', '不存在的内容xyz');
+      expect(results).toHaveLength(0);
+    });
+  });
+
+  describe('getMessagesByRound', () => {
+    it('应按轮次范围获取消息', async () => {
+      await memory.saveRound('s1', 'm1', 'r1');
+      await memory.saveRound('s1', 'm2', 'r2');
+      await memory.saveRound('s1', 'm3', 'r3');
+
+      const results = await memory.getMessagesByRound('s1', 1, 2);
+      // 第 1-2 轮，每轮 2 条
+      expect(results.length).toBe(4);
+    });
+  });
+
+  describe('traceByKeyword', () => {
+    it('无摘要时应直接搜索消息', async () => {
+      await memory.saveRound('s1', '讨论编程', '很好');
+      await memory.saveRound('s1', '继续编程', '继续');
+
+      const result = await memory.traceByKeyword('s1', '编程');
+      expect(result.summary).toBeNull();
+      expect(result.messages.length).toBeGreaterThan(0);
+    });
+  });
+
+  describe('不同会话隔离', () => {
+    it('不同 sessionId 的数据应隔离', async () => {
+      await memory.saveRound('s1', '会话1', '回复1');
+      await memory.saveRound('s2', '会话2', '回复2');
+
+      const ctx1 = await memory.buildContext('s1');
+      const ctx2 = await memory.buildContext('s2');
+
+      expect(ctx1.some(m => m.content === '会话1')).toBe(true);
+      expect(ctx1.some(m => m.content === '会话2')).toBe(false);
+      expect(ctx2.some(m => m.content === '会话2')).toBe(true);
+    });
+  });
+
+  describe('dispose', () => {
+    it('应清理所有数据', async () => {
+      await memory.saveRound('s1', '测试', '回复');
+      memory.dispose();
+
+      const context = await memory.buildContext('s1');
+      expect(context).toEqual([]);
+    });
+  });
+});

package/tests/output.test.ts

@@ -0,0 +1,128 @@
+/**
+ * AI Output 模块测试
+ */
+import { describe, it, expect } from 'vitest';
+import { parseOutput, renderToPlainText, renderToSatori } from '@zhin.js/ai';
+
+describe('parseOutput', () => {
+  it('纯文本应返回单个 TextElement', () => {
+    const result = parseOutput('Hello, world!');
+    expect(result).toHaveLength(1);
+    expect(result[0]).toMatchObject({ type: 'text', content: 'Hello, world!', format: 'markdown' });
+  });
+
+  it('空字符串应返回空白 TextElement', () => {
+    const result = parseOutput('');
+    expect(result).toHaveLength(1);
+    expect(result[0]).toMatchObject({ type: 'text', content: '', format: 'plain' });
+  });
+
+  it('应解析图片 ![alt](url)', () => {
+    const result = parseOutput('看看这张图 ![cat](https://example.com/cat.png)');
+    expect(result).toHaveLength(2);
+    expect(result[0]).toMatchObject({ type: 'text', content: '看看这张图' });
+    expect(result[1]).toMatchObject({ type: 'image', url: 'https://example.com/cat.png', alt: 'cat' });
+  });
+
+  it('应解析音频 [audio](url)', () => {
+    const result = parseOutput('[audio](https://example.com/song.mp3)');
+    expect(result).toHaveLength(1);
+    expect(result[0]).toMatchObject({ type: 'audio', url: 'https://example.com/song.mp3' });
+  });
+
+  it('应解析视频 [video](url)', () => {
+    const result = parseOutput('这是视频 [video](https://example.com/video.mp4)');
+    expect(result).toHaveLength(2);
+    expect(result[1]).toMatchObject({ type: 'video', url: 'https://example.com/video.mp4' });
+  });
+
+  it('应解析文件 [file:name](url)', () => {
+    const result = parseOutput('[file:report.pdf](https://example.com/report.pdf)');
+    expect(result).toHaveLength(1);
+    expect(result[0]).toMatchObject({ type: 'file', url: 'https://example.com/report.pdf', name: 'report.pdf' });
+  });
+
+  it('应解析 card 代码块', () => {
+    const raw = '```card\n{"title":"天气","description":"晴天"}\n```';
+    const result = parseOutput(raw);
+    expect(result.some(el => el.type === 'card')).toBe(true);
+    const card = result.find(el => el.type === 'card')!;
+    expect(card).toMatchObject({ type: 'card', title: '天气', description: '晴天' });
+  });
+
+  it('无效 card JSON 应降级为文本', () => {
+    const raw = '```card\n{invalid json}\n```';
+    const result = parseOutput(raw);
+    expect(result.some(el => el.type === 'text')).toBe(true);
+  });
+
+  it('应处理混合内容', () => {
+    const raw = '看看这个 ![](https://img.png) 更多文字';
+    const result = parseOutput(raw);
+    expect(result.length).toBeGreaterThanOrEqual(2);
+    expect(result.some(el => el.type === 'image')).toBe(true);
+  });
+});
+
+describe('renderToPlainText', () => {
+  it('应渲染文本元素', () => {
+    expect(renderToPlainText([{ type: 'text', content: 'Hello' }])).toBe('Hello');
+  });
+
+  it('应渲染图片为 [图片: alt]', () => {
+    const result = renderToPlainText([{ type: 'image', url: 'url', alt: '猫' }]);
+    expect(result).toBe('[图片: 猫]');
+  });
+
+  it('应渲染音频的 fallbackText', () => {
+    const result = renderToPlainText([{ type: 'audio', url: 'url', fallbackText: '一段音乐' }]);
+    expect(result).toBe('一段音乐');
+  });
+
+  it('应渲染卡片标题和字段', () => {
+    const result = renderToPlainText([{
+      type: 'card',
+      title: '天气',
+      description: '晴天',
+      fields: [{ label: '温度', value: '25°C' }],
+    }]);
+    expect(result).toContain('天气');
+    expect(result).toContain('晴天');
+    expect(result).toContain('温度: 25°C');
+  });
+
+  it('应渲染文件为 [文件: name]', () => {
+    const result = renderToPlainText([{ type: 'file', url: 'url', name: 'report.pdf' }]);
+    expect(result).toContain('[文件: report.pdf]');
+  });
+});
+
+describe('renderToSatori', () => {
+  it('应将文本包裹在 <p> 标签中', () => {
+    const result = renderToSatori([{ type: 'text', content: 'Hello' }]);
+    expect(result).toBe('<p>Hello</p>');
+  });
+
+  it('应生成 <img> 标签', () => {
+    const result = renderToSatori([{ type: 'image', url: 'https://img.png', alt: '猫' }]);
+    expect(result).toContain('<img');
+    expect(result).toContain('alt="猫"');
+  });
+
+  it('应转义 HTML 特殊字符', () => {
+    const result = renderToSatori([{ type: 'text', content: '<script>alert("xss")</script>' }]);
+    expect(result).not.toContain('<script>');
+    expect(result).toContain('&lt;script&gt;');
+  });
+
+  it('应渲染卡片结构', () => {
+    const result = renderToSatori([{
+      type: 'card',
+      title: '标题',
+      description: '描述',
+    }]);
+    expect(result).toContain('<div class="card">');
+    expect(result).toContain('<h3>标题</h3>');
+    expect(result).toContain('<p>描述</p>');
+  });
+});