@zhin.js/core 1.0.37 → 1.0.38

package/src/ai/zhin-agent/builtin-tools.ts

@@ -1,247 +0,0 @@
-/**
- * ZhinAgent 内置上下文工具工厂
- *
- * 这些工具依赖运行时上下文（sessionId / userId / context），
- * 由 ZhinAgent.process() 按需创建并注入到工具列表中。
- */
-
-import type { ToolContext } from '../../types.js';
-import type { AgentTool } from '../types.js';
-import type { ConversationMemory } from '../conversation-memory.js';
-import type { UserProfileStore } from '../user-profile.js';
-import type { FollowUpManager } from '../follow-up.js';
-import type { SubagentManager, SubagentOrigin } from '../subagent.js';
-
-export function createChatHistoryTool(sessionId: string, memory: ConversationMemory): AgentTool {
-  return {
-    name: 'chat_history',
-    description: '搜索与用户的历史聊天记录。可以按关键词搜索，也可以按对话轮次范围查询。当用户问到"之前聊过什么""我们讨论过什么"等回忆类问题时使用。',
-    parameters: {
-      type: 'object',
-      properties: {
-        keyword: {
-          type: 'string',
-          description: '搜索关键词（模糊匹配消息内容和摘要）。留空则返回最近几轮记录',
-        },
-        from_round: {
-          type: 'number',
-          description: '起始轮次（与 to_round 配合使用，精确查询某段对话）',
-        },
-        to_round: {
-          type: 'number',
-          description: '结束轮次',
-        },
-      },
-      required: ['keyword'],
-    },
-    tags: ['memory', 'history', '聊天记录', '回忆', '之前'],
-    keywords: ['之前', '历史', '聊过', '讨论过', '记得', '上次', '以前', '回忆'],
-    async execute(args: Record<string, any>) {
-      const { keyword, from_round, to_round } = args;
-      const currentRound = await memory.getCurrentRound(sessionId);
-
-      if (keyword) {
-        const result = await memory.traceByKeyword(sessionId, keyword);
-        const msgs = result.messages.map(m => {
-          const role = m.role === 'user' ? '用户' : '助手';
-          const time = new Date(m.time).toLocaleString('zh-CN');
-          return `[第${m.round}轮 ${time}] ${role}: ${m.content}`;
-        }).join('\n');
-
-        let output = `当前是第 ${currentRound} 轮对话。\n\n`;
-        if (result.summary) {
-          output += `📋 找到相关摘要（覆盖第${result.summary.fromRound}-${result.summary.toRound}轮）：\n${result.summary.summary}\n\n`;
-        }
-        output += msgs ? `💬 相关聊天记录：\n${msgs}` : '未找到包含该关键词的聊天记录。';
-        return output;
-      }
-
-      if (from_round != null && to_round != null) {
-        const messages = await memory.getMessagesByRound(sessionId, from_round, to_round);
-        if (messages.length === 0) {
-          return `第 ${from_round}-${to_round} 轮没有聊天记录。当前是第 ${currentRound} 轮。`;
-        }
-        const msgs = messages.map(m => {
-          const role = m.role === 'user' ? '用户' : '助手';
-          const time = new Date(m.time).toLocaleString('zh-CN');
-          return `[第${m.round}轮 ${time}] ${role}: ${m.content}`;
-        }).join('\n');
-        return `第 ${from_round}-${to_round} 轮聊天记录（当前第 ${currentRound} 轮）：\n${msgs}`;
-      }
-
-      const messages = await memory.getMessagesByRound(
-        sessionId,
-        Math.max(1, currentRound - 4),
-        currentRound,
-      );
-      if (messages.length === 0) {
-        return '暂无聊天记录。';
-      }
-      const msgs = messages.map(m => {
-        const role = m.role === 'user' ? '用户' : '助手';
-        return `[第${m.round}轮] ${role}: ${m.content}`;
-      }).join('\n');
-      return `最近的聊天记录（当前第 ${currentRound} 轮）：\n${msgs}`;
-    },
-  };
-}
-
-export function createUserProfileTool(userId: string, profiles: UserProfileStore): AgentTool {
-  return {
-    name: 'user_profile',
-    description: '读取或保存用户的个人偏好和信息。当用户告诉你他的名字、偏好、兴趣、习惯等个人信息时，用 set 操作保存。当需要了解用户偏好时，用 get 操作读取。',
-    parameters: {
-      type: 'object',
-      properties: {
-        action: {
-          type: 'string',
-          description: '操作类型: get（读取所有偏好）, set（保存偏好）, delete（删除偏好）',
-          enum: ['get', 'set', 'delete'],
-        },
-        key: {
-          type: 'string',
-          description: '偏好名称，如: name, style, interests, timezone, language 等',
-        },
-        value: {
-          type: 'string',
-          description: '偏好值（仅 set 操作需要）',
-        },
-      },
-      required: ['action'],
-    },
-    tags: ['profile', '偏好', '用户', '个性化', '记住'],
-    keywords: ['我叫', '我的名字', '记住我', '我喜欢', '我偏好', '我习惯', '叫我', '我是'],
-    async execute(args: Record<string, any>) {
-      const { action, key, value } = args;
-
-      switch (action) {
-        case 'get': {
-          const all = await profiles.getAll(userId);
-          const entries = Object.entries(all);
-          if (entries.length === 0) return '暂无保存的用户偏好。';
-          return '用户偏好：\n' + entries.map(([k, v]) => `  ${k}: ${v}`).join('\n');
-        }
-        case 'set': {
-          if (!key || !value) return '需要提供 key 和 value';
-          await profiles.set(userId, key, value);
-          return `已保存: ${key} = ${value}`;
-        }
-        case 'delete': {
-          if (!key) return '需要提供 key';
-          const deleted = await profiles.delete(userId, key);
-          return deleted ? `已删除: ${key}` : `未找到偏好: ${key}`;
-        }
-        default:
-          return '不支持的操作，请使用 get/set/delete';
-      }
-    },
-  };
-}
-
-export function createScheduleFollowUpTool(
-  sessionId: string,
-  context: ToolContext,
-  followUps: FollowUpManager,
-): AgentTool {
-  const platform = context.platform || '';
-  const botId = context.botId || '';
-  const senderId = context.senderId || '';
-  const sceneId = context.sceneId || '';
-  const sceneType = (context.message as any)?.$channel?.type || 'private';
-
-  return {
-    name: 'schedule_followup',
-    description: '安排或取消定时跟进提醒。创建新提醒会自动取消之前的提醒。提醒持久保存，重启不丢失。',
-    parameters: {
-      type: 'object',
-      properties: {
-        action: {
-          type: 'string',
-          description: '操作类型: create（创建提醒，默认）或 cancel（取消当前会话所有提醒）',
-          enum: ['create', 'cancel'],
-        },
-        delay_minutes: {
-          type: 'number',
-          description: '延迟时间，单位是分钟。注意：3 就是 3 分钟，不是 3 小时。举例: 3 = 3分钟后, 60 = 1小时后, 1440 = 1天后',
-        },
-        message: {
-          type: 'string',
-          description: '提醒消息内容',
-        },
-      },
-      required: ['action'],
-    },
-    tags: ['reminder', '提醒', '跟进', '定时'],
-    keywords: ['提醒', '提醒我', '过一会', '过一小时', '明天', '跟进', '别忘了', '记得提醒', '取消提醒'],
-    async execute(args: Record<string, any>) {
-      const { action = 'create', delay_minutes, message: msg } = args;
-
-      if (action === 'cancel') {
-        const count = await followUps.cancelBySession(sessionId);
-        return count > 0
-          ? `✅ 已取消 ${count} 个待执行的提醒`
-          : '当前没有待执行的提醒';
-      }
-
-      if (!delay_minutes || delay_minutes <= 0) return '延迟时间必须大于 0 分钟';
-      if (!msg) return '请提供提醒内容';
-
-      return followUps.schedule({
-        sessionId,
-        platform,
-        botId,
-        senderId,
-        sceneId,
-        sceneType,
-        message: msg,
-        delayMinutes: delay_minutes,
-      });
-    },
-  };
-}
-
-export function createSpawnTaskTool(
-  context: ToolContext,
-  manager: SubagentManager,
-): AgentTool {
-  const platform = context.platform || '';
-  const botId = context.botId || '';
-  const senderId = context.senderId || '';
-  const sceneId = context.sceneId || '';
-  const sceneType = (context.message as any)?.$channel?.type || 'private';
-
-  return {
-    name: 'spawn_task',
-    description: '将复杂或耗时的任务交给后台子 agent 异步处理。子 agent 拥有文件读写、Shell、网络搜索等能力，完成后会自动通知用户。适用于需要多步操作的文件处理、代码分析、数据收集等任务。',
-    parameters: {
-      type: 'object',
-      properties: {
-        task: {
-          type: 'string',
-          description: '要交给子 agent 完成的任务描述（尽量详细，包含目标、范围、期望输出）',
-        },
-        label: {
-          type: 'string',
-          description: '任务的简短标签（用于显示，可选）',
-        },
-      },
-      required: ['task'],
-    },
-    tags: ['agent', 'async', 'task', '后台', '子任务'],
-    keywords: ['后台', '异步', '子任务', 'spawn', 'background', '并行', '独立处理'],
-    async execute(args: Record<string, any>) {
-      const { task, label } = args;
-      if (!task) return '请提供任务描述';
-
-      const origin: SubagentOrigin = {
-        platform,
-        botId,
-        sceneId,
-        senderId,
-        sceneType,
-      };
-
-      return manager.spawn({ task, label, origin });
-    },
-  };
-}

package/src/ai/zhin-agent/config.ts

@@ -1,113 +0,0 @@
-/**
- * ZhinAgent 配置、常量、类型定义
- */
-
-import type { RateLimitConfig } from '../rate-limiter.js';
-import { DEFAULT_CONTEXT_TOKENS } from '../compaction.js';
-
-export type ModelSizeHint = 'small' | 'medium' | 'large';
-
-const SMALL_MODEL_RE = /[:\-_](0\.5|1\.?[58]?|[3-8])b\b/i;
-const MEDIUM_MODEL_RE = /[:\-_](14|[12][0-9]|32)b\b/i;
-
-/**
- * Infer model size from model name string.
- * Pattern: `:8b` → small, `:14b` → medium, else large.
- */
-export function inferModelSize(modelName: string): ModelSizeHint {
-  if (SMALL_MODEL_RE.test(modelName)) return 'small';
-  if (MEDIUM_MODEL_RE.test(modelName)) return 'medium';
-  return 'large';
-}
-
-/**
- * Resolve the effective model size hint.
- * Priority: explicit config > model name inference.
- */
-export function resolveModelSize(config: Required<ZhinAgentConfig>, modelName: string): ModelSizeHint {
-  if (config.modelSizeHint && config.modelSizeHint !== ('' as any)) return config.modelSizeHint;
-  return inferModelSize(modelName);
-}
-
-/**
- * Resolve the effective skill instruction max chars based on model size.
- */
-export function resolveSkillInstructionMaxChars(config: Required<ZhinAgentConfig>, modelName: string): number {
-  if (config.skillInstructionMaxChars && config.skillInstructionMaxChars > 0) return config.skillInstructionMaxChars;
-  const size = resolveModelSize(config, modelName);
-  switch (size) {
-    case 'small': return 1500;
-    case 'medium': return 4000;
-    case 'large': return 8000;
-  }
-}
-
-export const SECTION_SEP = '\n\n---\n\n';
-export const HISTORY_CONTEXT_MARKER = '[Chat messages since your last reply - for context]';
-export const CURRENT_MESSAGE_MARKER = '[Current message - respond to this]';
-
-export const PERM_MAP: Record<string, number> = {
-  user: 0,
-  group_admin: 1,
-  group_owner: 2,
-  bot_admin: 3,
-  owner: 4,
-};
-
-export type OnChunkCallback = (chunk: string, full: string) => void;
-
-export interface ZhinAgentConfig {
-  persona?: string;
-  maxIterations?: number;
-  timeout?: number;
-  preExecTimeout?: number;
-  maxSkills?: number;
-  maxTools?: number;
-  minTopicRounds?: number;
-  slidingWindowSize?: number;
-  topicChangeThreshold?: number;
-  rateLimit?: RateLimitConfig;
-  toneAwareness?: boolean;
-  visionModel?: string;
-  contextTokens?: number;
-  maxHistoryShare?: number;
-  disabledTools?: string[];
-  allowedTools?: string[];
-  execSecurity?: 'deny' | 'allowlist' | 'full';
-  execPreset?: 'readonly' | 'network' | 'development' | 'custom';
-  execAllowlist?: string[];
-  execAsk?: boolean;
-  maxSubagentIterations?: number;
-  subagentTools?: string[];
-  /** 模型大小提示，影响技能指令截断长度。留空则根据模型名自动推断 */
-  modelSizeHint?: 'small' | 'medium' | 'large';
-  /** 技能指令最大字符数（覆盖 modelSizeHint 推断值） */
-  skillInstructionMaxChars?: number;
-}
-
-export const DEFAULT_CONFIG: Required<ZhinAgentConfig> = {
-  persona: '你是一个友好的中文 AI 助手，擅长使用工具帮助用户解决问题。',
-  maxIterations: 5,
-  timeout: 60_000,
-  preExecTimeout: 10_000,
-  maxSkills: 3,
-  maxTools: 8,
-  minTopicRounds: 5,
-  slidingWindowSize: 5,
-  topicChangeThreshold: 0.15,
-  rateLimit: {},
-  toneAwareness: true,
-  visionModel: '',
-  contextTokens: DEFAULT_CONTEXT_TOKENS,
-  maxHistoryShare: 0.5,
-  disabledTools: [],
-  allowedTools: [],
-  execSecurity: 'deny',
-  execPreset: 'custom',
-  execAllowlist: [],
-  execAsk: false,
-  maxSubagentIterations: 15,
-  subagentTools: [],
-  modelSizeHint: '' as any,
-  skillInstructionMaxChars: 0,
-};

package/src/ai/zhin-agent/exec-policy.ts

@@ -1,78 +0,0 @@
-/**
- * ZhinAgent 执行策略 — bash 命令的安全检查与工具包装
- */
-
-import type { AgentTool } from '../types.js';
-import type { ZhinAgentConfig } from './config.js';
-
-// ── 预设命令白名单 ──────────────────────────────────────────────────
-
-const PRESET_READONLY = ['ls', 'cat', 'pwd', 'date', 'whoami', 'grep', 'find', 'head', 'tail', 'wc'];
-const PRESET_NETWORK = [...PRESET_READONLY, 'curl', 'wget', 'ping', 'dig'];
-const PRESET_DEVELOPMENT = [...PRESET_NETWORK, 'npm', 'npx', 'node', 'git', 'gh', 'python', 'python3', 'pip', 'pnpm', 'yarn'];
-
-export const EXEC_PRESETS: Record<string, string[]> = {
-  readonly: PRESET_READONLY,
-  network: PRESET_NETWORK,
-  development: PRESET_DEVELOPMENT,
-};
-
-/**
- * Resolves the effective allowlist by merging preset commands with custom allowlist.
- */
-export function resolveExecAllowlist(config: Required<ZhinAgentConfig>): string[] {
-  const preset = config.execPreset;
-  const presetList = (preset && preset !== 'custom') ? (EXEC_PRESETS[preset] ?? []) : [];
-  const custom = config.execAllowlist ?? [];
-  const merged = [...new Set([...presetList, ...custom])];
-  return merged;
-}
-
-/**
- * Check if a bash command is allowed under the current exec policy.
- * Throws an Error when the command is denied.
- */
-export function checkExecPolicy(config: Required<ZhinAgentConfig>, command: string): void {
-  const security = config.execSecurity ?? 'deny';
-  if (security === 'full') return;
-  if (security === 'deny') {
-    throw new Error('当前配置禁止执行 Shell 命令（execSecurity=deny）。如需开放请在配置中设置 ai.agent.execSecurity。');
-  }
-  // allowlist
-  const list = resolveExecAllowlist(config);
-  const cmd = (command || '').trim();
-  const allowed = list.some(pattern => {
-    try {
-      const re = new RegExp(pattern);
-      return re.test(cmd);
-    } catch {
-      return cmd === pattern || cmd.startsWith(pattern);
-    }
-  });
-  if (!allowed) {
-    const ask = config.execAsk;
-    throw new Error(
-      ask
-        ? '该命令不在允许列表中，需要审批后执行。当前版本请将命令加入 ai.agent.execAllowlist 或联系管理员。'
-        : '该命令不在允许列表中，已被拒绝执行。可将允许的命令模式加入 ai.agent.execAllowlist。',
-    );
-  }
-}
-
-/**
- * Wrap `bash` tools with exec policy enforcement.
- */
-export function applyExecPolicyToTools(config: Required<ZhinAgentConfig>, tools: AgentTool[]): AgentTool[] {
-  return tools.map(t => {
-    if (t.name !== 'bash') return t;
-    const original = t.execute;
-    return {
-      ...t,
-      execute: async (args: Record<string, any>) => {
-        const cmd = args?.command != null ? String(args.command) : '';
-        checkExecPolicy(config, cmd);
-        return original(args);
-      },
-    };
-  });
-}