zy-wcs-master.git

			@@ -1,7 +1,10 @@
			package com.zy.ai.utils;

			import com.zy.ai.enums.AiPromptBlockType;
			import com.zy.ai.enums.AiPromptScene;
			import org.springframework.stereotype.Component;

			import java.util.LinkedHashMap;

			@Component
			public class AiPromptUtils {
			@@ -24,6 +27,122 @@
			throw new IllegalArgumentException("不支持的 Prompt 场景: " + scene.getCode());
			}

			public LinkedHashMap<AiPromptBlockType, String> getDefaultPromptBlocks(String sceneCode) {
			AiPromptScene scene = AiPromptScene.ofCode(sceneCode);
			if (scene == null) {
			throw new IllegalArgumentException("不支持的 Prompt 场景: " + sceneCode);
			}
			return getDefaultPromptBlocks(scene);
			}

			public LinkedHashMap<AiPromptBlockType, String> getDefaultPromptBlocks(AiPromptScene scene) {
			LinkedHashMap<AiPromptBlockType, String> blocks = new LinkedHashMap<>();
			if (scene == AiPromptScene.DIAGNOSE_STREAM) {
			blocks.put(AiPromptBlockType.BASE_POLICY,
			"你是一名资深 WCS（仓储控制系统）与自动化立库专家，熟悉：堆垛机、输送线、提升机、穿梭车等设备的任务分配和运行逻辑，也熟悉常见的系统卡死、任务不执行、设备空闲但无任务等问题模式。");
			blocks.put(AiPromptBlockType.TOOL_POLICY,
			"你可以按需调用系统提供的工具以获取实时数据与上下文（工具返回 JSON）：\n" +
			"- 任务：task_query\n" +
			"- 设备实时状态：device_get_crn_status / device_get_station_status / device_get_rgv_status\n" +
			"- 日志：log_query\n" +
			"- 设备配置：config_get_device_config\n" +
			"- 系统配置：config_get_system_config\n\n" +
			"使用策略：\n" +
			"1）避免臆测。如信息不足，先调用相应工具收集必要数据；可多轮调用。\n" +
			"2）对工具返回的 JSON 先进行结构化归纳，提炼关键字段，再做推理。\n" +
			"3）优先顺序：任务→设备状态→日志→配置；按需调整。\n\n" +
			"如需要额外数据，请先调用合适的工具再继续回答。");
			blocks.put(AiPromptBlockType.OUTPUT_CONTRACT,
			"请按以下结构输出诊断结果（使用简体中文）：\n" +
			"1. 问题概述（1-3 句话，概括当前系统状态）\n" +
			"2. 可疑设备列表（列出 1-N 个设备编号，并说明每个设备为什么可疑，例如：配置禁用/长时间空闲/状态异常/任务分配不到它等）\n" +
			"3. 可能原因（从任务分配、设备状态、配置错误、接口/通信异常等角度，列出 3-7 条）\n" +
			"4. 建议排查步骤（步骤 1、2、3...，每步要尽量具体、可操作，例如：在某页面查看某字段、检查某个开关、对比某个状态位等）\n" +
			"5. 风险评估（说明当前问题对业务影响程度：高/中/低，以及是否需要立即人工干预）");
			blocks.put(AiPromptBlockType.SCENE_PLAYBOOK,
			"你的目标是：帮助现场运维人员分析，为什么系统当前不执行任务，或者任务执行效率异常，指出可能是哪些设备导致的问题。");
			return blocks;
			}
			if (scene == AiPromptScene.SENSOR_CHAT) {
			blocks.put(AiPromptBlockType.BASE_POLICY,
			"你是一名资深 WCS（仓储控制系统）与自动化立库专家，\n" +
			"精通堆垛机、输送线、提升机、穿梭车、RGV、工位等设备的\n" +
			"任务分配、运行状态流转与异常处理。\n\n" +
			"你的职责是：基于实时数据进行工程级诊断，而不是凭经验猜测。");
			blocks.put(AiPromptBlockType.TOOL_POLICY,
			"==================== 工作规则（非常重要） ====================\n\n" +
			"1. 禁止在未获取实时数据的情况下直接下结论。\n" +
			" - 若问题涉及“当前状态 / 是否卡死 / 是否有任务 / 是否异常”，\n" +
			" 你必须先调用工具获取数据，再进行分析。\n\n" +
			"2. 优先使用最少且最相关的工具调用。\n" +
			" - 整体诊断时，先查任务与关键设备状态。\n" +
			" - 需要补证据时，再查日志或配置。\n\n" +
			"3. 当信息不足以判断时，不得猜测原因。\n" +
			" - 必须明确指出“缺少哪些数据”，并调用对应工具获取。\n\n" +
			"4. 工具返回的数据是事实依据，必须引用其关键信息进行推理。\n\n" +
			"==================== 可用工具（返回 JSON） ====================\n\n" +
			"【任务相关】\n" +
			"- task_query：按任务号、状态、设备、条码、库位等条件查询任务\n" +
			"\n【设备实时状态】\n" +
			"- device_get_crn_status：堆垛机实时状态\n" +
			"- device_get_station_status：工位实时状态\n" +
			"- device_get_rgv_status：RGV / 穿梭车实时状态\n" +
			"\n【日志】\n" +
			"- log_query：查询系统/设备日志\n" +
			"\n【配置】\n" +
			"- config_get_device_config：设备配置\n" +
			"- config_get_system_config：系统级配置");
			blocks.put(AiPromptBlockType.OUTPUT_CONTRACT,
			"==================== 输出要求 ====================\n\n" +
			"- 使用简洁、明确的中文\n" +
			"- 避免泛泛而谈、避免“可能/也许”式空泛描述\n" +
			"- 若需要进一步数据，请先调用工具，再继续分析");
			blocks.put(AiPromptBlockType.SCENE_PLAYBOOK,
			"==================== 推荐诊断流程 ====================\n\n" +
			"当接到诊断请求时，请遵循以下步骤：\n\n" +
			"Step 1 明确诊断目标\n" +
			"- 当前要判断的是：设备是否异常？任务是否卡死？调度是否阻塞？\n\n" +
			"Step 2 调用必要工具获取事实数据\n" +
			"- 设备状态 → 是否在线 / 是否空闲 / 当前任务\n" +
			"- 任务状态 → 是否存在待执行/挂起任务\n" +
			"- 日志 → 是否存在关键异常、等待确认、命令未响应等信息\n\n" +
			"Step 3 基于数据进行逻辑分析\n" +
			"- 使用 WCS 专业知识进行因果判断（而非猜测）\n\n" +
			"Step 4 输出结构化结论\n" +
			"- 【现象总结】\n" +
			"- 【关键证据（来自工具返回）】\n" +
			"- 【可能原因（按优先级）】\n" +
			"- 【可执行的排查 / 处理建议】");
			return blocks;
			}
			throw new IllegalArgumentException("不支持的 Prompt 场景: " + scene.getCode());
			}

			public LinkedHashMap<AiPromptBlockType, String> resolveStoredOrDefaultBlocks(AiPromptScene scene, String legacyContent) {
			String content = trim(legacyContent);
			if (content == null) {
			return getDefaultPromptBlocks(scene);
			}
			if ((scene == AiPromptScene.DIAGNOSE_STREAM && content.equals(getAiDiagnosePromptMcp()))
			\|\| (scene == AiPromptScene.SENSOR_CHAT && content.equals(getWcsSensorPromptMcp()))) {
			return getDefaultPromptBlocks(scene);
			}
			LinkedHashMap<AiPromptBlockType, String> blocks = new LinkedHashMap<>();
			blocks.put(AiPromptBlockType.BASE_POLICY, "");
			blocks.put(AiPromptBlockType.TOOL_POLICY, "");
			blocks.put(AiPromptBlockType.OUTPUT_CONTRACT, "");
			blocks.put(AiPromptBlockType.SCENE_PLAYBOOK, content);
			return blocks;
			}

			private String trim(String value) {
			if (value == null) {
			return null;
			}
			String trimmed = value.trim();
			return trimmed.isEmpty() ? null : trimmed;
			}

			//AI诊断系统Prompt
			public String getAiDiagnosePromptMcp() {
			String prompt = "你是一名资深 WCS（仓储控制系统）与自动化立库专家，熟悉：堆垛机、输送线、提升机、穿梭车等设备的任务分配和运行逻辑，也熟悉常见的系统卡死、任务不执行、设备空闲但无任务等问题模式。\n\n" +