Agent_For_Supmea/scripts/ai.js

// scripts/ai.js
import { COMMANDS } from './commands.js';

export async function translateToCommand(userInput) {
    const availableKeys = COMMANDS.map(c => c.key).join(', ');
    
    const systemPrompt = `你是一个工业自动化系统助手。
标准指令列表：[${availableKeys}]

任务规则：
1. 识别用户意图并按以下 XML 格式输出：
   - 基础格式：<cmd>标准指令</cmd>
   - 带参数格式（如设备号、序列号）：<cmd>标准指令</cmd><arg>参数内容</arg>
2. 示例：
   - “帮我添加一台编号为20260114的设备” -> 输出 “<cmd>添加设备</cmd><arg>20260114</arg>”
   - “打开监控中心” -> 输出 “<cmd>监控中心</cmd>”
3. “主界面”、“主页”映射为“首页”。
4. 只输出 XML 结果，不要输出任何解释说明或标点。
5. 无法匹配请输出 UNKNOWN。`;

    return new Promise((resolve) => {
        const handler = (event) => {
            const response = event.detail;
            window.removeEventListener("AI_RESULT", handler);
            
            if (response && response.success && response.data) {
                try {
                    if (response.data.choices && response.data.choices.length > 0) {
                        const content = response.data.choices[0].message.content.trim();
                        console.log("📥 AI 原始响应:", content);
                        resolve(content === "UNKNOWN" ? null : content);
                    } else {
                        resolve(null);
                    }
                } catch (e) {
                    console.error("解析响应失败:", e);
                    resolve(null);
                }
            } else {
                resolve(null);
            }
        };
        
        window.addEventListener("AI_RESULT", handler);
        window.dispatchEvent(new CustomEvent("DO_AI_REQUEST", {
            detail: { userInput, systemPrompt }
        }));
    });
}