my-ai-chat-framework 1.0.2

package/src/core/Messages.js

@@ -0,0 +1,562 @@
+import MessageFormatter from './MessageFormatter.js';
+
+class Messages {
+  constructor() {
+    this.messages = [];
+    this.systemPrompts = [];
+    this.tools = [];
+    this.metadata = {
+      name: '',
+      createdAt: new Date(),
+      lastModified: new Date()
+    };
+  }
+
+  // ========== 核心消息方法 ==========
+
+  /**
+   * 添加消息（通用方法）
+   * @param {string} role - 角色：user/assistant/system/tool
+   * @param {string} content - 消息内容
+   * @param {object} metadata - 额外元数据
+   * @returns {number} 新消息的索引
+   */
+  addMessage(role, content, metadata = {}) {
+    // 验证角色
+    const validRoles = ['user', 'assistant', 'system', 'tool'];
+    if (!validRoles.includes(role)) {
+      throw new Error(`无效的角色: ${role}，有效值: ${validRoles.join(', ')}`);
+    }
+    
+    // 特殊处理：tool消息可以没有content（但必须有tool_call_id）
+    if (role === 'tool') {
+      // tool消息必须有tool_call_id
+      if (!metadata.tool_call_id) {
+        throw new Error('tool消息必须包含tool_call_id');
+      }
+      // tool消息的content可以为空
+    } else {
+      // 其他消息必须有内容
+      if (typeof content !== 'string') {
+        throw new Error('消息内容必须是字符串');
+      }
+      // 允许空字符串，但不允许undefined/null
+      if (content === undefined || content === null) {
+        throw new Error('消息内容不能为undefined或null');
+      }
+    }
+    
+    // 创建消息对象
+    const message = {
+      role,
+      content: content ? content.trim() : '', // 允许空字符串
+      timestamp: new Date()
+    };
+    
+    // 合并元数据（过滤掉undefined和null）
+    if (metadata && typeof metadata === 'object') {
+      for (const [key, value] of Object.entries(metadata)) {
+        if (value !== undefined && value !== null) {
+          message[key] = value;
+        }
+      }
+    }
+    
+    return this.messages.push(message) - 1;
+  }
+
+  /**
+   * 添加用户消息（便捷方法）
+   */
+  addUserMessage(content, metadata = {}) {
+    return this.addMessage('user', content, metadata);
+  }
+
+  /**
+   * 添加助手消息（便捷方法）
+   */
+  addAssistantMessage(content, metadata = {}) {
+    return this.addMessage('assistant', content, metadata);
+  }
+
+  /**
+   * 添加系统消息（便捷方法）
+   */
+  addSystemMessage(content, metadata = {}) {
+    return this.addMessage('system', content, metadata);
+  }
+
+  /**
+   * 添加工具消息（便捷方法）
+   */
+  addToolMessage(content, toolCallId, metadata = {}) {
+    if (!toolCallId || typeof toolCallId !== 'string') {
+      throw new Error('tool_call_id不能为空');
+    }
+    
+    return this.addMessage('tool', content, {
+      tool_call_id: toolCallId,
+      ...metadata
+    });
+  }
+
+  /**
+   * 批量添加消息
+   */
+  addMessages(messagesArray) {
+    if (!Array.isArray(messagesArray)) {
+      throw new Error('参数必须是数组');
+    }
+    
+    const indices = [];
+    for (const msg of messagesArray) {
+      if (!msg.role || !msg.content) {
+        throw new Error('消息必须包含role和content属性');
+      }
+      
+      const { role, content, ...metadata } = msg;
+      const index = this.addMessage(role, content, metadata);
+      indices.push(index);
+    }
+    
+    return indices;
+  }
+
+  // ========== 消息查询 ==========
+
+  /**
+   * 获取所有消息
+   */
+  getMessages() {
+    return [...this.messages];
+  }
+
+  /**
+   * 按角色获取消息
+   */
+  getMessagesByRole(role) {
+    const validRoles = ['user', 'assistant', 'system', 'tool'];
+    if (!validRoles.includes(role)) {
+      throw new Error(`无效的角色: ${role}`);
+    }
+    return this.messages.filter(msg => msg.role === role);
+  }
+
+  /**
+   * 获取最后一条消息
+   */
+  getLastMessage() {
+    return this.messages.length > 0 
+      ? { ...this.messages[this.messages.length - 1] }
+      : null;
+  }
+
+  /**
+   * 获取最后一条用户消息
+   */
+  getLastUserMessage() {
+    for (let i = this.messages.length - 1; i >= 0; i--) {
+      if (this.messages[i].role === 'user') {
+        return { ...this.messages[i] };
+      }
+    }
+    return null;
+  }
+
+  /**
+   * 获取最后一条助手消息
+   */
+  getLastAssistantMessage() {
+    for (let i = this.messages.length - 1; i >= 0; i--) {
+      if (this.messages[i].role === 'assistant') {
+        return { ...this.messages[i] };
+      }
+    }
+    return null;
+  }
+
+  /**
+   * 获取消息数量
+   */
+  getMessageCount() {
+    return this.messages.length;
+  }
+
+  // ========== 消息修改 ==========
+
+  /**
+   * 更新消息内容
+   */
+  updateMessage(index, content) {
+    if (!Number.isInteger(index) || index < 0 || index >= this.messages.length) {
+      throw new Error(`无效的消息索引: ${index}`);
+    }
+    
+    if (typeof content !== 'string' || !content.trim()) {
+      throw new Error('消息内容不能为空');
+    }
+    
+    this.messages[index].content = content.trim();
+    this.messages[index].lastModified = new Date();
+    return this;
+  }
+
+  /**
+   * 更新最后一条消息
+   */
+  updateLastMessage(content) {
+    if (this.messages.length === 0) {
+      throw new Error('没有消息可以修改');
+    }
+    return this.updateMessage(this.messages.length - 1, content);
+  }
+
+  /**
+   * 设置消息字段
+   */
+  setMessageField(index, field, value) {
+    if (!Number.isInteger(index) || index < 0 || index >= this.messages.length) {
+      throw new Error(`无效的消息索引: ${index}`);
+    }
+    
+    this.messages[index][field] = value;
+    this.messages[index].lastModified = new Date();
+    return this;
+  }
+
+  // ========== 消息删除 ==========
+
+  /**
+   * 删除消息
+   */
+  removeMessage(index) {
+    if (!Number.isInteger(index) || index < 0 || index >= this.messages.length) {
+      throw new Error(`无效的消息索引: ${index}`);
+    }
+    
+    return this.messages.splice(index, 1)[0];
+  }
+
+  /**
+   * 删除最后一条消息
+   */
+  removeLastMessage() {
+    if (this.messages.length === 0) {
+      throw new Error('没有消息可以删除');
+    }
+    return this.removeMessage(this.messages.length - 1);
+  }
+
+  /**
+   * 撤回到上一条助手消息
+   */
+  undoToAssistant() {
+    if (this.messages.length === 0) {
+      return [];
+    }
+    
+    // 找到最后一条助手消息的位置
+    let lastAssistantIndex = -1;
+    for (let i = this.messages.length - 1; i >= 0; i--) {
+      if (this.messages[i].role === 'assistant') {
+        lastAssistantIndex = i;
+        break;
+      }
+    }
+    
+    if (lastAssistantIndex >= 0) {
+      // 删除该助手消息之后的所有消息
+      return this.messages.splice(lastAssistantIndex + 1);
+    } else {
+      // 没有助手消息，清空所有
+      const removed = [...this.messages];
+      this.messages = [];
+      return removed;
+    }
+  }
+
+  /**
+   * 是否可以撤回
+   */
+  canUndo() {
+    return this.messages.some(msg => msg.role === 'assistant');
+  }
+
+  /**
+   * 清空所有消息
+   */
+  clearMessages() {
+    const removed = [...this.messages];
+    this.messages = [];
+    this.metadata.lastModified = new Date();
+    return removed;
+  }
+
+  // ========== 系统提示管理 ==========
+
+  /**
+   * 添加系统提示
+   */
+  addSystemPrompt(content, enabled = true) {
+    if (typeof content !== 'string' || !content.trim()) {
+      throw new Error('系统提示内容不能为空');
+    }
+    
+    const prompt = {
+      role: 'system',
+      content: content.trim(),
+      enabled: Boolean(enabled),
+      createdAt: new Date()
+    };
+    
+    return this.systemPrompts.push(prompt) - 1;
+  }
+
+  /**
+   * 获取启用的系统提示
+   */
+  getEnabledSystemPrompts() {
+    return this.systemPrompts
+      .filter(prompt => prompt.enabled)
+      .map(prompt => ({ role: prompt.role, content: prompt.content }));
+  }
+
+  /**
+   * 切换系统提示状态
+   */
+  toggleSystemPrompt(index) {
+    if (!Number.isInteger(index) || index < 0 || index >= this.systemPrompts.length) {
+      throw new Error(`无效的系统提示索引: ${index}`);
+    }
+    
+    this.systemPrompts[index].enabled = !this.systemPrompts[index].enabled;
+    return this;
+  }
+
+  /**
+   * 更新系统提示内容
+   */
+  updateSystemPrompt(index, content) {
+    if (!Number.isInteger(index) || index < 0 || index >= this.systemPrompts.length) {
+      throw new Error(`无效的系统提示索引: ${index}`);
+    }
+    
+    if (typeof content !== 'string' || !content.trim()) {
+      throw new Error('系统提示内容不能为空');
+    }
+    
+    this.systemPrompts[index].content = content.trim();
+    this.systemPrompts[index].lastModified = new Date();
+    return this;
+  }
+
+  /**
+   * 删除系统提示
+   */
+  removeSystemPrompt(index) {
+    if (!Number.isInteger(index) || index < 0 || index >= this.systemPrompts.length) {
+      throw new Error(`无效的系统提示索引: ${index}`);
+    }
+    
+    return this.systemPrompts.splice(index, 1)[0];
+  }
+
+  // ========== 工具管理 ==========
+
+  /**
+   * 添加工具
+   */
+  addTool(toolDefinition) {
+    if (typeof toolDefinition === 'object' && toolDefinition !== null) {
+      // 如果已经是格式化好的工具对象
+      if (!toolDefinition.type || toolDefinition.type !== 'function') {
+        throw new Error(`无效的工具类型，期望: 'function'，实际: '${toolDefinition.type || 'undefined'}'`);
+      }
+      if (!toolDefinition.function?.name?.trim()) {
+        throw new Error('工具名称不能为空');
+      }
+      return this.tools.push(toolDefinition) - 1;
+    }
+
+    // 如果是工具定义对象，使用MessageFormatter构造
+    if (toolDefinition.name && toolDefinition.description && toolDefinition.parameters) {
+      const constructedTool = MessageFormatter.constructTool(toolDefinition);
+      return this.tools.push(constructedTool) - 1;
+    }
+
+    throw new Error('无效的工具格式');
+  }
+
+  /**
+   * 从定义添加工具
+   */
+  addToolFromDefinition(name, description, parameters) {
+    const tool = MessageFormatter.constructTool({ name, description, parameters });
+    return this.tools.push(tool) - 1;
+  }
+
+  /**
+   * 获取所有工具
+   */
+  getTools() {
+    return [...this.tools];
+  }
+
+  /**
+   * 删除工具
+   */
+  removeTool(index) {
+    if (!Number.isInteger(index) || index < 0 || index >= this.tools.length) {
+      throw new Error(`无效的工具索引: ${index}`);
+    }
+    
+    return this.tools.splice(index, 1)[0];
+  }
+
+  /**
+   * 清空所有工具
+   */
+  clearTools() {
+    const removed = [...this.tools];
+    this.tools = [];
+    return removed;
+  }
+
+  // ========== 元数据管理 ==========
+
+  setName(name) {
+    if (typeof name !== 'string' || !name.trim()) {
+      throw new Error('名称不能为空');
+    }
+    this.metadata.name = name.trim();
+    this.metadata.lastModified = new Date();
+    return this;
+  }
+
+  getName() {
+    return this.metadata.name;
+  }
+
+  // ========== 兼容性方法 ==========
+
+  /**
+   * 获取最近的消息（兼容旧方法）
+   */
+  getRecentMessages(baseRounds, cycleRounds) {
+    const messages = this.messages.map(item => {
+      // 使用MessageFormatter.toStandardFormat来确保一致性
+      return MessageFormatter.toStandardFormat(item);
+    });
+    
+    const msgTotal = messages.length;
+    const totalRounds = msgTotal / 2;
+
+    if (!baseRounds || isNaN(baseRounds) || baseRounds <= 0) {
+      return messages;
+    }
+
+    const maxCycleRoundMultiple = cycleRounds > 0
+      ? Math.floor(totalRounds / cycleRounds) * cycleRounds
+      : 0;
+    const actualRounds = baseRounds + (totalRounds - maxCycleRoundMultiple);
+    const actualMsgCount = actualRounds * 2;
+    const n = Math.max(msgTotal - actualMsgCount, 0);
+
+    return messages.slice(n);
+  }
+
+  /**
+   * 获取所有格式化消息（兼容旧方法）
+   */
+  getMessagesFormatted(baseRounds, cycleRounds) {
+    const systems = this.getEnabledSystemPrompts();
+    const hintSystem = {
+      role: 'system',
+      content: `你的名字||标题是：${this.metadata.name || '未知'}`
+    };
+    systems.unshift(hintSystem);
+    
+    const recentMessages = this.getRecentMessages(baseRounds, cycleRounds);
+    return [...systems, ...recentMessages];
+  }
+
+  // ========== 导入导出 ==========
+
+  export() {
+    return {
+      __format_version: 2,
+      exportedAt: new Date().toISOString(),
+      metadata: { ...this.metadata },
+      payload: {
+        messages: JSON.parse(JSON.stringify(this.messages)),
+        systemPrompts: JSON.parse(JSON.stringify(this.systemPrompts)),
+        tools: JSON.parse(JSON.stringify(this.tools))
+      }
+    };
+  }
+
+  import(data) {
+    const obj = typeof data === 'string' ? JSON.parse(data) : data;
+    const parsed = Messages.parseCompatible(obj);
+    
+    this.messages = parsed.messages || [];
+    this.systemPrompts = parsed.systemPrompts || [];
+    this.tools = parsed.tools || [];
+    this.metadata = parsed.metadata || { 
+      name: '', 
+      createdAt: new Date(), 
+      lastModified: new Date() 
+    };
+    
+    this.metadata.lastModified = new Date();
+    return this;
+  }
+
+  static parseCompatible(obj) {
+    // 处理新格式（v2）
+    if (obj?.__format_version === 2 && obj.payload) {
+      return {
+        messages: obj.payload.messages || [],
+        systemPrompts: obj.payload.systemPrompts || [],
+        tools: obj.payload.tools || [],
+        metadata: obj.metadata || { 
+          name: '', 
+          createdAt: new Date(), 
+          lastModified: new Date() 
+        }
+      };
+    }
+    
+    // 处理旧格式（v1）
+    if (obj?.__format_version === 1 && obj.payload) {
+      return {
+        messages: obj.payload.messages || [],
+        systemPrompts: obj.payload.systems || [],
+        tools: obj.payload.tools || [],
+        metadata: { 
+          name: obj.payload.hintData?.name || '',
+          createdAt: new Date(),
+          lastModified: new Date()
+        }
+      };
+    }
+
+    // 处理更旧的格式
+    if (obj?.messages && Array.isArray(obj.messages)) {
+      return {
+        messages: obj.messages || [],
+        systemPrompts: obj.systems || [],
+        tools: obj.tools || [],
+        metadata: { 
+          name: obj.hintData?.name || '',
+          createdAt: new Date(),
+          lastModified: new Date()
+        }
+      };
+    }
+
+    throw new Error('无法识别的聊天数据格式');
+  }
+}
+
+export default Messages;

package/src/core/RequestBuilder.js

@@ -0,0 +1,96 @@
+import MessageFormatter from './MessageFormatter.js';
+import ToolManager from './ToolManager.js';
+import EventEmitter from './EventEmitter.js';
+
+/**
+ * 请求构建器
+ * 负责构建符合不同API规范的请求体
+ */
+class RequestBuilder {
+  constructor(model, messagesInstance, config = {}) {
+    this.model = model;
+    this.messages = messagesInstance;
+    this.config = {
+      // temperature: 0.7,
+      // max_tokens: 2000,
+      // stream: false, // 默认关闭流式传输
+      ...config
+    };
+    this.toolManager = null; // 可选的ToolManager引用
+  }
+
+  /**
+   * 设置ToolManager引用
+   */
+  setToolManager(toolManager) {
+    this.toolManager = toolManager;
+    return this;
+  }
+
+  /**
+   * 构建OpenAI兼容格式的请求体
+   */
+  buildOpenAIFormat(baseRounds, cycleRounds) {
+    // 使用新的MessageFormatter方法
+    const formattedMessages = MessageFormatter.formatMessages(
+      this.messages, 
+      baseRounds, 
+      cycleRounds
+    );
+    
+    const request = {
+      model: this.model,
+      messages: formattedMessages,
+      ...this.config
+    };
+
+    // 优先从ToolManager获取工具定义，其次从Messages获取
+    let tools = [];
+    if (this.toolManager && this.toolManager.hasTools()) {
+      tools = this.toolManager.getToolDefinitions();
+    } else {
+      tools = this.messages.getTools();
+    }
+    
+    if (tools && tools.length > 0) {
+      request.tools = tools;
+      request.tool_choice = 'auto';
+    }
+
+    return request;
+  }
+
+  /**
+   * 构建请求体并返回JSON字符串
+   */
+  toJSON(baseRounds, cycleRounds) {
+    const request = this.buildOpenAIFormat(baseRounds, cycleRounds);
+    return JSON.stringify(request);
+  }
+
+  /**
+   * 构建请求体并返回对象
+   */
+  toObject(baseRounds, cycleRounds) {
+    return this.buildOpenAIFormat(baseRounds, cycleRounds);
+  }
+
+  /**
+   * 设置配置项
+   */
+  setConfig(key, value) {
+    this.config[key] = value;
+    return this; // 支持链式调用
+  }
+
+  /**
+   * 批量设置配置项
+   */
+  setConfigs(configs) {
+    Object.assign(this.config, configs);
+    return this;
+  }
+}
+
+export default RequestBuilder;
+