mm_os 4.2.9 → 4.3.0

package/ai.js

@@ -0,0 +1,130 @@
+/**
+ * ai.js
+ * AI模块管理类
+ * 负责加载并初始化mm_ai模块，桥接前后端AI对话
+ */
+
+const path = require('path');
+const fs = require('fs');
+
+async function tryRequire() {
+  try {
+    return require('mm_ai');
+  } catch (error) {
+    console.log('mm_ai模块未安装，尝试安装...');
+    await require('npm').exec('npm i mm_ai');
+    return require('mm_ai');
+  }
+}
+
+/**
+ * 覆写 Agent.prototype.filterMessages，将消息中的 /file/ 路径转为 LLM 可读格式
+ * @param {Array} messages - 消息记录
+ * @returns {Array} 转换后的消息记录
+ */
+function overrideFilterMessages(messages) {
+  if (!messages || !messages.length) {
+    return messages;
+  }
+  var result = [];
+  for (var i = 0; i < messages.length; i++) {
+    var msg = messages[i];
+    if (msg.role !== 'user' || typeof msg.content !== 'string') {
+      result.push(msg);
+      continue;
+    }
+    var content = msg.content;
+    var parts = [];
+    var last_idx = 0;
+    var file_re = /!?\[([^\]]*)\]\((\/file\/[^)]+)\)/g;
+    var match;
+    while ((match = file_re.exec(content)) !== null) {
+      if (match.index > last_idx) {
+        parts.push({ type: 'text', text: content.substring(last_idx, match.index) });
+      }
+      var alt = match[1] || '文件';
+      var file_url = match[2];
+      var rel_path = '.' + file_url;
+      var full_path = rel_path.fullname($.run_path);
+      if (fs.existsSync(full_path)) {
+        var ext = path.extname(full_path).toLowerCase();
+        var img_exts = ['.jpg', '.jpeg', '.png', '.gif', '.webp', '.bmp', '.svg'];
+        var text_exts = ['.txt', '.js', '.ts', '.py', '.json', '.md', '.css', '.html', '.xml',
+          '.yaml', '.yml', '.ini', '.cfg', '.conf', '.sh', '.bat', '.ps1', '.sql',
+          '.java', '.go', '.rs', '.rb', '.php', '.c', '.cpp', '.h', '.hpp',
+          '.log', '.csv', '.env', '.toml'];
+        if (img_exts.indexOf(ext) !== -1) {
+          try {
+            var img_data = fs.readFileSync(full_path);
+            var base64 = img_data.toString('base64');
+            var mime = 'image/' + (ext === '.jpg' ? 'jpeg' : ext.replace('.', ''));
+            parts.push({ type: 'image_url', image_url: { url: 'data:' + mime + ';base64,' + base64 } });
+          } catch (e) {
+            parts.push({ type: 'text', text: '[图片加载失败: ' + alt + ']' });
+          }
+        } else if (text_exts.indexOf(ext) !== -1) {
+          try {
+            var file_content = fs.readFileSync(full_path, 'utf-8');
+            parts.push({ type: 'text', text: '--- ' + alt + ' ---\n' + file_content + '\n---' });
+          } catch (e) {
+            parts.push({ type: 'text', text: '[文件读取失败: ' + alt + ']' });
+          }
+        } else {
+          parts.push({ type: 'text', text: '[文件: ' + alt + ']' });
+        }
+      } else {
+        parts.push({ type: 'text', text: match[0] });
+      }
+      last_idx = match.index + match[0].length;
+    }
+    if (last_idx < content.length) {
+      parts.push({ type: 'text', text: content.substring(last_idx) });
+    }
+    var user_content;
+    if (parts.length === 0) {
+      user_content = content;
+    } else if (parts.length === 1 && parts[0].type === 'text') {
+      user_content = parts[0].text;
+    } else {
+      user_content = parts;
+    }
+    result.push({ ...msg, content: user_content });
+  }
+  return result;
+}
+
+/**
+ * 初始化AI模块
+ * 创建MM_ai实例（mm_ai内部已注册chat:in事件处理）
+ * @param {object} server 服务器实例
+ */
+async function initAI(server) {
+  // 检查AI开关
+  if (!server.config.sys || !server.config.sys.ai) {
+    return null;
+  }
+
+  try {
+    var mod = await tryRequire();
+    if (!mod || !mod.MM_ai) {
+      server.log('warn', 'mm_ai模块加载失败');
+      return null;
+    }
+
+    // 覆写 Agent.prototype.filterMessages 处理文件路径
+    var { Agent } = require('mm_ai/core/agent/index.js');
+    Agent.prototype.filterMessages = overrideFilterMessages;
+
+    var ai = new mod.MM_ai(server.config, server);
+    server.ai = ai;
+    await ai.run();
+    return ai;
+  } catch (error) {
+    server.log('error', 'AI模块加载失败', error);
+    return null;
+  }
+}
+
+module.exports = {
+  initAI
+};

package/package.json

@@ -1,7 +1,7 @@
 {
 	"name": "mm_os",
 	"description": "MM_OS服务端架构，用于快速构建应用程序，支持网站建设、小程序后台、AI应用、物联网(IOT/AIOT)、游戏服务端等多种场景。",
-	"version": "4.2.9",
+	"version": "4.3.0",
 	"main": "index.js",
 	"keywords": [
 		"AI",
@@ -79,6 +79,7 @@
 		"core/",
 		"ulits/",
 		"tcp.js",
-		"software.js"
+		"software.js",
+		"ai.js"
 	]
 }