@zhin.js/agent 0.0.2 → 0.0.4

package/src/storage.ts

@@ -1,135 +1,9 @@
 /**
- * StorageBackend — Unified storage abstraction layer
- *
- * Provides a common interface for AI sub-systems (session, context, memory,
- * user profile, follow-up) to switch between in-memory and database backends
- * without changing business logic.
- *
- * Usage:
- *   const backend = new MemoryStorageBackend<MyRecord>();
- *   // ... later, when DB is ready:
- *   const dbBackend = new DatabaseStorageBackend<MyRecord>(model, { keyField: 'session_id' });
- *   service.upgradeBackend(dbBackend);
- */
-
-export interface StorageBackend<T extends Record<string, any>> {
-  get(key: string): Promise<T | null>;
-  set(key: string, value: T): Promise<void>;
-  delete(key: string): Promise<boolean>;
-  list(filter?: Partial<T>): Promise<T[]>;
-  clear(): Promise<void>;
-  readonly type: 'memory' | 'database';
-}
-
-/**
- * In-memory storage backend.
- */
-export class MemoryStorageBackend<T extends Record<string, any>> implements StorageBackend<T> {
-  readonly type = 'memory' as const;
-  private data = new Map<string, T>();
-
-  async get(key: string): Promise<T | null> {
-    return this.data.get(key) ?? null;
-  }
-
-  async set(key: string, value: T): Promise<void> {
-    this.data.set(key, value);
-  }
-
-  async delete(key: string): Promise<boolean> {
-    return this.data.delete(key);
-  }
-
-  async list(filter?: Partial<T>): Promise<T[]> {
-    const all = Array.from(this.data.values());
-    if (!filter) return all;
-    return all.filter(item => {
-      for (const [k, v] of Object.entries(filter)) {
-        if (item[k] !== v) return false;
-      }
-      return true;
-    });
-  }
-
-  async clear(): Promise<void> {
-    this.data.clear();
-  }
-}
-
-/**
- * Database model interface aligned with @zhin.js/database's RelatedModel API.
- */
-export interface DbModel {
-  select(...fields: string[]): any;
-  create(data: Record<string, any>): Promise<any>;
-  update(data: Partial<any>): any;
-  delete(condition: Record<string, any>): any;
-}
-
-/**
- * Database-backed storage backend.
- */
-export class DatabaseStorageBackend<T extends Record<string, any>> implements StorageBackend<T> {
-  readonly type = 'database' as const;
-
-  constructor(
-    private model: DbModel,
-    private options: {
-      /** The field name used as lookup key (e.g. 'session_id', 'user_id') */
-      keyField: string;
-    },
-  ) {}
-
-  async get(key: string): Promise<T | null> {
-    const rows: T[] = await this.model
-      .select()
-      .where({ [this.options.keyField]: key });
-    return rows[0] ?? null;
-  }
-
-  async set(key: string, value: T): Promise<void> {
-    const existing = await this.get(key);
-    if (existing) {
-      await this.model
-        .update(value)
-        .where({ [this.options.keyField]: key });
-    } else {
-      await this.model.create({ ...value, [this.options.keyField]: key });
-    }
-  }
-
-  async delete(key: string): Promise<boolean> {
-    try {
-      await this.model.delete({ [this.options.keyField]: key });
-      return true;
-    } catch {
-      return false;
-    }
-  }
-
-  async list(filter?: Partial<T>): Promise<T[]> {
-    if (filter) {
-      return this.model.select().where(filter);
-    }
-    return this.model.select();
-  }
-
-  async clear(): Promise<void> {
-    const all = await this.list();
-    for (const item of all) {
-      const key = (item as Record<string, unknown>)[this.options.keyField] as string | undefined;
-      if (key) await this.delete(key);
-    }
-  }
-}
-
-/**
- * Helper to create a swappable backend ref.
- * Call `swap(newBackend)` to atomically upgrade from memory to database.
- */
-export function createSwappableBackend<T extends Record<string, any>>(
-  initial: StorageBackend<T>,
-): { backend: StorageBackend<T>; swap: (next: StorageBackend<T>) => void } {
-  const ref = { backend: initial, swap: (next: StorageBackend<T>) => { ref.backend = next; } };
-  return ref;
-}
+ * Re-export from @zhin.js/ai for backward compatibility.
+ */
+export {
+  MemoryStorageBackend,
+  DatabaseStorageBackend,
+  createSwappableBackend,
+} from '@zhin.js/ai';
+export type { StorageBackend, DbModel } from '@zhin.js/ai';

package/src/tone-detector.ts

@@ -1,89 +1,5 @@
 /**
- * ToneDetector — 轻量级情绪/语气检测
- *
- * 通过标点符号、emoji 密度、关键词分析用户语气，
- * 生成一条 hint 注入 system prompt，让 AI 的回复匹配用户情绪。
- *
- * 零 LLM 开销，纯正则/统计分析。
+ * Re-export from @zhin.js/ai for backward compatibility.
  */
-
-export type Tone = 'neutral' | 'frustrated' | 'excited' | 'questioning' | 'sad' | 'urgent';
-
-interface ToneResult {
-  tone: Tone;
-  hint: string;
-}
-
-// 常见负面情绪词
-const FRUSTRATED_WORDS = /不行|不对|又错|还是不|怎么回事|搞不定|烦死|崩溃|无语|什么鬼|bug|报错|失败|出问题/;
-const SAD_WORDS = /难过|伤心|失落|遗憾|可惜|唉|哎|不开心|郁闷|心累/;
-const URGENT_WORDS = /急|赶紧|马上|立刻|紧急|尽快|快点|asap|hurry/i;
-const EXCITED_WORDS = /太好了|太棒了|厉害|牛|可以|成功|搞定|完美|赞|nice|amazing|awesome|cool/i;
-
-/**
- * 检测用户消息的情绪语气
- */
-export function detectTone(message: string): ToneResult {
-  const len = message.length;
-  if (len === 0) return { tone: 'neutral', hint: '' };
-
-  // 统计特征
-  const exclamations = (message.match(/!/g) || []).length + (message.match(/！/g) || []).length;
-  const questions = (message.match(/\?/g) || []).length + (message.match(/？/g) || []).length;
-  const ellipsis = (message.match(/\.\.\./g) || []).length + (message.match(/…/g) || []).length;
-  const capsRatio = len > 5 ? (message.match(/[A-Z]/g) || []).length / len : 0;
-
-  // emoji 检测（常见 Unicode 范围）
-  const emojiCount = (message.match(/[\u{1F600}-\u{1F64F}\u{1F300}-\u{1F5FF}\u{1F680}-\u{1F6FF}\u{1F1E0}-\u{1F1FF}\u{2600}-\u{26FF}\u{2700}-\u{27BF}]/gu) || []).length;
-
-  // 关键词检测
-  const isFrustrated = FRUSTRATED_WORDS.test(message);
-  const isSad = SAD_WORDS.test(message);
-  const isUrgent = URGENT_WORDS.test(message);
-  const isExcited = EXCITED_WORDS.test(message);
-
-  // 判定优先级: frustrated > urgent > sad > excited > questioning > neutral
-  if (isFrustrated || (exclamations >= 3 && !isExcited)) {
-    return {
-      tone: 'frustrated',
-      hint: '用户似乎有些沮丧或受挫，请用耐心、理解的语气回复，先表示共情再提供帮助。',
-    };
-  }
-
-  if (isUrgent) {
-    return {
-      tone: 'urgent',
-      hint: '用户似乎很着急，请直接给出解决方案，减少寒暄，优先效率。',
-    };
-  }
-
-  if (isSad || ellipsis >= 2) {
-    return {
-      tone: 'sad',
-      hint: '用户的语气似乎有些低落，请用温暖、关心的语气回复。',
-    };
-  }
-
-  if (isExcited || (emojiCount >= 2 && exclamations >= 1)) {
-    return {
-      tone: 'excited',
-      hint: '用户的心情不错，可以用更活泼、热情的语气回复。',
-    };
-  }
-
-  if (questions >= 2 || (questions >= 1 && len < 20)) {
-    return {
-      tone: 'questioning',
-      hint: '', // 提问是正常的，不需要特殊 hint
-    };
-  }
-
-  if (capsRatio > 0.5 && len > 10) {
-    return {
-      tone: 'frustrated',
-      hint: '用户使用了大量大写字母，可能在表达强烈情绪，请注意语气。',
-    };
-  }
-
-  return { tone: 'neutral', hint: '' };
-}
+export { detectTone } from '@zhin.js/ai';
+export type { Tone } from '@zhin.js/ai';

package/tests/ai/setup.ts

@@ -1,17 +1,29 @@
 /**
  * Agent 模块测试环境设置
  * 提供 Mock 与测试辅助，类型与规范从 @zhin.js/core 引用
+ * 通用 AI mocks 从 @zhin.js/ai 测试 setup 导入
  */
 import { vi } from 'vitest';
 import type { Message, MessageElement, Tool, ToolContext } from '@zhin.js/core';
-import type { AIConfig, ChatMessage } from '@zhin.js/core';
-
-export const createMockLogger = () => ({
-  debug: vi.fn(),
-  info: vi.fn(),
-  warn: vi.fn(),
-  error: vi.fn(),
-});
+import type { AIConfig } from '@zhin.js/core';
+
+// Import and re-export generic AI mocks from ai package tests
+import {
+  createMockLogger,
+  createMockProvider,
+  createChatMessage,
+  delay,
+  waitFor,
+  collectAsyncGenerator,
+} from '../../../ai/tests/setup.js';
+export {
+  createMockLogger,
+  createMockProvider,
+  createChatMessage,
+  delay,
+  waitFor,
+  collectAsyncGenerator,
+};
 
 export const createMockPlugin = (name = 'test-plugin') => ({
   name,
@@ -134,54 +146,6 @@ export const createMockTool = (options: MockToolOptions): Tool => {
   return tool;
 };
 
-export interface MockProviderOptions {
-  name?: string;
-  response?: string | AsyncGenerator<string>;
-  toolCalls?: Array<{ name: string; arguments: Record<string, any> }>;
-  error?: Error;
-}
-
-export const createMockProvider = (options: MockProviderOptions = {}) => {
-  const {
-    name = 'mock',
-    response = '这是 AI 的回复',
-    toolCalls = [],
-    error,
-  } = options;
-
-  if (error) {
-    return {
-      name,
-      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()),
-    healthCheck: vi.fn().mockResolvedValue(true),
-  };
-};
-
 export const createToolContext = (options: Partial<ToolContext> = {}): ToolContext => ({
   platform: 'test',
   scope: 'group',
@@ -209,34 +173,6 @@ export const createMockAIConfig = (overrides: Partial<AIConfig> = {}): AIConfig
   ...overrides,
 });
 
-export const waitFor = async <T>(promise: Promise<T>, timeout = 5000): Promise<T> => {
-  return Promise.race([
-    promise,
-    new Promise<T>((_, reject) =>
-      setTimeout(() => reject(new Error('Timeout')), timeout)
-    ),
-  ]);
-};
-
-export const collectAsyncGenerator = async <T>(generator: AsyncGenerator<T>): Promise<T[]> => {
-  const results: T[] = [];
-  for await (const item of generator) {
-    results.push(item);
-  }
-  return results;
-};
-
-export const delay = (ms: number): Promise<void> =>
-  new Promise((resolve) => setTimeout(resolve, ms));
-
-export const createChatMessage = (
-  role: 'user' | 'assistant' | 'system',
-  content: string
-): ChatMessage => ({
-  role,
-  content,
-});
-
 export const assertToolParameters = (
   tool: Tool,
   expectedProperties: string[],