foliko 1.0.86 → 1.1.0

package/src/core/agent-context.js

@@ -0,0 +1,188 @@
+/**
+ * AgentContext - Agent 级别的上下文封装
+ *
+ * 职责：
+ * 1. 隔离不同 Agent 实例的状态
+ * 2. 管理 Agent 级工具注册
+ * 3. 管理 Agent 级技能
+ * 4. 管理 Agent 级系统提示词
+ */
+
+class AgentContext {
+  /**
+   * @param {string} agentId - Agent ID
+   * @param {Agent} agent - Agent 实例引用
+   * @param {Object} options - 配置选项
+   */
+  constructor(agentId, agent, options = {}) {
+    this.agentId = agentId;
+    this.agent = agent; // Agent 实例引用
+
+    this.name = options.name || agent?.name || agentId;
+
+    // Agent 级工具注册（覆盖全局）
+    this.tools = new Map();
+
+    // Agent 级技能
+    this.skills = [];
+
+    // Agent 级系统提示词
+    this.systemPrompt = null;
+
+    // 自定义数据
+    this.customData = new Map();
+
+    // 元数据
+    this.metadata = {
+      createdAt: Date.now(),
+      lastActive: Date.now(),
+    };
+  }
+
+  // ==================== 工具管理 ====================
+
+  /**
+   * 注册工具
+   * @param {Object} toolDef - 工具定义
+   */
+  registerTool(toolDef) {
+    this.tools.set(toolDef.name, toolDef);
+  }
+
+  /**
+   * 批量注册工具
+   * @param {Array<Object>} toolDefs - 工具定义数组
+   */
+  registerTools(toolDefs) {
+    for (const toolDef of toolDefs) {
+      this.registerTool(toolDef);
+    }
+  }
+
+  /**
+   * 注销工具
+   * @param {string} name - 工具名
+   */
+  unregisterTool(name) {
+    this.tools.delete(name);
+  }
+
+  /**
+   * 获取工具
+   * @param {string} name - 工具名
+   * @returns {Object|null}
+   */
+  getTool(name) {
+    return this.tools.get(name) || null;
+  }
+
+  /**
+   * 获取所有工具
+   * @returns {Array}
+   */
+  getAllTools() {
+    return Array.from(this.tools.values());
+  }
+
+  /**
+   * 检查是否有工具
+   * @param {string} name - 工具名
+   * @returns {boolean}
+   */
+  hasTool(name) {
+    return this.tools.has(name);
+  }
+
+  // ==================== 技能管理 ====================
+
+  /**
+   * 添加技能
+   * @param {Object} skill - 技能对象
+   */
+  addSkill(skill) {
+    this.skills.push(skill);
+  }
+
+  /**
+   * 移除技能
+   * @param {string} skillName - 技能名
+   */
+  removeSkill(skillName) {
+    this.skills = this.skills.filter((s) => s.name !== skillName);
+  }
+
+  /**
+   * 获取所有技能
+   * @returns {Array}
+   */
+  getAllSkills() {
+    return [...this.skills];
+  }
+
+  // ==================== 系统提示词 ====================
+
+  /**
+   * 设置系统提示词
+   * @param {string} prompt - 提示词
+   */
+  setSystemPrompt(prompt) {
+    this.systemPrompt = prompt;
+  }
+
+  /**
+   * 获取系统提示词
+   * @returns {string|null}
+   */
+  getSystemPrompt() {
+    return this.systemPrompt;
+  }
+
+  // ==================== 自定义数据 ====================
+
+  /**
+   * 设置自定义数据
+   * @param {string} key - 键
+   * @param {*} value - 值
+   */
+  setCustomData(key, value) {
+    this.customData.set(key, value);
+  }
+
+  /**
+   * 获取自定义数据
+   * @param {string} key - 键
+   * @param {*} defaultValue - 默认值
+   * @returns {*}
+   */
+  getCustomData(key, defaultValue = undefined) {
+    return this.customData.get(key) ?? defaultValue;
+  }
+
+  /**
+   * 删除自定义数据
+   * @param {string} key - 键
+   */
+  deleteCustomData(key) {
+    this.customData.delete(key);
+  }
+
+  // ==================== 生命周期 ====================
+
+  /**
+   * 更新最后活跃时间
+   */
+  touch() {
+    this.metadata.lastActive = Date.now();
+  }
+
+  /**
+   * 销毁
+   */
+  destroy() {
+    this.tools.clear();
+    this.skills = [];
+    this.customData.clear();
+  }
+}
+
+module.exports = { AgentContext };

package/src/core/agent.js

@@ -53,10 +53,6 @@ class Agent extends EventEmitter {
     // 子Agent管理
     this._subAgents = new Map();
 
-    // 消息队列（用于 pushMessage）
-    this._messageQueue = [];
-    this._isProcessingQueue = false;
-
     // System prompt 构建器
     this._systemPromptBuilder = new SystemPromptBuilder();
     this._registerDefaultPromptParts();
@@ -101,7 +97,7 @@ class Agent extends EventEmitter {
     });
 
     // 4. 工具列表 (优先级 400)
-    this._systemPromptBuilder.register('tools', 400, () => this._buildToolsDescription());
+    //this._systemPromptBuilder.register('tools', 400, () => this._buildToolsDescription());
 
     // 5. 技能列表 (优先级 500)
     this._systemPromptBuilder.register('skills', 500, () => this._buildSkillsDescription());
@@ -526,17 +522,21 @@ class Agent extends EventEmitter {
 4. **工具优先**：可用工具列表会提供，格式为 toolName(toolArgs)。不确定用哪个工具时，优先调用可能相关的工具。
 5. **结果导向**：调用工具后，基于返回结果回答，不要重复工具的内部实现细节。
 6. **多步骤任务**：复杂任务拆解为多个工具调用，逐步完成，每步基于结果决定下一步。
+7. **步骤未完成必须提示继续**：如果任务需要多步骤操作，在返回前必须明确告知用户当前进度和下一步操作。如果回复内容为空或不完整，必须说明"步骤进行中，请继续"。
 
 
 【响应规范】
 - 直接给出结论或结果，不说"我需要..."、"我建议..."等铺垫话术
 - 如果工具返回错误，说明错误原因并给出解决建议
 - 如果信息不足，先调用工具获取必要信息，不要假设或猜测
+- **重要：如果任务需要多步骤操作，在回复末尾必须提示用户"请继续"或"是否继续下一步"**
+
 
 【禁止事项】
 - 不调用工具就直接回答
 - 编造不存在的数据、文件、订单、用户等信息
-- 回复含糊不清，让用户无法确定答案是否正确`;
+- 回复含糊不清，让用户无法确定答案是否正确
+- **多步骤任务未完成就结束，不提示用户继续**`;
   }
 
   /**
@@ -734,10 +734,6 @@ class Agent extends EventEmitter {
    * 发送消息
    */
   async chat(message, options = {}) {
-    if (this._status === 'busy') {
-      throw new Error('Agent is busy');
-    }
-
     if (this._status === 'error') {
       this._status = 'idle';
     }
@@ -755,21 +751,16 @@ class Agent extends EventEmitter {
 
       this._syncTools();
 
+      // 通过 handler 的 enqueue 处理（自动排队）
       const result = await this._chatHandler.chat(enhancedMessage, options);
+
       this._status = 'idle';
       this.emit('status', { status: 'idle' });
 
-      // 处理队列中的下一条消息
-      setImmediate(() => this._processQueue());
-
       return result;
     } catch (err) {
       this._status = 'error';
       this.emit('status', { status: 'error', error: err.message });
-
-      // 发生错误时也要处理队列
-      setImmediate(() => this._processQueue());
-
       throw err;
     }
   }
@@ -789,49 +780,10 @@ class Agent extends EventEmitter {
       options.sessionId = ctx.sessionId;
     }
 
-    // 如果忙碌，进入队列等待
-    if (this._status === 'busy') {
-      const queuedItem = { message, options, resolve: null, reject: null };
-      const promise = new Promise((resolve, reject) => {
-        queuedItem.resolve = resolve;
-        queuedItem.reject = reject;
-      });
-      this._messageQueue.push(queuedItem);
-      // Agent busy, message queued
-      return promise;
-    }
-
-    // 空闲则直接处理
+    // 直接调用 chat，由 handler 的 enqueue 处理排队
     return this.chat(message, options);
   }
 
-  /**
-   * 处理消息队列
-   * @private
-   */
-  async _processQueue() {
-    if (this._isProcessingQueue || this._messageQueue.length === 0) {
-      return;
-    }
-
-    this._isProcessingQueue = true;
-
-    // 强制设置为 idle，确保 this.chat() 能执行
-    this._status = 'idle';
-
-    while (this._messageQueue.length > 0) {
-      const item = this._messageQueue.shift();
-      try {
-        const result = await this.chat(item.message, item.options);
-        item.resolve(result);
-      } catch (err) {
-        item.reject(err);
-      }
-    }
-
-    this._isProcessingQueue = false;
-  }
-
   /**
    * 发送消息（流式）
    */
@@ -861,16 +813,10 @@ class Agent extends EventEmitter {
       yield* this._chatHandler.chatStream(enhancedMessage, options);
       this._status = 'idle';
       this.emit('status', { status: 'idle' });
-
-      // 处理队列中的下一条消息
-      setImmediate(() => this._processQueue());
     } catch (err) {
       this._status = 'error';
       this.emit('status', { status: 'error', error: err.message });
 
-      // 发生错误时也要处理队列
-      setImmediate(() => this._processQueue());
-
       throw err;
     }
   }

package/src/core/context-manager.js

@@ -0,0 +1,283 @@
+/**
+ * ContextManager - 上下文管理器
+ *
+ * 职责：
+ * 1. 统一管理所有 Context 的生命周期
+ * 2. 提供上下文查询接口
+ * 3. 处理上下文清理和资源释放
+ */
+
+const { RequestContext } = require('./request-context');
+const { SessionContext } = require('./session-context');
+const { AgentContext } = require('./agent-context');
+
+class ContextManager {
+  /**
+   * @param {Framework} framework - 框架实例
+   */
+  constructor(framework) {
+    this.framework = framework;
+
+    // Context 存储
+    this._requestContexts = new Map(); // requestId → RequestContext
+    this._sessionContexts = new Map(); // sessionId → SessionContext
+    this._agentContexts = new Map(); // agentId → AgentContext
+
+    // 引用计数（用于自动清理）
+    this._refCounts = new Map();
+
+    // 事件转发
+    this._setupEventForwarding();
+  }
+
+  /**
+   * 设置事件转发
+   * @private
+   */
+  _setupEventForwarding() {
+    if (this.framework) {
+      this.framework.on('session:context-created', (ctx) => {
+        // 可以在这里添加日志、监控等
+      });
+      this.framework.on('session:context-destroyed', (ctx) => {
+        // 可以在这里添加清理逻辑
+      });
+    }
+  }
+
+  // ==================== Request Context ====================
+
+  /**
+   * 创建 Request Context
+   * @param {Object} options - 请求选项
+   * @returns {RequestContext}
+   */
+  createRequestContext(options = {}) {
+    const ctx = new RequestContext(options);
+    this._requestContexts.set(ctx.requestId, ctx);
+    this._refCounts.set(`request:${ctx.requestId}`, 1);
+    return ctx;
+  }
+
+  /**
+   * 获取 Request Context
+   * @param {string} requestId - 请求 ID
+   * @returns {RequestContext|null}
+   */
+  getRequestContext(requestId) {
+    return this._requestContexts.get(requestId) || null;
+  }
+
+  /**
+   * 释放 Request Context
+   * @param {string} requestId - 请求 ID
+   */
+  releaseRequestContext(requestId) {
+    const ctx = this._requestContexts.get(requestId);
+    if (ctx) {
+      ctx.cancel();
+      this._requestContexts.delete(requestId);
+      this._refCounts.delete(`request:${requestId}`);
+    }
+  }
+
+  /**
+   * 增加 Request Context 引用计数
+   * @param {string} requestId - 请求 ID
+   */
+  retainRequestContext(requestId) {
+    const key = `request:${requestId}`;
+    this._refCounts.set(key, (this._refCounts.get(key) || 0) + 1);
+  }
+
+  // ==================== Session Context ====================
+
+  /**
+   * 获取或创建 Session Context
+   * @param {string} sessionId - 会话 ID
+   * @param {Object} options - 配置选项
+   * @returns {SessionContext}
+   */
+  getOrCreateSessionContext(sessionId, options = {}) {
+    let ctx = this._sessionContexts.get(sessionId);
+    if (!ctx) {
+      ctx = new SessionContext(sessionId, this.framework, options);
+      this._sessionContexts.set(sessionId, ctx);
+      this._refCounts.set(`session:${sessionId}`, 1);
+    } else {
+      // 增加引用计数
+      const key = `session:${sessionId}`;
+      this._refCounts.set(key, (this._refCounts.get(key) || 0) + 1);
+    }
+    return ctx;
+  }
+
+  /**
+   * 获取 Session Context
+   * @param {string} sessionId - 会话 ID
+   * @returns {SessionContext|null}
+   */
+  getSessionContext(sessionId) {
+    return this._sessionContexts.get(sessionId) || null;
+  }
+
+  /**
+   * 释放 Session Context（减少引用计数）
+   * @param {string} sessionId - 会话 ID
+   * @param {boolean} [force=false] - 强制销毁
+   */
+  releaseSessionContext(sessionId, force = false) {
+    const key = `session:${sessionId}`;
+    const count = this._refCounts.get(key) || 0;
+
+    if (force || count <= 1) {
+      const ctx = this._sessionContexts.get(sessionId);
+      if (ctx) {
+        ctx.destroy();
+        this.framework.emit('session:context-destroyed', ctx);
+      }
+      this._sessionContexts.delete(sessionId);
+      this._refCounts.delete(key);
+    } else {
+      this._refCounts.set(key, count - 1);
+    }
+  }
+
+  /**
+   * 增加 Session Context 引用计数
+   * @param {string} sessionId - 会话 ID
+   */
+  retainSessionContext(sessionId) {
+    const key = `session:${sessionId}`;
+    this._refCounts.set(key, (this._refCounts.get(key) || 0) + 1);
+  }
+
+  /**
+   * 检查 Session Context 是否存在
+   * @param {string} sessionId - 会话 ID
+   * @returns {boolean}
+   */
+  hasSessionContext(sessionId) {
+    return this._sessionContexts.has(sessionId);
+  }
+
+  // ==================== Agent Context ====================
+
+  /**
+   * 获取或创建 Agent Context
+   * @param {string} agentId - Agent ID
+   * @param {Agent} agent - Agent 实例
+   * @param {Object} options - 配置选项
+   * @returns {AgentContext}
+   */
+  getOrCreateAgentContext(agentId, agent, options = {}) {
+    let ctx = this._agentContexts.get(agentId);
+    if (!ctx) {
+      ctx = new AgentContext(agentId, agent, options);
+      this._agentContexts.set(agentId, ctx);
+    }
+    return ctx;
+  }
+
+  /**
+   * 获取 Agent Context
+   * @param {string} agentId - Agent ID
+   * @returns {AgentContext|null}
+   */
+  getAgentContext(agentId) {
+    return this._agentContexts.get(agentId) || null;
+  }
+
+  /**
+   * 删除 Agent Context
+   * @param {string} agentId - Agent ID
+   */
+  removeAgentContext(agentId) {
+    const ctx = this._agentContexts.get(agentId);
+    if (ctx) {
+      ctx.destroy();
+      this._agentContexts.delete(agentId);
+    }
+  }
+
+  // ==================== 批量操作 ====================
+
+  /**
+   * 销毁所有 Session Context
+   */
+  destroyAllSessionContexts() {
+    for (const [sessionId] of this._sessionContexts) {
+      this.releaseSessionContext(sessionId, true);
+    }
+  }
+
+  /**
+   * 清理过期的 Request Context
+   * @param {number} [maxAge=300000] - 最大存活时间（毫秒），默认 5 分钟
+   */
+  cleanupExpiredRequests(maxAge = 300000) {
+    const now = Date.now();
+    for (const [requestId, ctx] of this._requestContexts) {
+      if (now - ctx.startTime > maxAge) {
+        this.releaseRequestContext(requestId);
+      }
+    }
+  }
+
+  /**
+   * 清理所有 Context
+   */
+  destroyAll() {
+    // 清理 Request Contexts
+    for (const [requestId] of this._requestContexts) {
+      this.releaseRequestContext(requestId);
+    }
+
+    // 清理 Session Contexts
+    this.destroyAllSessionContexts();
+
+    // 清理 Agent Contexts
+    for (const [agentId] of this._agentContexts) {
+      this.removeAgentContext(agentId);
+    }
+  }
+
+  // ==================== 统计 ====================
+
+  /**
+   * 获取统计信息
+   * @returns {Object}
+   */
+  getStats() {
+    return {
+      activeRequests: this._requestContexts.size,
+      activeSessions: this._sessionContexts.size,
+      activeAgents: this._agentContexts.size,
+      refCounts: Object.fromEntries(this._refCounts),
+    };
+  }
+
+  /**
+   * 获取内存使用估计
+   * @returns {Object}
+   */
+  getMemoryUsage() {
+    let sessionMessagesSize = 0;
+    let sessionVariablesSize = 0;
+
+    for (const ctx of this._sessionContexts.values()) {
+      sessionMessagesSize += ctx.messageStore.messages.length;
+      sessionVariablesSize += ctx.variables.size;
+    }
+
+    return {
+      requestContexts: this._requestContexts.size,
+      sessionContexts: this._sessionContexts.size,
+      agentContexts: this._agentContexts.size,
+      totalSessionMessages: sessionMessagesSize,
+      totalSessionVariables: sessionVariablesSize,
+    };
+  }
+}
+
+module.exports = { ContextManager };