Upload 64 files

ai-tools.js

@@ -9,22 +9,22 @@ const mongoTools = [
     functionDeclarations: [
       {
         name: "query_database",
-        description: "查询学校数据库中的信息。当用户询问具体的学生、成绩、考勤或班级数据时，必须使用此工具。支持的集合(collections): 'Student'(学生), 'Score'(成绩), 'Attendance'(考勤), 'Class'(班级)。",
+        description: "查询学校数据库。⚠️字段名必须准确：班级字段名是 'className' (禁止用 'class')，学生姓名在Student表是 'name'，在成绩表是 'studentName'。支持的集合: 'Student', 'Score', 'Attendance', 'Class'。",
         parameters: {
           type: "OBJECT",
           properties: {
             collection: {
               type: "STRING",
-              description: "要查询的集合名称，例如 'Student', 'Score', 'Attendance'。",
+              description: "集合名称",
               enum: ["Student", "Score", "Attendance", "Class"]
             },
             filter: {
               type: "OBJECT",
-              description: "Mongoose/MongoDB 查询过滤条件的 JSON 对象。例如: {name: '张三'} 或 {score: {$lt: 60}}。不要包含 schoolId，系统会自动注入。",
+              description: "Mongoose查询条件JSON。例: {className:'一年级(1)班', name:'张三'}。注意：查询班级时必须使用 'className' 字段。",
             },
             limit: {
               type: "NUMBER",
-              description: "限制返回条数，默认 5，最大 20。"
+              description: "默认 5"
             }
           },
           required: ["collection", "filter"]
@@ -58,9 +58,9 @@ function injectSecurityFilter(filter, user, role, schoolId) {
         return safeFilter;
     }
 
+    // 老师只能看自己相关的班级逻辑可以在这里加强
     if (role === 'TEACHER') {
-        // 简单权限控制：老师只能查自己相关，或全校公开数据
-        // 实际逻辑可根据需求扩展
+       // 暂时不做强制过滤，依赖业务层逻辑
     }
 
     if (role === 'STUDENT') {
@@ -72,12 +72,41 @@ function injectSecurityFilter(filter, user, role, schoolId) {
     return safeFilter;
 }
 
+/**
+ * 辅助：递归修正查询字段名 (Deep Normalization)
+ * AI 经常把 className 写成 class，把 studentName 写成 name，这里做统一修正
+ */
+function normalizeQueryFields(query, collection) {
+    if (!query || typeof query !== 'object') return query;
+    
+    if (Array.isArray(query)) {
+        return query.map(item => normalizeQueryFields(item, collection));
+    }
+
+    const newQuery = {};
+    for (const key in query) {
+        let newKey = key;
+        
+        // 1. 修正班级字段: class -> className
+        if (key === 'class') newKey = 'className';
+
+        // 2. 修正名字字段: Score表里叫 studentName, Student表里叫 name
+        if (collection === 'Score' || collection === 'Attendance') {
+            if (key === 'name') newKey = 'studentName';
+        }
+
+        // 递归处理值 (例如 $or 数组内部的对象)
+        newQuery[newKey] = normalizeQueryFields(query[key], collection);
+    }
+    return newQuery;
+}
+
 /**
  * 3. 工具执行器 (Executor)
  */
 async function executeMongoTool(functionCall, user, role, schoolId) {
-    // 兼容 OpenAI 格式 (arguments 是字符串) 和 Gemini 格式 (args 是对象)
     let args = functionCall.args;
+    // 兼容 OpenAI 格式
     if (typeof functionCall.arguments === 'string') {
         try {
             args = JSON.parse(functionCall.arguments);
@@ -89,17 +118,20 @@ async function executeMongoTool(functionCall, user, role, schoolId) {
 
     const { collection, filter = {}, limit = 5 } = args || {};
     
+    // 🛠️ 关键修复：在注入安全字段前，先修正 AI 的字段命名错误
+    const normalizedFilter = normalizeQueryFields(filter, collection);
+
     // 🛡️ 安全注入
-    const safeFilter = injectSecurityFilter(filter, user, role, schoolId);
+    const safeFilter = injectSecurityFilter(normalizedFilter, user, role, schoolId);
     const safeLimit = Math.min(Math.max(limit, 1), 20); 
 
-    // --- 🔍 MCP LOGGING START ---
+    // --- 🔍 MCP LOGGING ---
     console.log(`\n================= [MCP TOOL CALL] =================`);
     console.log(`🛠️  Tool:        query_database`);
     console.log(`📂 Collection:  ${collection}`);
-    console.log(`📥 AI Params:   ${JSON.stringify(filter)}`);
-    console.log(`🔒 Safe Query:  ${JSON.stringify(safeFilter)}`);
-    console.log(`👤 User Role:   ${role} (${user.username})`);
+    console.log(`📥 AI Params:   ${JSON.stringify(filter)}`); // 原始
+    console.log(`🔧 Normalized:  ${JSON.stringify(normalizedFilter)}`); // 修正后
+    console.log(`🔒 Safe Query:  ${JSON.stringify(safeFilter)}`); // 最终
     console.log(`---------------------------------------------------`);
 
     try {
@@ -109,10 +141,17 @@ async function executeMongoTool(functionCall, user, role, schoolId) {
         switch (collection) {
             case "Student":
                 fields = "name studentNo className gender flowerBalance seatNo -_id";
+                // 模糊搜索支持
+                if (safeFilter.name && !safeFilter.name.$regex) {
+                    safeFilter.name = { $regex: safeFilter.name, $options: 'i' };
+                }
                 result = await Student.find(safeFilter).select(fields).limit(safeLimit).lean();
                 break;
             case "Score":
                 fields = "studentName courseName score type examName -_id";
+                if (safeFilter.studentName && !safeFilter.studentName.$regex) {
+                    safeFilter.studentName = { $regex: safeFilter.studentName, $options: 'i' };
+                }
                 result = await Score.find(safeFilter).select(fields).sort({ _id: -1 }).limit(safeLimit).lean();
                 break;
             case "Attendance":
@@ -125,24 +164,24 @@ async function executeMongoTool(functionCall, user, role, schoolId) {
                 break;
             default:
                 console.log(`❌ [MCP ERROR] Unknown collection: ${collection}`);
-                console.log(`===================================================\n`);
                 return { error: "Unknown collection" };
         }
 
         console.log(`✅ [MCP SUCCESS] Found ${result.length} records.`);
         if (result.length > 0) {
-            console.log(`📄 Sample Data: ${JSON.stringify(result[0])}`);
+            console.log(`📄 Sample: ${JSON.stringify(result[0])}`);
+        } else {
+            console.log(`⚠️ No records found. Hint: Check if schoolId matches.`);
         }
         console.log(`===================================================\n`);
 
         if (result.length === 0) {
-            return { info: "未找到符合条件的数据。" };
+            return { info: "未找到符合条件的数据。请确认查询条件是否准确（如姓名是否正确）。" };
         }
         return result;
 
     } catch (error) {
         console.error("❌ [MCP EXCEPTION]", error.message);
-        console.log(`===================================================\n`);
         return { error: "Database query failed", details: error.message };
     }
 }