mm_ai 1.0.0

package/index.js

@@ -0,0 +1,560 @@
+require('mm_expand');
+const {
+  Log
+} = require('mm_logs');
+const {
+  cacheAdmin
+} = require('mm_cache');
+const {
+  sqlAdmin
+} = require('mm_sql');
+const {
+  Config
+} = require('mm_config');
+const {
+  Manager,
+  Drive
+} = require('mm_machine');
+const { Agent } = require('./core/agent/index.js');
+const { MCP } = require('./core/mcp/index.js');
+const { LLM } = require('./core/llm/index.js');
+const { Tool } = require('./core/tool/index.js');
+const { Skill } = require('./core/skill/index.js');
+const { Knowledge } = require('./core/knowledge/index.js');
+const { Work } = require('./core/work/index.js');
+const { Prompt } = require('./core/prompt/index.js');
+const { Resource } = require('./core/resource/index.js');
+const { Project } = require('./core/project/index.js');
+const { Chat } = require('./chat.js');
+
+/**
+ * MM_ai 类
+ */
+class MM_ai extends Drive {
+  static config = {
+    name: 'os',
+    title: '操作系统',
+    description: '操作系统',
+    diy_sql: true,
+    diy_cache: true,
+    diy_filer: false,
+    sql: {
+      // mysql、sqlite
+      way: 'mysql'
+    },
+    sqlite: {
+      dir: './db/'.fullname(),
+      user: 'root',
+      password: '',
+      database: 'mm',
+      charset: 'utf8mb4',
+      connect_timeout: 5000,
+      connection_limit: 500
+    },
+    mysql: {
+      host: '127.0.0.1',
+      port: 3306,
+      user: 'root',
+      password: 'Asd159357',
+      database: 'mm',
+      charset: 'utf8mb4',
+      timezone: '+08:00',
+      connect_timeout: 5000,
+      connection_limit: 500
+    },
+    cache: {
+      // 缓存方式memory、redis、mongodb
+      way: 'redis'
+    },
+    // 缓存基础配置
+    cachebase: {
+      dir: './cache/'.fullname(),
+      expire: 0,
+      max: 0,
+      persist: false, // 是否启用持久化
+      save_interval: 60000 // 保存间隔（毫秒）
+    },
+    // 缓存配置
+    redis: {
+      // 服务器地址
+      host: 'localhost',
+      // 端口号
+      port: 6379,
+      // 密码
+      password: 'asd159357',
+      // 数据库
+      database: 0,
+      // 前缀
+      prefix: 'mm_',
+      // 连接超时时间(ms)
+      connect_timeout: 5000,
+      // 重试次数
+      retry_count: 3,
+      // 重试间隔(ms)
+      retry_delay: 1000
+    },
+    llm_key: {
+      "deepseek": "sk-8e58b0e550714c929b0562d6cb9c1c04",
+      "openai": "sk-xxxx",
+      "glm": "sk-xxxx",
+      "minimax": "sk-xxxx",
+      "qwen": "sk-xxxx",
+      "doubao": "sk-xxxx",
+      "step": "sk-xxxx"
+    },
+    source_url: {
+      "deepseek": "https://api.deepseek.com",
+      "openai": "https://api.openai.com/v1",
+      "glm": "https://open.bigmodel.cn/api/paas/v4",
+      "minimax": "https://api.minimaxi.chat/v1",
+      "qwen": "https://dashscope.aliyuncs.com/compatible-mode/v1",
+      "doubao": "https://ark.cn-beijing.volces.com/api/v3",
+      "step": "https://api.stepfun.com/v1",
+      "cogvideox": "https://open.bigmodel.cn/api/paas/v4",
+      "comflyUI": "http://127.0.0.1:8188",
+      "stable_diffusion": "http://127.0.0.1:7860"
+    }
+  };
+
+  /**
+   * 构造函数
+   * @param {object} config 配置项
+   * @param {object} parent 父对象
+   */
+  constructor(config, parent) {
+    super({ ...MM_ai.config, ...config }, parent);
+    // 智能体集合（负责调用工具，相当于AI的手脚）
+    this.agent = {};
+    // 大模型集合（负责处理自然语言，充当AI的主脑）
+    this.llm = {};
+    // 技能集合（负责执行任务，相当于AI的能力）
+    this.skill = {};
+    // 工作集合（负责完成工作，相当于AI的具体行为）
+    this.work = {};
+    // 项目合集（负责管理项目，相当于AI团队任务）
+    this.project = {};
+    // 工具集合（负责调用内部服务，相当于AI的动手工具）
+    this.tool = {};
+    // MCP集合（负责调用外部服务，相当于AI的动手工具）
+    this.mcp = {};
+    // 资源集合（负责加载外部资源，如模型、数据等）
+    this.resource = {};
+    // 提示词集合（负责生成提示词，相当于AI的行为举止）
+    this.prompt = {};
+    // 常识知识集合（负责提供共享的领域常识，相当于AI的共识知识库）
+    this.knowledge = {};
+
+    // 管理器集合
+    this.manager = {};
+
+    // 是否是控制台模式，默认不是
+    this.is_console_mode = false;
+  }
+}
+
+/**
+ * 预置
+ */
+MM_ai.prototype._preset = function () {
+  // 数据库，调用统一数据库接口，也允许重定义数据库
+  this.sql = null;
+  // 缓存，调用统一缓存接口，也允许重定义缓存
+  this.cache = null;
+  // 文件管理，调用统一文件管理接口，也允许重定义文件管理
+  this.filer = null;
+  /**
+   * 获取服务器实例
+   * @returns {object} 服务器实例对象
+   */
+  this.getServer = function () {
+    return $.server;
+  };
+};
+
+/**
+ * 创建管理器实例
+ * @param {string} name 管理器名称
+ * @param {string} title 管理器标题
+ * @param {Function} cls 管理器类
+ * @param {string} root_dir 管理器所在目录，默认是ai
+ * @returns {object} 管理器实例
+ */
+MM_ai.prototype._createManager = function (name, title, cls, root_dir) {
+  var manager = new Manager(
+    {
+      name: name,
+      title: title,
+      filename: name + '.json',
+      tpl_dir: `./core/${name}/`.fullname(__dirname),
+      base_dir: `./common/${name}/`.fullname(__dirname),
+      dir: `./${root_dir || 'ai'}/${name}/`.fullname()
+    },
+    this,
+    this[name],
+    cls
+  );
+  // 设置到对应的管理器属性
+  this.manager[name] = manager;
+  return manager;
+}
+
+/**
+ * 初始化管理
+ */
+MM_ai.prototype._initManager = async function () {
+  // 管理器配置列表
+  var mgrConfigs = [
+    { name: 'llm', title: '大模型管理', Module: LLM },
+    { name: 'agent', title: '智能体管理器', Module: Agent },
+    { name: 'mcp', title: 'MCP管理器', Module: MCP },
+    { name: 'resource', title: '资源管理器', Module: Resource },
+    { name: 'prompt', title: '提示词管理器', Module: Prompt },
+    { name: 'skill', title: '技能管理器', Module: Skill },
+    { name: 'knowledge', title: '常识知识管理器', Module: Knowledge },
+    { name: 'work', title: '工作管理器', Module: Work },
+    { name: 'tool', title: '工具管理器', Module: Tool },
+    { name: 'project', title: '项目管理器', Module: Project, root_dir: '/' }
+  ];
+
+  // 管理器实例集合
+  var managers = [];
+
+  // 使用for循环初始化所有管理器
+  for (var i = 0; i < mgrConfigs.length; i++) {
+    var config = mgrConfigs[i];
+    var manager = this._createManager(config.name, config.title, config.Module, config.root_dir);
+    managers.push(manager);
+  }
+
+  // 并行执行所有管理器的初始化操作
+  var promises = [];
+  for (var j = 0; j < managers.length; j++) {
+    promises.push(managers[j].do('init'));
+  }
+
+  await Promise.all(promises);
+};
+
+/**
+ * 初始化资源
+ */
+MM_ai.prototype._initSources = async function () {
+  // 初始化顺序 llm > prompt > resource > project > mcp > tool > skill > agent
+  await Promise.all([
+    this.manager.llm.runAll('init', this.getEventer(), this.getLogger()),
+    this.manager.prompt.runAll('init', this.getEventer(), this.getLogger()),
+    this.manager.resource.runAll('init', this, this.getEventer(), this.getLogger()),
+    this.manager.mcp.runAll('init', this, this.getEventer(), this.getLogger()),
+    this.manager.tool.runAll('init', this, this.getEventer(), this.getLogger()),
+    this.manager.knowledge.runAll('init', this, this.getEventer(), this.getLogger()),
+    this.manager.work.runAll('init', this.getEventer(), this.getLogger()),
+    this.manager.project.runAll('init', this.getEventer(), this.getLogger())
+  ]);
+  await this.manager.skill.runAll('init', this.tool, this.mcp, this.knowledge, this.work, this.getEventer(), this.getLogger());
+  await this.manager.agent.runAll('init', this.llm, this.skill, this.tool, this.prompt, this.getEventer(), this.getLogger());
+};
+
+/**
+ * 加载资源
+ */
+MM_ai.prototype._loadSources = async function () {
+  // 加载无需依赖关系，所以可以同时加载，内部并行执行
+  await Promise.all([
+    this.manager.llm.runAll('load'),
+    this.manager.agent.runAll('load'),
+    this.manager.mcp.runAll('load'),
+    this.manager.prompt.runAll('load'),
+    this.manager.resource.runAll('load'),
+    this.manager.tool.runAll('load'),
+    this.manager.work.runAll('load'),
+    this.manager.knowledge.runAll('load'),
+    this.manager.skill.runAll('load'),
+    this.manager.project.runAll('load')
+  ]);
+};
+
+/**
+ * 启动资源
+ */
+MM_ai.prototype._startSources = async function () {
+  // 启动无顺序，但需要等待所有启动完成，内部依次执行
+  await Promise.all([
+    this.manager.llm.runAll('start'),
+    this.manager.prompt.runAll('start'),
+    this.manager.mcp.runAll('start'),
+    this.manager.agent.runAll('start'),
+    this.manager.resource.runAll('start'),
+    this.manager.tool.runAll('start'),
+    this.manager.work.runAll('start'),
+    this.manager.knowledge.runAll('start'),
+    this.manager.skill.runAll('start'),
+    this.manager.project.runAll('start')
+  ]);
+};
+
+/**
+ * 初始化基础设施
+ */
+MM_ai.prototype._initModules = function () {
+  // 模板引擎
+  require('mm_tpl');
+};
+
+/**
+ * 初始化工具
+ */
+MM_ai.prototype._initTool = function () {
+  this.logger = new Log(this.config.log);
+  $.log = this.logger;
+  $.config = new Config();
+  $.config.setOptions(this.config);
+};
+
+/**
+ * 初始化基础设施
+ */
+MM_ai.prototype._initBase = async function () {
+  if (!this.sql || this.config.diy_sql) {
+    // 获取数据库配置
+    var db = { type: 'mysql', ...this.config.sql };
+    if (this.config.mysql) {
+      db.mysql = this.config.mysql;
+    }
+    if (this.config.sqlite) {
+      db.sqlite = this.config.sqlite;
+    }
+    this.sql = sqlAdmin('sys', db);
+    $.sql = this.sql;
+    await $.sql.open();
+  }
+  if (!this.cache || this.config.diy_cache) {
+    // 获取缓存配置
+    var cache = { type: 'memory', ...this.config.cache };
+    if (this.config.cachebase) {
+      cache.cachebase = this.config.cachebase;
+    }
+    if (this.config.redis) {
+      cache = { ...cache, ...this.config.redis };
+    }
+    if (this.config.mongodb) {
+      cache.mongodb = this.config.mongodb;
+    }
+    this.cache = cacheAdmin('sys', cache);
+    $.cache = this.cache;
+    await $.cache.connect();
+  }
+
+  if (!this.filer || this.config.diy_filer) {
+    this.filer = $.file;
+  }
+};
+
+/**
+ * 初始化
+ * @param {object} server - 服务器对象
+ * @param {object} eventer - 事件总线对象
+ * @param {object} logger - 日志对象
+ * @private
+ */
+MM_ai.prototype._initCore = async function (server, eventer, logger) {
+  // 注册全局异常处理器，防止未捕获异常导致进程崩溃
+  process.on('uncaughtException', function (err) {
+    console.error('未捕获的异常:', err);
+  });
+  process.on('unhandledRejection', function (reason) {
+    console.error('未处理的 Promise 拒绝:', reason);
+  });
+
+  if (logger) {
+    this.getLogger = function () {
+      return logger;
+    };
+  }
+  if (eventer) {
+    this.getEventer = function () {
+      return eventer;
+    };
+  }
+  if (server) {
+    this.getServer = function () {
+      return server;
+    };
+    this.sql = server.sql;
+    this.cache = server.cache;
+    this.filer = server.filer;
+  }
+  // 初始化工具
+  this._initTool();
+  // 初始化模块
+  this._initModules();
+  // 初始化基础设施
+  await this._initBase();
+  // 初始化管理器
+  await this._initManager();
+  // 加载资源
+  await this._loadSources();
+  // 初始化资源
+  await this._initSources();
+};
+
+/**
+ * 启动
+ */
+MM_ai.prototype._startCore = async function () {
+  // 启动资源
+  await this._startSources();
+};
+
+/**
+ * 聊天方法
+ * @param {string} content - 输入内容
+ * @returns {Promise<object>} - 智能体响应
+ */
+MM_ai.prototype.chat = async function (content) {
+  let agent = this.getAgent(this.agent_name);
+  if (!agent) {
+    throw new Error('智能体不存在');
+  }
+  // 处理用户输入
+  let res = await agent.chat(this.user_name, '', content);
+  return res;
+};
+
+/**
+ * 获取智能体
+ * @param {string} agent_name - 智能体名称
+ * @returns {object} - 智能体对象
+ */
+MM_ai.prototype.getAgent = function (agent_name) {
+  return this.agent[agent_name || this.agent_name];
+};
+
+/**
+ * 获取项目
+ * @param {string} project_name - 项目名称
+ * @returns {object} - 项目对象
+ */
+MM_ai.prototype.getProject = function (project_name) {
+  return this.project[project_name];
+};
+
+/**
+ * 获取用户信息
+ * @param {string} agent_name - 智能体名称(英文名，唯一标识)
+ * @param {string} user_name - 用户名称(英文名，唯一标识)
+ * @returns {Object} - 包含用户信息的对象
+ */
+MM_ai.prototype.getUser = function (agent_name, user_name) {
+  let agent = this.getAgent(agent_name);
+  if (!agent) {
+    throw new Error(`智能体${agent_name}不存在`);
+  }
+  return agent.getUser(user_name);
+};
+
+/**
+ * 获取任务
+ * @param {string} project_name - 项目名称(英文名，唯一标识)
+ * @param {string} task_name - 任务名称(英文名，唯一标识)
+ * @returns {object} - 任务对象
+ */
+MM_ai.prototype.getTask = function (project_name, task_name) {
+  let project = this.getProject(project_name);
+  if (!project) {
+    throw new Error(`项目${project_name}不存在`);
+  }
+  return project.getTask(task_name);
+};
+
+/**
+ * 获取技能
+ * @param {string} skill_name - 技能名称(英文名，唯一标识)
+ * @returns {object} - 技能对象
+ */
+MM_ai.prototype.getSkill = function (skill_name) {
+  return this.skill[skill_name];
+};
+
+/**
+ * 获取工具
+ * @param {string} tool_name - 工具名称(英文名，唯一标识)
+ * @returns {object} - 工具对象
+ */
+MM_ai.prototype.getTool = function (tool_name) {
+  return this.tool[tool_name];
+};
+
+/**
+ * 获取LLM模型
+ * @param {string} llm_name - LLM模型名称
+ * @returns {object} - LLM模型对象
+ */
+MM_ai.prototype.getLLM = function (llm_name) {
+  return this.llm[llm_name];
+};
+
+/**
+ * 获取工作
+ * @param {string} work_name - 工作名称(英文名，唯一标识)
+ * @returns {object} - 工作对象
+ */
+MM_ai.prototype.getWork = function (work_name) {
+  return this.work[work_name];
+};
+
+/**
+ * 获取提示词管理器
+ * @param {string} prompt_name - 提示词管理器名称(英文名，唯一标识)
+ * @returns {object} - 工作对象
+ */
+MM_ai.prototype.getPrompt = function (prompt_name) {
+  return this.prompt[prompt_name];
+};
+
+/**
+ * 获取技能模块
+ * @param {string} name - 技能名称
+ * @returns {Promise} - 技能模块
+ */
+MM_ai.prototype.getSkill = function (skill_name) {
+  return this.skill[skill_name];
+};
+
+/**
+ * 控制台聊天交互
+ */
+MM_ai.prototype.console = async function () {
+  console.log('AI聊天机器人已启动，输入"退出"或"quit"结束对话;输入"清空"或"clear"清空控制台内容;\n');
+  let chat = new Chat({}, this);
+  await chat.run();
+};
+
+/**
+ * 运行服务器
+ */
+MM_ai.prototype.run = async function () {
+  try {
+    let state = this.getState();
+    if (state === 'created' || state === 'creating') {
+      await this.do('init');
+    }
+    await this.do('start');
+    this.log('info', '服务器启动成功');
+  } catch (error) {
+    this.log('error', '服务器启动失败:', error);
+    throw error;
+  }
+};
+
+/**
+ * 注册事件
+ * @param {string} event - 事件名称
+ * @param {function} callback - 事件回调函数
+ */
+MM_ai.prototype.onEvent = function (event, callback) {
+  this.getEventer().on(event, callback);
+};
+
+module.exports = {
+  MM_ai
+};

package/package.json

@@ -0,0 +1,49 @@
+{
+  "name": "mm_ai",
+  "version": "1.0.0",
+  "description": "AI聊天模块",
+  "main": "chat.js",
+  "scripts": {
+    "start": "node ./example/index.js",
+    "test": "node ./example/index.js",
+    "dev": "nodemon ./example/index.js"
+  },
+  "dependencies": {
+    "@mariox/weather-mcp-server": "^1.1.1",
+    "abcjs": "^6.6.3",
+    "agent-browser": "^0.31.1",
+    "compressing": "^2.1.1",
+    "doc-to-md-rag": "^1.0.1",
+    "docx": "^9.7.1",
+    "koa": "^3.2.1",
+    "koa-body": "^8.0.0",
+    "koa-send": "^5.0.1",
+    "mm_cache": "^1.5.0",
+    "mm_expand": "^2.4.4",
+    "mm_html": "^1.1.7",
+    "mm_https": "^1.6.6",
+    "mm_logs": "^1.6.9",
+    "mm_machine": "^2.9.0",
+    "mm_sql": "^1.5.7",
+    "openai": "^6.45.0",
+    "playwright": "^1.61.1",
+    "robotjs": "^0.7.1",
+    "sharp": "^0.35.3",
+    "tencentcloud-sdk-nodejs": "^4.1.264",
+    "xlsx": "^0.18.5"
+  },
+  "keywords": [
+    "ai",
+    "chat",
+    "deepseek"
+  ],
+  "author": "",
+  "license": "MIT",
+  "devDependencies": {
+    "@types/jest": "^30.0.0",
+    "eslint-formatter-table": "^7.32.1",
+    "eslint-plugin-jsdoc": "^63.0.12",
+    "jest": "^30.4.2",
+    "mm_eslint": "^1.7.5"
+  }
+}