@zhin.js/agent 0.0.1 → 0.0.3

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

@@ -95,7 +95,7 @@ export function createZhinAgentContext(refs: AIServiceRefs): void {
       setCronManager({ cronFeature, engine: cronEngine });
     }
 
-    // Unified scheduler (at/every/cron + Heartbeat)
+    // Unified scheduler (at/every/cron)
     const dataDir = path.join(process.cwd(), 'data');
     const scheduler = new Scheduler({
       storePath: path.join(dataDir, 'scheduler-jobs.json'),
@@ -108,8 +108,6 @@ export function createZhinAgentContext(refs: AIServiceRefs): void {
           sceneId: 'scheduler',
         });
       },
-      heartbeatEnabled: true,
-      heartbeatIntervalMs: 30 * 60 * 1000,
     });
     setScheduler(scheduler);
     scheduler.start().catch((e) => logger.warn('Scheduler start failed: ' + (e as Error).message));

package/src/init/register-management-tools.ts

@@ -12,7 +12,7 @@ export function registerManagementTools(): void {
   useContext('ai', 'tool', (ai, toolService) => {
     if (!ai || !toolService) return;
 
-    const listModelsTool = new ZhinTool('ai.models')
+    const listModelsTool = new ZhinTool('ai_models')
       .desc('列出所有可用的 AI 模型')
       .keyword('模型', '可用模型', 'ai模型', 'model', 'models')
       .tag('ai', 'management')
@@ -30,7 +30,7 @@ export function registerManagementTools(): void {
         return r;
       });
 
-    const clearSessionTool = new ZhinTool('ai.clear')
+    const clearSessionTool = new ZhinTool('ai_clear')
       .desc('清除当前对话的历史记录')
       .keyword('清除', '清空', '重置', 'clear', 'reset')
       .tag('ai', 'session')
@@ -47,7 +47,7 @@ export function registerManagementTools(): void {
         return '✅ 对话历史已清除';
       });
 
-    const healthCheckTool = new ZhinTool('ai.health')
+    const healthCheckTool = new ZhinTool('ai_health')
       .desc('检查 AI 服务的健康状态')
       .keyword('健康', '状态', '检查', 'health', 'status')
       .tag('ai', 'management')

package/src/output.ts

@@ -1,261 +1,15 @@
 /**
- * @zhin.js/ai - Multi-modal Output Pipeline
- *
- * 统一输出类型系统：AI 回复 → OutputElement[] → Adapter 渲染
- *
- * 支持的输出类型:
- *   - text:  纯文本 / markdown
- *   - image: 图片 URL 或 base64
- *   - audio: 音频 URL 或 base64
- *   - video: 视频 URL
- *   - card:  结构化卡片 (标题 + 正文 + 字段 + 按钮)
- *   - file:  文件附件
- */
-
-// ============================================================================
-// OutputElement 类型定义
-// ============================================================================
-
-export interface TextElement {
-  type: 'text';
-  content: string;
-  /** markdown / plain */
-  format?: 'markdown' | 'plain';
-}
-
-export interface ImageElement {
-  type: 'image';
-  url: string;
-  /** base64 数据 (url 不可用时的后备) */
-  base64?: string;
-  alt?: string;
-  width?: number;
-  height?: number;
-}
-
-export interface AudioElement {
-  type: 'audio';
-  url: string;
-  base64?: string;
-  duration?: number;
-  /** 语音转文字的后备文本 */
-  fallbackText?: string;
-}
-
-export interface VideoElement {
-  type: 'video';
-  url: string;
-  coverUrl?: string;
-  duration?: number;
-  fallbackText?: string;
-}
-
-export interface CardField {
-  label: string;
-  value: string;
-  inline?: boolean;
-}
-
-export interface CardButton {
-  text: string;
-  /** 点击后发送的命令 */
-  command?: string;
-  /** 点击后打开的 URL */
-  url?: string;
-}
-
-export interface CardElement {
-  type: 'card';
-  title: string;
-  description?: string;
-  fields?: CardField[];
-  imageUrl?: string;
-  buttons?: CardButton[];
-  color?: string;
-}
-
-export interface FileElement {
-  type: 'file';
-  url: string;
-  name: string;
-  size?: number;
-  mimeType?: string;
-}
-
-export type OutputElement =
-  | TextElement
-  | ImageElement
-  | AudioElement
-  | VideoElement
-  | CardElement
-  | FileElement;
-
-// ============================================================================
-// 输出解析器 — 将 AI 原始回复转换为 OutputElement[]
-// ============================================================================
-
-/**
- * 从 AI 回复文本中解析出结构化的 OutputElement[]
- *
- * 识别规则:
- *   - ![alt](url)            → ImageElement
- *   - [audio](url)           → AudioElement
- *   - [video](url)           → VideoElement
- *   - ```card ... ```        → CardElement (JSON)
- *   - [file:name](url)       → FileElement
- *   - 其余文本               → TextElement
- */
-export function parseOutput(raw: string): OutputElement[] {
-  if (!raw || !raw.trim()) return [{ type: 'text', content: '', format: 'plain' }];
-
-  const elements: OutputElement[] = [];
-  let remaining = raw;
-
-  // eslint-disable-next-line no-constant-condition
-  while (remaining.length > 0) {
-    // ── 尝试匹配 card 代码块 ──
-    const cardMatch = remaining.match(/^([\s\S]*?)```card\s*\n([\s\S]*?)```/);
-    if (cardMatch) {
-      // 前缀文本
-      if (cardMatch[1].trim()) {
-        elements.push({ type: 'text', content: cardMatch[1].trim(), format: 'markdown' });
-      }
-      // 解析卡片 JSON
-      try {
-        const cardData = JSON.parse(cardMatch[2]);
-        elements.push({ ...cardData, type: 'card' });
-      } catch {
-        elements.push({ type: 'text', content: cardMatch[2], format: 'plain' });
-      }
-      remaining = remaining.slice(cardMatch[0].length);
-      continue;
-    }
-
-    // ── 尝试匹配 image: ![alt](url) ──
-    const imgMatch = remaining.match(/^([\s\S]*?)!\[([^\]]*)\]\(([^)]+)\)/);
-    if (imgMatch && imgMatch[1].length < 500) {
-      if (imgMatch[1].trim()) {
-        elements.push({ type: 'text', content: imgMatch[1].trim(), format: 'markdown' });
-      }
-      elements.push({ type: 'image', url: imgMatch[3], alt: imgMatch[2] || undefined });
-      remaining = remaining.slice(imgMatch[0].length);
-      continue;
-    }
-
-    // ── 尝试匹配 audio: [audio](url) ──
-    const audioMatch = remaining.match(/^([\s\S]*?)\[audio\]\(([^)]+)\)/i);
-    if (audioMatch && audioMatch[1].length < 500) {
-      if (audioMatch[1].trim()) {
-        elements.push({ type: 'text', content: audioMatch[1].trim(), format: 'markdown' });
-      }
-      elements.push({ type: 'audio', url: audioMatch[2] });
-      remaining = remaining.slice(audioMatch[0].length);
-      continue;
-    }
-
-    // ── 尝试匹配 video: [video](url) ──
-    const videoMatch = remaining.match(/^([\s\S]*?)\[video\]\(([^)]+)\)/i);
-    if (videoMatch && videoMatch[1].length < 500) {
-      if (videoMatch[1].trim()) {
-        elements.push({ type: 'text', content: videoMatch[1].trim(), format: 'markdown' });
-      }
-      elements.push({ type: 'video', url: videoMatch[2] });
-      remaining = remaining.slice(videoMatch[0].length);
-      continue;
-    }
-
-    // ── 尝试匹配 file: [file:name](url) ──
-    const fileMatch = remaining.match(/^([\s\S]*?)\[file:([^\]]+)\]\(([^)]+)\)/i);
-    if (fileMatch && fileMatch[1].length < 500) {
-      if (fileMatch[1].trim()) {
-        elements.push({ type: 'text', content: fileMatch[1].trim(), format: 'markdown' });
-      }
-      elements.push({ type: 'file', url: fileMatch[3], name: fileMatch[2] });
-      remaining = remaining.slice(fileMatch[0].length);
-      continue;
-    }
-
-    // ── 无匹配 → 全部作为文本 ──
-    elements.push({ type: 'text', content: remaining.trim(), format: 'markdown' });
-    break;
-  }
-
-  return elements.length > 0 ? elements : [{ type: 'text', content: raw, format: 'plain' }];
-}
-
-// ============================================================================
-// OutputElement 渲染工具
-// ============================================================================
-
-/**
- * 将 OutputElement[] 降级为纯文本（用于不支持富媒体的平台）
- */
-export function renderToPlainText(elements: OutputElement[]): string {
-  return elements.map(el => {
-    switch (el.type) {
-      case 'text':
-        return el.content;
-      case 'image':
-        return el.alt ? `[图片: ${el.alt}]` : `[图片: ${el.url}]`;
-      case 'audio':
-        return el.fallbackText || `[音频: ${el.url}]`;
-      case 'video':
-        return el.fallbackText || `[视频: ${el.url}]`;
-      case 'card': {
-        const parts = [el.title];
-        if (el.description) parts.push(el.description);
-        if (el.fields?.length) {
-          for (const f of el.fields) parts.push(`${f.label}: ${f.value}`);
-        }
-        return parts.join('\n');
-      }
-      case 'file':
-        return `[文件: ${el.name}] ${el.url}`;
-      default:
-        return '';
-    }
-  }).filter(Boolean).join('\n');
-}
-
-/**
- * 将 OutputElement[] 渲染为 Satori 兼容的 HTML 片段
- */
-export function renderToSatori(elements: OutputElement[]): string {
-  return elements.map(el => {
-    switch (el.type) {
-      case 'text':
-        return `<p>${escapeHtml(el.content)}</p>`;
-      case 'image':
-        return `<img src="${escapeHtml(el.url)}"${el.alt ? ` alt="${escapeHtml(el.alt)}"` : ''}/>`;
-      case 'audio':
-        return el.fallbackText
-          ? `<p>${escapeHtml(el.fallbackText)}</p>`
-          : `<audio src="${escapeHtml(el.url)}"/>`;
-      case 'video':
-        return el.fallbackText
-          ? `<p>${escapeHtml(el.fallbackText)}</p>`
-          : `<video src="${escapeHtml(el.url)}"/>`;
-      case 'card': {
-        let html = `<div class="card">`;
-        html += `<h3>${escapeHtml(el.title)}</h3>`;
-        if (el.description) html += `<p>${escapeHtml(el.description)}</p>`;
-        if (el.imageUrl) html += `<img src="${escapeHtml(el.imageUrl)}"/>`;
-        if (el.fields?.length) {
-          for (const f of el.fields) {
-            html += `<p><b>${escapeHtml(f.label)}</b>: ${escapeHtml(f.value)}</p>`;
-          }
-        }
-        html += `</div>`;
-        return html;
-      }
-      case 'file':
-        return `<a href="${escapeHtml(el.url)}">${escapeHtml(el.name)}</a>`;
-      default:
-        return '';
-    }
-  }).filter(Boolean).join('\n');
-}
-
-function escapeHtml(s: string): string {
-  return s.replace(/&/g, '&amp;').replace(/</g, '&lt;').replace(/>/g, '&gt;').replace(/"/g, '&quot;');
-}
+ * Re-export from @zhin.js/ai for backward compatibility.
+ */
+export { parseOutput, renderToPlainText, renderToSatori } from '@zhin.js/ai';
+export type {
+  TextElement,
+  ImageElement,
+  AudioElement,
+  VideoElement,
+  CardField,
+  CardButton,
+  CardElement,
+  FileElement,
+  OutputElement,
+} from '@zhin.js/ai';

package/src/rate-limiter.ts

@@ -1,129 +1,5 @@
 /**
- * RateLimiter — AI 请求频率限制
- *
- * 防止单用户过度消耗 GPU/API 资源。
- *
- * 策略:
- *   1. 滑动窗口速率限制 — 每分钟 N 次请求
- *   2. 优雅降级 — 超限时返回友好提示而非静默丢弃
+ * Re-export from @zhin.js/ai for backward compatibility.
  */
-
-import { Logger } from '@zhin.js/core';
-
-const logger = new Logger(null, 'RateLimiter');
-
-// ============================================================================
-// 配置
-// ============================================================================
-
-export interface RateLimitConfig {
-  /** 每分钟最大请求数（默认 20） */
-  maxRequestsPerMinute?: number;
-  /** 冷却时间（秒），超限后需等待（默认 10） */
-  cooldownSeconds?: number;
-}
-
-const DEFAULTS: Required<RateLimitConfig> = {
-  maxRequestsPerMinute: 20,
-  cooldownSeconds: 10,
-};
-
-// ============================================================================
-// 类型
-// ============================================================================
-
-interface UserBucket {
-  /** 最近请求的时间戳列表（ms） */
-  timestamps: number[];
-  /** 冷却结束时间（ms），0 表示无冷却 */
-  cooldownUntil: number;
-}
-
-export interface RateLimitResult {
-  allowed: boolean;
-  /** 如果被拒绝，返回友好提示 */
-  message?: string;
-  /** 需等待的秒数 */
-  retryAfterSeconds?: number;
-}
-
-// ============================================================================
-// RateLimiter
-// ============================================================================
-
-export class RateLimiter {
-  private config: Required<RateLimitConfig>;
-  private buckets: Map<string, UserBucket> = new Map();
-  private cleanupTimer?: ReturnType<typeof setInterval>;
-
-  constructor(config?: RateLimitConfig) {
-    this.config = { ...DEFAULTS, ...config };
-    // 定期清理过期的 bucket（每 5 分钟）
-    this.cleanupTimer = setInterval(() => this.cleanup(), 5 * 60 * 1000);
-  }
-
-  /**
-   * 检查请求是否被允许
-   */
-  check(userId: string): RateLimitResult {
-    const now = Date.now();
-    let bucket = this.buckets.get(userId);
-
-    if (!bucket) {
-      bucket = { timestamps: [], cooldownUntil: 0 };
-      this.buckets.set(userId, bucket);
-    }
-
-    // 1. 检查冷却期
-    if (bucket.cooldownUntil > now) {
-      const waitSec = Math.ceil((bucket.cooldownUntil - now) / 1000);
-      return {
-        allowed: false,
-        message: `请稍等 ${waitSec} 秒后再发消息哦～消息太频繁啦 😊`,
-        retryAfterSeconds: waitSec,
-      };
-    }
-
-    // 2. 滑动窗口：清理 1 分钟前的时间戳
-    const windowStart = now - 60_000;
-    bucket.timestamps = bucket.timestamps.filter(t => t > windowStart);
-
-    // 3. 检查速率
-    if (bucket.timestamps.length >= this.config.maxRequestsPerMinute) {
-      bucket.cooldownUntil = now + this.config.cooldownSeconds * 1000;
-      const waitSec = this.config.cooldownSeconds;
-      logger.warn(`User ${userId} rate limited: ${bucket.timestamps.length} requests in 1 min`);
-      return {
-        allowed: false,
-        message: `你发消息太快啦，请等 ${waitSec} 秒后再试～`,
-        retryAfterSeconds: waitSec,
-      };
-    }
-
-    // 4. 记录这次请求
-    bucket.timestamps.push(now);
-    return { allowed: true };
-  }
-
-  /**
-   * 清理长期不活跃的 bucket
-   */
-  private cleanup(): void {
-    const now = Date.now();
-    const staleThreshold = 10 * 60 * 1000; // 10 分钟
-    for (const [userId, bucket] of this.buckets) {
-      const latest = bucket.timestamps[bucket.timestamps.length - 1] ?? 0;
-      if (now - latest > staleThreshold && bucket.cooldownUntil < now) {
-        this.buckets.delete(userId);
-      }
-    }
-  }
-
-  dispose(): void {
-    if (this.cleanupTimer) {
-      clearInterval(this.cleanupTimer);
-      this.cleanupTimer = undefined;
-    }
-    this.buckets.clear();
-  }
-}
+export { RateLimiter } from '@zhin.js/ai';
+export type { RateLimitConfig, RateLimitResult } from '@zhin.js/ai';