Upload 63 files

ai-context.js

@@ -0,0 +1,235 @@
+
+const { 
+    User, Student, Score, AttendanceModel, ClassModel, 
+    LeaveRequestModel, TodoModel, School 
+} = require('./models');
+
+/**
+ * 格式化当前日期
+ */
+const getCurrentDateInfo = () => {
+    const now = new Date();
+    const days = ['周日', '周一', '周二', '周三', '周四', '周五', '周六'];
+    return `${now.getFullYear()}年${now.getMonth() + 1}月${now.getDate()}日 ${days[now.getDay()]}`;
+};
+
+/**
+ * 构建学生画像上下文 (学生视角)
+ */
+async function buildStudentContext(username, schoolId) {
+    const student = await Student.findOne({ 
+        $or: [{ studentNo: username }, { name: username }],
+        schoolId 
+    });
+
+    if (!student) return "无法找到该学生的详细档案。";
+
+    // 1. 获取近期成绩 (最近10条，让AI掌握更多趋势)
+    const recentScores = await Score.find({ 
+        studentNo: student.studentNo, 
+        schoolId 
+    }).sort({ _id: -1 }).limit(10);
+
+    // 2. 获取考勤概况
+    const attendanceStats = await AttendanceModel.aggregate([
+        { $match: { studentId: student._id.toString() } },
+        { $group: { _id: "$status", count: { $sum: 1 } } }
+    ]);
+    const absentCount = attendanceStats.find(a => a._id === 'Absent')?.count || 0;
+    const leaveCount = attendanceStats.find(a => a._id === 'Leave')?.count || 0;
+
+    // 3. 获取待办事项
+    const user = await User.findOne({ username, schoolId });
+    const todos = user ? await TodoModel.find({ userId: user._id, isCompleted: false }).limit(5) : [];
+
+    let prompt = `
+### 当前用户身份：学生 (个人视图)
+- **姓名**: ${student.name}
+- **班级**: ${student.className}
+- **学号**: ${student.studentNo}
+- **积分(小红花)**: ${student.flowerBalance} 🌺
+
+### 个人学习数据
+`;
+
+    if (recentScores.length > 0) {
+        prompt += `- **近期成绩历史**: ${recentScores.map(s => `${s.courseName}: ${s.score} (${s.type || '考试'})`).join('; ')}\n`;
+        // 计算简单平均分
+        const avg = (recentScores.reduce((acc, s) => acc + s.score, 0) / recentScores.length).toFixed(1);
+        prompt += `- **近期平均分**: ${avg}\n`;
+    } else {
+        prompt += `- **近期成绩**: 暂无记录\n`;
+    }
+
+    if (absentCount > 0 || leaveCount > 0) {
+        prompt += `- **考勤异常**: 本学期缺勤 ${absentCount} 次，请假 ${leaveCount} 次。\n`;
+    } else {
+        prompt += `- **考勤状况**: 全勤，表现极佳。\n`;
+    }
+
+    if (todos.length > 0) {
+        prompt += `- **未完成待办**: ${todos.map(t => t.content).join('; ')}\n`;
+    }
+
+    return prompt;
+}
+
+/**
+ * 构建教师画像上下文 (增强版 - 全班详情)
+ */
+async function buildTeacherContext(username, schoolId) {
+    const user = await User.findOne({ username, schoolId });
+    if (!user) return "无法找到该教师档案。";
+
+    const className = user.homeroomClass;
+    let prompt = `
+### 当前用户身份：教师
+- **姓名**: ${user.trueName || username}
+- **任教科目**: ${user.teachingSubject || '未设置'}
+`;
+
+    if (className) {
+        // 1. 获取全班学生列表
+        const students = await Student.find({ className, schoolId });
+        const studentNos = students.map(s => s.studentNo);
+        const studentIds = students.map(s => s._id.toString());
+
+        // 2. 获取全班考勤统计 (Group by Student)
+        const attendanceRaw = await AttendanceModel.aggregate([
+            { $match: { studentId: { $in: studentIds }, status: { $in: ['Absent', 'Leave'] } } },
+            { $group: { _id: "$studentId", absent: { $sum: { $cond: [{ $eq: ["$status", "Absent"] }, 1, 0] } }, leave: { $sum: { $cond: [{ $eq: ["$status", "Leave"] }, 1, 0] } } } }
+        ]);
+        const attendanceMap = {};
+        attendanceRaw.forEach(a => attendanceMap[a._id] = a);
+
+        // 3. 获取全班近期成绩 (为了Token效率，我们只取每人最近3次考试或全班最近的100条成绩记录)
+        const recentScores = await Score.find({ schoolId, studentNo: { $in: studentNos } }).sort({ _id: -1 }).limit(200);
+        const scoreMap = {}; // studentNo -> [Score Obj]
+        recentScores.forEach(s => {
+            if (!scoreMap[s.studentNo]) scoreMap[s.studentNo] = [];
+            scoreMap[s.studentNo].push(s);
+        });
+
+        // 4. 构建“全息花名册”字符串
+        let rosterStr = "";
+        const failingStudents = [];
+
+        students.forEach(s => {
+            const att = attendanceMap[s._id.toString()] || { absent: 0, leave: 0 };
+            const myScores = scoreMap[s.studentNo] || [];
+            
+            // 计算个人概况
+            let scoreStr = "暂无成绩";
+            let avgScore = 0;
+            if (myScores.length > 0) {
+                // 只取最近3门
+                const latest3 = myScores.slice(0, 3);
+                scoreStr = latest3.map(sc => `${sc.courseName}:${sc.score}`).join(', ');
+                avgScore = latest3.reduce((a,b)=>a+b.score,0) / latest3.length;
+                
+                // 检查不及格
+                if (latest3.some(sc => sc.score < 60)) {
+                    failingStudents.push(s.name);
+                }
+            }
+
+            // 格式: [姓名](座号): 考勤[缺x, 假y], 成绩[科目:分, ...], 积分: z
+            rosterStr += `- **${s.name}** (座号:${s.seatNo || '-'}): 考勤[缺${att.absent}/假${att.leave}], 小红花:${s.flowerBalance}, 近期成绩[${scoreStr}]\n`;
+        });
+
+        // 5. 待办审批
+        const pendingLeaves = await LeaveRequestModel.find({ className, schoolId, status: 'Pending' }).limit(5);
+        
+        prompt += `- **班级**: ${className} (共${students.length}人)\n`;
+        
+        if (pendingLeaves.length > 0) {
+            prompt += `\n### 🔴 紧急待办\n`;
+            prompt += `你有 ${pendingLeaves.length} 条请假申请待审批 (申请人: ${pendingLeaves.map(l => l.studentName).join(', ')})。\n`;
+        }
+
+        if (failingStudents.length > 0) {
+            prompt += `\n### ⚠️ 学情预警\n`;
+            prompt += `以下学生近期有不及格记录，请重点关注: ${failingStudents.join(', ')}\n`;
+        }
+
+        prompt += `\n### 📋 班级学生全息档案 (Roster)\n`;
+        prompt += `(这是你班级所有学生的详细数据，当用户询问具体学生时，请在此处检索)\n`;
+        prompt += rosterStr;
+
+    } else {
+        prompt += `- **班级**: 目前没有担任班主任，暂无详细学生数据。\n`;
+    }
+
+    return prompt;
+}
+
+/**
+ * 构建管理员/校长画像上下文
+ */
+async function buildAdminContext(role, schoolId) {
+    let prompt = `### 当前用户身份：${role === 'PRINCIPAL' ? '校长' : '超级管理员'}\n`;
+    
+    if (role === 'PRINCIPAL' && schoolId) {
+        const school = await School.findById(schoolId);
+        const totalStudents = await Student.countDocuments({ schoolId });
+        const totalTeachers = await User.countDocuments({ schoolId, role: 'TEACHER' });
+        
+        // 今日缺勤详细名单
+        const today = new Date().toISOString().split('T')[0];
+        const absences = await AttendanceModel.find({ schoolId, date: today, status: { $in: ['Absent', 'Leave'] } });
+        const absentNames = absences.map(a => `${a.studentName}(${a.className})`).join(', ');
+
+        // 全校均分
+        const recentScores = await Score.find({ schoolId }).sort({_id:-1}).limit(100);
+        let avgScore = 0;
+        if (recentScores.length) avgScore = (recentScores.reduce((a,b)=>a+b.score,0)/recentScores.length).toFixed(1);
+
+        prompt += `- **学校**: ${school ? school.name : '未知'}\n`;
+        prompt += `- **宏观数据**: 教师 ${totalTeachers} 人，学生 ${totalStudents} 人，近期全校抽样平均分 ${avgScore}。\n`;
+        prompt += `- **今日出勤**: 缺勤/请假 ${absences.length} 人。名单: ${absentNames || '无'}。\n`;
+    }
+
+    return prompt;
+}
+
+/**
+ * 主入口：构建用户上下文 Prompt
+ * @param {string} username - 请求头中的用户名
+ * @param {string} role - 请求头中的角色
+ * @param {string} schoolId - 请求头中的学校ID
+ */
+async function buildUserContext(username, role, schoolId) {
+    try {
+        const dateStr = getCurrentDateInfo();
+        let roleContext = "";
+
+        if (role === 'STUDENT') {
+            roleContext = await buildStudentContext(username, schoolId);
+        } else if (role === 'TEACHER') {
+            roleContext = await buildTeacherContext(username, schoolId);
+        } else if (role === 'ADMIN' || role === 'PRINCIPAL') {
+            roleContext = await buildAdminContext(role, schoolId);
+        }
+
+        // 组装最终 System Instruction 片段
+        return `
+---
+【上下文注入信息 (Context Injection) - 绝密】
+当前系统时间: ${dateStr}
+以下是当前用户的核心数据和其管辖范围内的详细档案。
+${roleContext}
+
+【AI 行为准则】
+1. 你拥有上述所有数据的“上帝视角”。当老师问“张三的情况”时，请直接从【班级学生全息档案】中提取张三的成绩、考勤和小红花数据进行回答，不要说“我不知道”。
+2. 回答要具体。例如：不要说“他成绩一般”，要说“他最近数学考了60分，英语考了85分，属于偏科现象”。
+3. 如果数据中显示学生有缺勤或不及格，请在回答末尾给出具体的教学干预建议。
+4. 保持语气专业、辅助性强。
+---
+`;
+    } catch (e) {
+        console.error("Context build failed:", e);
+        return ""; // 失败时降级为空，不影响主流程
+    }
+}
+
+module.exports = { buildUserContext };

ai-routes.js

@@ -3,6 +3,7 @@ const express = require('express');
 const router = express.Router();
 const OpenAI = require('openai');
 const { ConfigModel, User, AIUsageModel } = require('./models');
+const { buildUserContext } = require('./ai-context');
 
 // ... (Key Management, Usage Tracking, Helpers, Provider Management functions remain same as before)
 // Fetch keys from DB + merge with ENV variables
@@ -165,7 +166,6 @@ async function streamOpenRouter(baseParams, res) {
             
             try {
                 console.log(`[AI] 🚀 Attempting ${providerLabel} Model: ${modelName} (URL: ${baseURL})`);
-                // console.log(`[AI] Payload Messages:`, JSON.stringify(messages).substring(0, 200) + "..."); // Debug log
                 
                 const stream = await client.chat.completions.create({ model: modelName, messages, stream: true });
                 
@@ -188,7 +188,6 @@ async function streamOpenRouter(baseParams, res) {
                     console.log(`[AI] 🔄 Rate limit/Quota for ${modelName}, switching...`);
                     break; // Switch to next provider/model logic if implemented in loop, here break to next model in loop
                 }
-                // For OpenAI client, some errors might be specific
             }
         }
     }
@@ -292,9 +291,6 @@ async function streamContentWithSmartFallback(baseParams, res) {
                 deprioritizeProvider(provider); 
                 continue; 
             }
-            // If it's a hard error (e.g. network), we might also want to switch, 
-            // but strict quota error check usually suffices for fallback logic.
-            // For robustness, let's allow fallback on most errors in the loop:
             continue; 
         }
     }
@@ -320,8 +316,6 @@ router.get('/live-access', checkAIAccess, async (req, res) => {
     try {
         const keys = await getKeyPool('gemini');
         if (keys.length === 0) return res.status(503).json({ error: 'No API keys available' });
-        // Return the first available key. In a real prod environment, you might issue a short-lived proxy token.
-        // For this architecture, we return the key to allow direct WebSocket connection.
         res.json({ key: keys[0] });
     } catch (e) { res.status(500).json({ error: e.message }); }
 });
@@ -352,6 +346,12 @@ router.post('/reset-pool', checkAIAccess, (req, res) => {
 
 router.post('/chat', checkAIAccess, async (req, res) => {
     const { text, audio, history } = req.body;
+    
+    // Extract headers for context building
+    const username = req.headers['x-user-username'];
+    const userRole = req.headers['x-user-role'];
+    const schoolId = req.headers['x-school-id'];
+
     res.setHeader('Content-Type', 'text/event-stream');
     res.setHeader('Cache-Control', 'no-cache');
     res.setHeader('Connection', 'keep-alive');
@@ -367,9 +367,15 @@ router.post('/chat', checkAIAccess, async (req, res) => {
         if (currentParts.length === 0) currentParts.push({ text: "Hello" });
         fullContents.push({ role: 'user', parts: currentParts });
 
+        // --- NEW: Inject Context ---
+        const contextPrompt = await buildUserContext(username, userRole, schoolId);
+        const baseSystemInstruction = "你是一位友善、耐心且知识渊博的中小学AI助教。请用简洁、鼓励性的语言回答学生的问题。回复支持 Markdown 格式。";
+        const combinedSystemInstruction = `${baseSystemInstruction}\n${contextPrompt}`;
+        // ---------------------------
+
         const answerText = await streamContentWithSmartFallback({
             contents: fullContents, 
-            config: { systemInstruction: "你是一位友善、耐心且知识渊博的中小学AI助教。请用简洁、鼓励性的语言回答学生的问题。回复支持 Markdown 格式。" }
+            config: { systemInstruction: combinedSystemInstruction }
         }, res);
 
         if (answerText) {