@zhin.js/agent 0.1.0 → 0.1.2

package/src/index.ts

@@ -1,183 +0,0 @@
-/**
- * @zhin.js/agent — AI Agent composition
- * Composes providers, tools, skills from @zhin.js/core into session, ZhinAgent, init.
- */
-
-// Re-export types from core for convenience
-export type {
-  AIConfig,
-  AIProvider,
-  ChatMessage,
-  ContentPart,
-  AgentTool,
-  AgentConfig,
-  AgentResult,
-} from '@zhin.js/core';
-
-export { Agent, createAgent, formatToolTitle } from './agent.js';
-export type { AgentState, AgentEvents } from './agent.js';
-
-export { AIService } from './service.js';
-
-export {
-  SessionManager,
-  MemorySessionManager,
-  DatabaseSessionManager,
-  createMemorySessionManager,
-  createDatabaseSessionManager,
-  AI_SESSION_MODEL,
-} from './session.js';
-
-export {
-  ContextManager,
-  createContextManager,
-  CHAT_MESSAGE_MODEL,
-  CONTEXT_SUMMARY_MODEL,
-} from './context-manager.js';
-export type { ContextConfig, MessageRecord } from './context-manager.js';
-
-export { ZhinAgent } from './zhin-agent/index.js';
-export type { ZhinAgentConfig, OnChunkCallback } from './zhin-agent/index.js';
-
-export { PERM_MAP, DEFAULT_CONFIG as ZHIN_AGENT_DEFAULT_CONFIG, SECTION_SEP } from './zhin-agent/config.js';
-export {
-  checkExecPolicy, applyExecPolicyToTools, resolveExecAllowlist, EXEC_PRESETS,
-  isDangerousCommand, stripEnvVarPrefix, stripSafeWrappers, splitCompoundCommand, extractCommandName,
-  type ExecPolicyResult,
-} from './zhin-agent/exec-policy.js';
-export { collectRelevantTools, toAgentTool } from './zhin-agent/tool-collector.js';
-export { buildRichSystemPrompt, buildContextHint, buildEnhancedPersona, buildUserMessageWithHistory, contentToText } from './zhin-agent/prompt.js';
-export type { RichSystemPromptContext } from './zhin-agent/prompt.js';
-export { createChatHistoryTool, createUserProfileTool, createScheduleFollowUpTool, createSpawnTaskTool } from './zhin-agent/builtin-tools.js';
-
-export {
-  ConversationMemory,
-  AI_MESSAGE_MODEL,
-  AI_SUMMARY_MODEL,
-} from './conversation-memory.js';
-export type { ConversationMemoryConfig } from './conversation-memory.js';
-
-export { UserProfileStore, AI_USER_PROFILE_MODEL } from './user-profile.js';
-
-export { RateLimiter } from './rate-limiter.js';
-export type { RateLimitConfig, RateLimitResult } from './rate-limiter.js';
-
-export { FollowUpManager, AI_FOLLOWUP_MODEL } from './follow-up.js';
-export type { FollowUpRecord, FollowUpSender } from './follow-up.js';
-
-export { SubagentManager } from './subagent.js';
-export type {
-  SubagentOrigin,
-  SubagentResultSender,
-  SpawnOptions,
-  SubagentManagerOptions,
-} from './subagent.js';
-
-export {
-  PersistentCronEngine,
-  readCronJobsFile,
-  writeCronJobsFile,
-  getCronJobsFilePath,
-  generateCronJobId,
-  createCronTools,
-  setCronManager,
-  getCronManager,
-  CRON_JOBS_FILENAME,
-} from './cron-engine.js';
-export type {
-  CronJobRecord,
-  CronRunner,
-  AddCronFn,
-  PersistentCronEngineOptions,
-  CronManager,
-} from './cron-engine.js';
-
-export {
-  estimateTokens,
-  estimateMessagesTokens,
-  splitMessagesByTokenShare,
-  chunkMessagesByMaxTokens,
-  computeAdaptiveChunkRatio,
-  resolveContextWindowTokens,
-  evaluateContextWindowGuard,
-  summarizeWithFallback,
-  summarizeInStages,
-  pruneHistoryForContext,
-  compactSession,
-  DEFAULT_CONTEXT_TOKENS,
-} from './compaction.js';
-export type { ContextWindowSource, ContextWindowInfo, ContextWindowGuardResult, PruneResult } from './compaction.js';
-
-export {
-  loadBootstrapFiles,
-  buildContextFiles,
-  buildBootstrapContextSection,
-  loadSoulPersona,
-  loadToolsGuide,
-  loadAgentsMemory,
-  clearBootstrapCache,
-} from './bootstrap.js';
-export type { BootstrapFile, ContextFile } from './bootstrap.js';
-
-export {
-  registerAIHook,
-  unregisterAIHook,
-  triggerAIHook,
-  createAIHookEvent,
-  clearAIHooks,
-  getRegisteredAIHookKeys,
-} from './hooks.js';
-export type {
-  AIHookEvent,
-  AIHookEventType,
-  AIHookHandler,
-  MessageReceivedEvent,
-  MessageSentEvent,
-  SessionCompactEvent,
-  SessionNewEvent,
-  AgentBootstrapEvent,
-  ToolCallEvent,
-} from './hooks.js';
-
-export { detectTone } from './tone-detector.js';
-export type { Tone } from './tone-detector.js';
-
-export {
-  parseOutput,
-  renderToPlainText,
-  renderToSatori,
-} from './output.js';
-export type {
-  OutputElement,
-  TextElement,
-  ImageElement,
-  AudioElement,
-  VideoElement,
-  CardElement,
-  FileElement,
-  CardField,
-  CardButton,
-} from './output.js';
-
-export {
-  calculatorTool,
-  timeTool,
-  searchTool,
-  codeRunnerTool,
-  httpTool,
-  memoryTool,
-  getBuiltinTools,
-  getAllBuiltinTools,
-} from './tools.js';
-
-export { initAgentModule } from './init.js';
-
-export {
-  MemoryStorageBackend,
-  DatabaseStorageBackend,
-  createSwappableBackend,
-} from './storage.js';
-export type {
-  StorageBackend,
-  DbModel,
-} from './storage.js';

package/src/init/create-zhin-agent.ts

@@ -1,161 +0,0 @@
-/**
- * Create ZhinAgent global brain and wire up sub-systems
- * (follow-up sender, subagent manager, cron engine, scheduler).
- */
-import * as path from 'path';
-import { getPlugin, Scheduler, getScheduler, setScheduler, type MessageType, type SendOptions } from '@zhin.js/core';
-import { ModelRegistry, computeTierScore } from '@zhin.js/ai';
-import { ZhinAgent } from '../zhin-agent/index.js';
-import { createBuiltinTools } from '../builtin-tools.js';
-import { collectPluginSkillSearchRoots } from '../discovery-utils.js';
-import { resolveSkillInstructionMaxChars, DEFAULT_CONFIG } from '../zhin-agent/config.js';
-import { PersistentCronEngine, setCronManager } from '../cron-engine.js';
-import type { AIServiceRefs } from './shared-refs.js';
-
-export function createZhinAgentContext(refs: AIServiceRefs): void {
-  const plugin = getPlugin();
-  const { useContext, root, logger } = plugin;
-
-  useContext('ai', (ai) => {
-    if (!ai.isReady()) {
-      logger.warn('AI Service not ready, ZhinAgent not created');
-      return;
-    }
-
-    const provider = ai.getProvider();
-    const agentConfig = ai.getAgentConfig();
-    const agent = new ZhinAgent(provider, agentConfig);
-    refs.zhinAgent = agent;
-
-    const skillRegistry = root.inject('skill');
-    if (skillRegistry) agent.setSkillRegistry(skillRegistry);
-
-    // Model Registry: discover models and wire to agent
-    const dataDir = path.join(process.cwd(), 'data');
-    const modelRegistry = new ModelRegistry(dataDir);
-    const hadCache = modelRegistry.loadCache();
-    agent.setModelRegistry(modelRegistry);
-    ai.setModelRegistry(modelRegistry);
-    // Discover models in background (don't block startup)
-    (async () => {
-      try {
-        const discovered = await modelRegistry.discover(provider);
-        modelRegistry.saveCache();
-        if (discovered.length > 0) {
-          provider.models = discovered
-            .sort((a, b) => computeTierScore(b.id) - computeTierScore(a.id))
-            .map(m => m.id);
-        }
-        if (hadCache) {
-          logger.debug(`ModelRegistry: refreshed ${discovered.length} models from ${provider.name}`);
-        } else {
-          logger.info(`ModelRegistry: discovered ${discovered.length} models from ${provider.name}`);
-        }
-      } catch (e) {
-        logger.warn(`ModelRegistry: discovery failed for ${provider.name}: ${(e as Error).message}`);
-      }
-    })();
-
-    // Follow-up reminder sender
-    agent.setFollowUpSender(async (record) => {
-      const adapter = root.inject(record.platform) as { sendMessage?: (opts: SendOptions) => Promise<string> } | undefined;
-      if (!adapter || typeof adapter.sendMessage !== 'function') {
-        logger.warn(`[跟进提醒] 找不到适配器: ${record.platform}`);
-        return;
-      }
-      await adapter.sendMessage({
-        context: record.platform,
-        bot: record.bot_id,
-        id: record.scene_id,
-        type: record.scene_type as MessageType,
-        content: `⏰ 定时提醒：${record.message}`,
-      });
-    });
-
-    // Subagent manager for background tasks
-    agent.initSubagentManager(() => {
-      const modelName = provider.models[0] || '';
-      const fullConfig = { ...DEFAULT_CONFIG, ...agentConfig } as Required<import('../zhin-agent/config.js').ZhinAgentConfig>;
-      const zhinTools = createBuiltinTools({
-        plugin,
-        skillInstructionMaxChars: resolveSkillInstructionMaxChars(fullConfig, modelName),
-        pluginSkillRootsResolver: () => collectPluginSkillSearchRoots(root),
-      });
-      return zhinTools.map(zt => {
-        const t = zt.toTool();
-        return {
-          name: t.name,
-          description: t.description,
-          parameters: t.parameters,
-          execute: t.execute as (args: Record<string, any>) => Promise<unknown>,
-          tags: t.tags,
-          keywords: t.keywords,
-        };
-      });
-    });
-    agent.setSubagentSender(async (origin, content) => {
-      const adapter = root.inject(origin.platform) as { sendMessage?: (opts: SendOptions) => Promise<string> } | undefined;
-      if (!adapter || typeof adapter.sendMessage !== 'function') {
-        logger.warn(`[子任务] 找不到适配器: ${origin.platform}`);
-        return;
-      }
-      await adapter.sendMessage({
-        context: origin.platform,
-        bot: origin.botId,
-        id: origin.sceneId,
-        type: origin.sceneType as MessageType,
-        content,
-      });
-    });
-
-    // Persistent cron engine
-    let cronEngine: PersistentCronEngine | null = null;
-    const cronFeature = root.inject('cron') as import('@zhin.js/core').CronFeature | undefined;
-    if (cronFeature && typeof cronFeature.add === 'function') {
-      const addCron: import('../cron-engine.js').AddCronFn = (c) => cronFeature.add(c, 'cron-engine');
-      const runner = async (prompt: string) => {
-        if (!refs.zhinAgent) return;
-        await refs.zhinAgent.process(prompt, {
-          platform: 'cron',
-          senderId: 'system',
-          sceneId: 'cron',
-        });
-      };
-      cronEngine = new PersistentCronEngine({ dataDir, addCron, runner });
-      cronEngine.load();
-      setCronManager({ cronFeature, engine: cronEngine });
-    }
-
-    // Unified scheduler (at/every/cron)
-    const scheduler = new Scheduler({
-      storePath: path.join(dataDir, 'scheduler-jobs.json'),
-      workspace: process.cwd(),
-      onJob: async (job) => {
-        if (!refs.zhinAgent) return;
-        await refs.zhinAgent.process(job.payload.message, {
-          platform: 'cron',
-          senderId: 'system',
-          sceneId: 'scheduler',
-        });
-      },
-    });
-    setScheduler(scheduler);
-    scheduler.start().catch((e) => logger.warn('Scheduler start failed: ' + (e as Error).message));
-
-    logger.debug('ZhinAgent created');
-    return () => {
-      setCronManager(null);
-      if (cronEngine) {
-        cronEngine.unload();
-        cronEngine = null;
-      }
-      const s = getScheduler();
-      if (s) {
-        s.stop();
-        setScheduler(null);
-      }
-      agent.dispose();
-      refs.zhinAgent = null;
-    };
-  });
-}

package/src/init/register-ai-service.ts

@@ -1,53 +0,0 @@
-/**
- * Register AIService as a plugin context.
- */
-import './types.js';
-import { getPlugin, type Plugin } from '@zhin.js/core';
-import type { AIConfig } from '@zhin.js/core';
-import { AIService } from '../service.js';
-import type { AIServiceRefs } from './shared-refs.js';
-
-export function registerAIService(refs: AIServiceRefs): void {
-  const plugin = getPlugin();
-  const { provide, root, logger } = plugin;
-
-  provide<'ai'>({
-    name: 'ai',
-    description: 'AI Service - Multi-model LLM integration',
-    async mounted(_p: Plugin) {
-      const configService = root.inject('config');
-      const appConfig =
-        configService?.getPrimary<{ ai?: AIConfig }>() || {};
-      const config = appConfig.ai || {};
-
-      if (config.enabled === false) {
-        logger.info('AI Service is disabled');
-        return undefined as unknown as AIService;
-      }
-
-      const service = new AIService(config);
-      refs.aiService = service;
-      service.setPlugin(root);
-
-      const providers = service.listProviders();
-      if (providers.length === 0) {
-        logger.warn(
-          'No AI providers configured. Please add API keys in zhin.config (yml/json/toml)',
-        );
-      } else {
-        logger.info(
-          `AI Service started with providers: ${providers.join(', ')}`,
-        );
-      }
-
-      return service;
-    },
-    async dispose(service) {
-      if (service) {
-        service.dispose();
-        refs.aiService = null;
-        logger.info('AI Service stopped');
-      }
-    },
-  });
-}

package/src/init/register-ai-trigger.ts

@@ -1,253 +0,0 @@
-/**
- * Register AI trigger handler via MessageDispatcher or fallback middleware.
- */
-import './types.js';
-import { getPlugin, Message, shouldTriggerAI, inferSenderPermissions, parseRichMediaContent, mergeAITriggerConfig } from '@zhin.js/core';
-import type { Tool, ToolContext } from '@zhin.js/core';
-import type { ContentPart } from '@zhin.js/core';
-import type { OutputElement } from '@zhin.js/ai';
-import type { AIServiceRefs } from './shared-refs.js';
-
-/**
- * Extract multimodal ContentPart[] from a Message's structured $content segments.
- * Handles image, video, audio, and face/sticker types.
- * Falls back to raw string parsing for image URLs when $content has no media segments.
- */
-function extractMediaParts(message: Message<any>): ContentPart[] {
-  const parts: ContentPart[] = [];
-
-  // 1. Extract from structured $content segments
-  if (Array.isArray(message.$content)) {
-    for (const seg of message.$content) {
-      if (typeof seg === 'string' || !seg || !seg.type) continue;
-      const { type, data } = seg;
-      switch (type) {
-        case 'image': {
-          const url = data?.url || data?.file || data?.src;
-          if (url) parts.push({ type: 'image_url', image_url: { url } });
-          break;
-        }
-        case 'video': {
-          const url = data?.url || data?.file || data?.src;
-          if (url) parts.push({ type: 'video_url', video_url: { url } });
-          break;
-        }
-        case 'audio':
-        case 'record':
-        case 'voice': {
-          const dataStr = data?.data || data?.base64;
-          if (dataStr) {
-            const fmt = data?.format === 'wav' ? 'wav' : 'mp3';
-            parts.push({ type: 'audio', audio: { data: dataStr, format: fmt } });
-          } else {
-            const url = data?.url || data?.file || data?.src;
-            if (url) {
-              // Audio URL: describe as text since most LLMs can't play audio URLs directly
-              parts.push({ type: 'text', text: `[用户发送了一段语音: ${url}]` });
-            }
-          }
-          break;
-        }
-        case 'face':
-        case 'sticker':
-        case 'emoji': {
-          const id = String(data?.id ?? data?.face_id ?? '');
-          const text = data?.text || data?.name || data?.describe;
-          if (id) parts.push({ type: 'face', face: { id, text } });
-          break;
-        }
-      }
-    }
-  }
-
-  // 2. Fallback: parse image URLs from $raw for adapters that don't use structured $content
-  if (parts.length === 0) {
-    const raw = typeof message.$raw === 'string' ? message.$raw : JSON.stringify(message.$raw || '');
-
-    const xmlMatches = raw.match(/<image[^>]+url="([^"]+)"/g);
-    if (xmlMatches) {
-      for (const m of xmlMatches) {
-        const urlMatch = m.match(/url="([^"]+)"/);
-        if (urlMatch) parts.push({ type: 'image_url', image_url: { url: urlMatch[1] } });
-      }
-    }
-
-    const cqMatches = raw.match(/\[CQ:image[^\]]*url=([^\],]+)/g);
-    if (cqMatches) {
-      for (const m of cqMatches) {
-        const urlMatch = m.match(/url=([^\],]+)/);
-        if (urlMatch) parts.push({ type: 'image_url', image_url: { url: urlMatch[1] } });
-      }
-    }
-  }
-
-  return parts;
-}
-
-function renderOutput(elements: OutputElement[]): string {
-  const parts: string[] = [];
-  for (const el of elements) {
-    switch (el.type) {
-      case 'text':
-        if (el.content) parts.push(el.content);
-        break;
-      case 'image':
-        parts.push(`<image url="${el.url}"/>`);
-        break;
-      case 'audio':
-        parts.push(`<audio url="${el.url}"/>`);
-        break;
-      case 'video':
-        parts.push(`<video url="${el.url}"/>`);
-        break;
-      case 'card': {
-        const cp = [`📋 ${el.title}`];
-        if (el.description) cp.push(el.description);
-        if (el.fields?.length)
-          for (const f of el.fields) cp.push(`  ${f.label}: ${f.value}`);
-        if (el.imageUrl) cp.push(`<image url="${el.imageUrl}"/>`);
-        parts.push(cp.join('\n'));
-        break;
-      }
-      case 'file':
-        parts.push(`📎 ${el.name}: ${el.url}`);
-        break;
-    }
-  }
-  return parts.join('\n') || '';
-}
-
-export function registerAITrigger(refs: AIServiceRefs): void {
-  const plugin = getPlugin();
-  const { useContext, root, logger } = plugin;
-
-  useContext('ai', (ai) => {
-    const rawConfig = ai.getTriggerConfig();
-    const triggerConfig = mergeAITriggerConfig(rawConfig);
-    if (!triggerConfig.enabled) {
-      logger.info('AI Trigger is disabled');
-      return;
-    }
-
-    const dispatcherSvc = root.inject('dispatcher') as
-      | { replyWithPolish?: (m: Message<any>, s: 'ai' | 'command', c: unknown) => Promise<unknown> }
-      | undefined;
-
-    const handleAIMessage = async (
-      message: Message<any>,
-      content: string,
-    ) => {
-      const replyOutbound = async (payload: unknown) => {
-        if (dispatcherSvc && typeof dispatcherSvc.replyWithPolish === 'function') {
-          return dispatcherSvc.replyWithPolish(message, 'ai', payload as any);
-        }
-        return message.$reply(payload as any);
-      };
-
-      const t0 = performance.now();
-      if (!ai.isReady()) return;
-      if (triggerConfig.thinkingMessage)
-        await replyOutbound(triggerConfig.thinkingMessage);
-
-      const permissions = inferSenderPermissions(message, triggerConfig);
-
-      // 从 bot 配置中查找 owner（bots[].owner）
-      const adapterInstance = root.inject(message.$adapter) as
-        | { bots?: Map<string, { $config?: Record<string, any> }> }
-        | undefined;
-      const botConfig = adapterInstance?.bots?.get(message.$bot)?.$config as Record<string, any> | undefined;
-      const botOwner: string | undefined = botConfig?.owner;
-
-      // 用 bot 级别 owner 覆盖权限判断
-      const isOwner = botOwner ? String(message.$sender.id) === String(botOwner) : permissions.isOwner;
-      const permissionLevel = isOwner ? 'owner' as const : permissions.permissionLevel;
-
-      const toolContext: ToolContext = {
-        platform: message.$adapter,
-        botId: message.$bot,
-        sceneId: message.$channel?.id || message.$sender.id,
-        senderId: message.$sender.id,
-        message,
-        scope: permissions.scope,
-        senderPermissionLevel: permissionLevel,
-        isGroupAdmin: permissions.isGroupAdmin,
-        isGroupOwner: permissions.isGroupOwner,
-        isBotAdmin: isOwner || permissions.isBotAdmin,
-        isOwner,
-      };
-
-      const tCollect = performance.now();
-      const toolService = root.inject('tool');
-      let externalTools: Tool[] = [];
-      if (toolService) {
-        externalTools = toolService.collectAll(root);
-        externalTools = toolService.filterByContext(externalTools, toolContext);
-      }
-      logger.debug(`[AI Handler] 工具收集: ${externalTools.length} 个, ${(performance.now() - tCollect).toFixed(0)}ms`);
-
-      try {
-        const timeout = new Promise<never>((_, rej) =>
-          setTimeout(() => rej(new Error('AI 响应超时')), triggerConfig.timeout),
-        );
-
-        let responseText: string;
-        if (refs.zhinAgent) {
-          const mediaParts = extractMediaParts(message);
-          let elements: OutputElement[];
-          if (mediaParts.length > 0) {
-            const parts: ContentPart[] = [];
-            if (content) parts.push({ type: 'text', text: content });
-            parts.push(...mediaParts);
-            elements = await Promise.race([
-              refs.zhinAgent.processMultimodal(parts, toolContext),
-              timeout,
-            ]);
-          } else {
-            elements = await Promise.race([
-              refs.zhinAgent.process(content, toolContext, externalTools),
-              timeout,
-            ]);
-          }
-          responseText = renderOutput(elements);
-        } else {
-          const response = await Promise.race([
-            ai.process(content, toolContext, externalTools),
-            timeout,
-          ]);
-          responseText = typeof response === 'string' ? response : '';
-        }
-
-        if (responseText) await replyOutbound(parseRichMediaContent(responseText));
-        logger.info(`[AI Handler] 总耗时: ${(performance.now() - t0).toFixed(0)}ms`);
-      } catch (error) {
-        const msg = error instanceof Error ? error.message : String(error);
-        logger.warn(`[AI Handler] 失败 (${(performance.now() - t0).toFixed(0)}ms): ${msg}`);
-        await replyOutbound(triggerConfig.errorTemplate.replace('{error}', msg));
-      }
-    };
-
-    const dispatcher = root.inject('dispatcher');
-
-    if (dispatcher && typeof dispatcher.setAIHandler === 'function') {
-      dispatcher.setAITriggerMatcher((message: Message<any>) =>
-        shouldTriggerAI(message, triggerConfig),
-      );
-      dispatcher.setAIHandler(handleAIMessage);
-      logger.debug('AI Handler registered via MessageDispatcher');
-      return () => { logger.info('AI Handler unregistered'); };
-    }
-
-    const aiMw = async (
-      message: Message<any>,
-      next: () => Promise<void>,
-    ) => {
-      const { triggered, content } = shouldTriggerAI(message, triggerConfig);
-      if (!triggered) return await next();
-      await handleAIMessage(message, content);
-      await next();
-    };
-    const dispose = root.addMiddleware(aiMw);
-    logger.debug('AI Trigger middleware registered (fallback mode)');
-    return () => { dispose(); };
-  });
-}