@zhin.js/ai 1.0.0 → 1.0.2

package/src/output.ts

@@ -0,0 +1,261 @@
+/**
+ * @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;');
+}

package/src/providers/anthropic.ts

@@ -178,14 +178,18 @@ export class AnthropicProvider extends BaseProvider {
     'claude-3-sonnet-20240229',
     'claude-3-haiku-20240307',
   ];
+  contextWindow: number;
+  capabilities = { vision: true, streaming: true, toolCalling: true, thinking: false };
 
   private baseUrl: string;
   private anthropicVersion: string;
 
   constructor(config: AnthropicConfig = {}) {
     super(config);
+    this.contextWindow = config.contextWindow ?? 200000;
     this.baseUrl = config.baseUrl || 'https://api.anthropic.com';
     this.anthropicVersion = config.anthropicVersion || '2023-06-01';
+    if (config.models?.length) this.models = config.models;
   }
 
   protected async fetch<T>(url: string, options: RequestInit & { json?: any } = {}): Promise<T> {

package/src/providers/ollama.ts

@@ -3,7 +3,7 @@
  * 支持本地 Ollama 模型
  */
 
-import { Logger } from '@zhin.js/core';
+import { Logger } from '@zhin.js/logger';
 import { BaseProvider } from './base.js';
 import type {
   ProviderConfig,
@@ -19,6 +19,8 @@ const logger = new Logger(null, 'Ollama');
 export interface OllamaConfig extends ProviderConfig {
   host?: string;
   models?: string[];
+  /** Ollama 上下文窗口大小（token 数），默认 32768。影响多轮对话和技能指令的保持能力 */
+  num_ctx?: number;
 }
 
 /**
@@ -81,13 +83,17 @@ function toOllamaTools(tools?: ToolDefinition[]): any[] | undefined {
 export class OllamaProvider extends BaseProvider {
   name = 'ollama';
   models: string[];
+  contextWindow: number;
+  capabilities = { vision: true, streaming: true, toolCalling: true, thinking: true };
 
   private host: string;
+  private numCtx: number;
 
   constructor(config: OllamaConfig = {}) {
     super(config);
     this.host = config.host || config.baseUrl || 'http://localhost:11434';
-    // 使用配置中的模型列表，如果没有则使用默认列表
+    this.numCtx = config.contextWindow ?? config.num_ctx ?? 32768;
+    this.contextWindow = this.numCtx;
     this.models = config.models?.length ? config.models : [
       'llama3.3',
       'llama3.2',
@@ -113,9 +119,16 @@ export class OllamaProvider extends BaseProvider {
       model: request.model,
       messages,
       stream: false,
-      options: {},
+      options: {
+        num_ctx: this.numCtx,
+      },
     };
 
+    // think 参数：控制 qwen3 等模型的思考模式
+    if (request.think !== undefined) {
+      ollamaRequest.think = request.think;
+    }
+
     if (request.temperature !== undefined) {
       ollamaRequest.options.temperature = request.temperature;
     }
@@ -181,9 +194,15 @@ export class OllamaProvider extends BaseProvider {
       model: request.model,
       messages,
       stream: true,
-      options: {},
+      options: {
+        num_ctx: this.numCtx,
+      },
     };
 
+    if (request.think !== undefined) {
+      ollamaRequest.think = request.think;
+    }
+
     if (request.temperature !== undefined) {
       ollamaRequest.options.temperature = request.temperature;
     }

package/src/providers/openai.ts

@@ -28,13 +28,17 @@ export class OpenAIProvider extends BaseProvider {
     'o1-preview',
     'o3-mini',
   ];
+  contextWindow: number;
+  capabilities = { vision: true, streaming: true, toolCalling: true, thinking: false };
 
   private baseUrl: string;
 
   constructor(config: OpenAIConfig = {}) {
     super(config);
+    this.contextWindow = config.contextWindow ?? 128000;
     this.baseUrl = config.baseUrl || 'https://api.openai.com/v1';
-    
+    if (config.models?.length) this.models = config.models;
+
     if (config.organization) {
       this.config.headers = {
         ...this.config.headers,
@@ -111,6 +115,7 @@ export class DeepSeekProvider extends OpenAIProvider {
       ...config,
       baseUrl: config.baseUrl || 'https://api.deepseek.com/v1',
     });
+    if (config.models?.length) this.models = config.models;
   }
 
   async listModels(): Promise<string[]> {
@@ -134,6 +139,7 @@ export class MoonshotProvider extends OpenAIProvider {
       ...config,
       baseUrl: config.baseUrl || 'https://api.moonshot.cn/v1',
     });
+    if (config.models?.length) this.models = config.models;
   }
 
   async listModels(): Promise<string[]> {
@@ -159,6 +165,7 @@ export class ZhipuProvider extends OpenAIProvider {
       ...config,
       baseUrl: config.baseUrl || 'https://open.bigmodel.cn/api/paas/v4',
     });
+    if (config.models?.length) this.models = config.models;
   }
 
   async listModels(): Promise<string[]> {

package/src/rate-limiter.ts

@@ -0,0 +1,129 @@
+/**
+ * RateLimiter — AI 请求频率限制
+ *
+ * 防止单用户过度消耗 GPU/API 资源。
+ *
+ * 策略:
+ *   1. 滑动窗口速率限制 — 每分钟 N 次请求
+ *   2. 优雅降级 — 超限时返回友好提示而非静默丢弃
+ */
+
+import { Logger } from '@zhin.js/logger';
+
+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();
+  }
+}

package/src/session.ts

@@ -9,7 +9,7 @@
  * - 更长的上下文记忆能力
  */
 
-import { Logger } from '@zhin.js/core';
+import { Logger } from '@zhin.js/logger';
 import type { ChatMessage, SessionConfig, Session } from './types.js';
 
 const logger = new Logger(null, 'AI-Session');
@@ -105,8 +105,8 @@ export class MemorySessionManager implements ISessionManager {
   private trimMessages(session: Session): void {
     const maxHistory = session.config.maxHistory ?? this.config.maxHistory;
     if (session.messages.length > maxHistory) {
-      const systemMessages = session.messages.filter(m => m.role === 'system');
-      const otherMessages = session.messages.filter(m => m.role !== 'system');
+      const systemMessages = session.messages.filter((m: ChatMessage) => m.role === 'system');
+      const otherMessages = session.messages.filter((m: ChatMessage) => m.role !== 'system');
       const keepCount = maxHistory - systemMessages.length;
       session.messages = [...systemMessages, ...otherMessages.slice(-keepCount)];
     }
@@ -118,7 +118,7 @@ export class MemorySessionManager implements ISessionManager {
 
   setSystemPrompt(sessionId: string, prompt: string): void {
     const session = this.get(sessionId);
-    session.messages = session.messages.filter(m => m.role !== 'system');
+    session.messages = session.messages.filter((m: ChatMessage) => m.role !== 'system');
     session.messages.unshift({ role: 'system', content: prompt });
     session.updatedAt = Date.now();
   }
@@ -130,7 +130,7 @@ export class MemorySessionManager implements ISessionManager {
   reset(sessionId: string): void {
     const session = this.sessions.get(sessionId);
     if (session) {
-      const systemMessages = session.messages.filter(m => m.role === 'system');
+      const systemMessages = session.messages.filter((m: ChatMessage) => m.role === 'system');
       session.messages = systemMessages;
       session.updatedAt = Date.now();
     }
@@ -192,14 +192,17 @@ export class DatabaseSessionManager implements ISessionManager {
   private saveTimer?: ReturnType<typeof setTimeout>;
   private model: any; // 数据库模型
 
+  /** 内存缓存上限，超出时淘汰最久未访问的条目 */
+  private static readonly MAX_CACHE_SIZE = 2000;
+
   constructor(
     model: any,
     config: { maxHistory?: number; expireMs?: number } = {}
   ) {
     this.model = model;
     this.config = {
-      maxHistory: config.maxHistory ?? 200, // 数据库支持更长的历史
-      expireMs: config.expireMs ?? 7 * 24 * 60 * 60 * 1000, // 7 天过期
+      maxHistory: config.maxHistory ?? 200,
+      expireMs: config.expireMs ?? 7 * 24 * 60 * 60 * 1000,
     };
 
     // 定期清理过期会话（每小时）
@@ -211,13 +214,20 @@ export class DatabaseSessionManager implements ISessionManager {
    */
   private async loadSession(sessionId: string): Promise<Session | null> {
     try {
-      const records = await this.model.select({ session_id: sessionId });
+      const records = await this.model.select().where({ session_id: sessionId });
       if (records && records.length > 0) {
         const record = records[0] as SessionRecord;
+        // SQLite 中 json 类型存储为 TEXT，读回时需要解析
+        const messages = typeof record.messages === 'string'
+          ? JSON.parse(record.messages)
+          : (record.messages || []);
+        const config = typeof record.config === 'string'
+          ? JSON.parse(record.config)
+          : (record.config || { provider: 'openai' });
         return {
           id: record.session_id,
-          config: record.config || { provider: 'openai' },
-          messages: record.messages || [],
+          config: Array.isArray(config) ? { provider: 'openai' } : config,
+          messages: Array.isArray(messages) ? messages : [],
           createdAt: record.created_at,
           updatedAt: record.updated_at,
         };
@@ -250,7 +260,7 @@ export class DatabaseSessionManager implements ISessionManager {
 
     for (const session of sessions) {
       try {
-        const existing = await this.model.select({ session_id: session.id });
+        const existing = await this.model.select().where({ session_id: session.id });
         const record: Partial<SessionRecord> = {
           session_id: session.id,
           messages: session.messages,
@@ -259,7 +269,7 @@ export class DatabaseSessionManager implements ISessionManager {
         };
 
         if (existing && existing.length > 0) {
-          await this.model.update(record, { session_id: session.id });
+          await this.model.update(record).where({ session_id: session.id });
         } else {
           record.created_at = session.createdAt;
           await this.model.create(record);
@@ -279,7 +289,6 @@ export class DatabaseSessionManager implements ISessionManager {
       session = await this.loadSession(sessionId) ?? undefined;
 
       if (!session) {
-        // 创建新会话
         session = {
           id: sessionId,
           config: config || { provider: 'openai' },
@@ -289,6 +298,11 @@ export class DatabaseSessionManager implements ISessionManager {
         };
       }
 
+      this.evictCacheIfNeeded();
+      this.cache.set(sessionId, session);
+    } else {
+      // LRU: 重新插入以更新 Map 的迭代顺序
+      this.cache.delete(sessionId);
       this.cache.set(sessionId, session);
     }
 
@@ -298,13 +312,28 @@ export class DatabaseSessionManager implements ISessionManager {
     return session;
   }
 
+  /**
+   * 当缓存超过上限时，淘汰最久未访问的条目。
+   * 利用 Map 的插入顺序（最旧的在前）。
+   */
+  private evictCacheIfNeeded(): void {
+    while (this.cache.size >= DatabaseSessionManager.MAX_CACHE_SIZE) {
+      const oldest = this.cache.keys().next().value;
+      if (oldest !== undefined) {
+        this.cache.delete(oldest);
+      } else {
+        break;
+      }
+    }
+  }
+
   async has(sessionId: string): Promise<boolean> {
     if (this.cache.has(sessionId)) {
       return true;
     }
 
     try {
-      const records = await this.model.select({ session_id: sessionId });
+      const records = await this.model.select().where({ session_id: sessionId });
       return records && records.length > 0;
     } catch {
       return false;
@@ -322,8 +351,8 @@ export class DatabaseSessionManager implements ISessionManager {
   private trimMessages(session: Session): void {
     const maxHistory = session.config.maxHistory ?? this.config.maxHistory;
     if (session.messages.length > maxHistory) {
-      const systemMessages = session.messages.filter(m => m.role === 'system');
-      const otherMessages = session.messages.filter(m => m.role !== 'system');
+      const systemMessages = session.messages.filter((m: ChatMessage) => m.role === 'system');
+      const otherMessages = session.messages.filter((m: ChatMessage) => m.role !== 'system');
       const keepCount = maxHistory - systemMessages.length;
       session.messages = [...systemMessages, ...otherMessages.slice(-keepCount)];
     }
@@ -336,7 +365,7 @@ export class DatabaseSessionManager implements ISessionManager {
 
   async setSystemPrompt(sessionId: string, prompt: string): Promise<void> {
     const session = await this.get(sessionId);
-    session.messages = session.messages.filter(m => m.role !== 'system');
+    session.messages = session.messages.filter((m: ChatMessage) => m.role !== 'system');
     session.messages.unshift({ role: 'system', content: prompt });
     session.updatedAt = Date.now();
     this.schedulesSave(session);
@@ -357,7 +386,7 @@ export class DatabaseSessionManager implements ISessionManager {
 
   async reset(sessionId: string): Promise<void> {
     const session = await this.get(sessionId);
-    const systemMessages = session.messages.filter(m => m.role === 'system');
+    const systemMessages = session.messages.filter((m: ChatMessage) => m.role === 'system');
     session.messages = systemMessages;
     session.updatedAt = Date.now();
     this.schedulesSave(session);