From aeb124afcef69c8e43230bc0b31cee0616a5d9c9 Mon Sep 17 00:00:00 2001
From: Junjie <fallin.jie@qq.com>
Date: 星期四, 12 三月 2026 13:53:47 +0800
Subject: [PATCH] #

---
 src/main/java/com/zy/ai/utils/AiPromptUtils.java |   19 +++++++++++++++++++
 1 files changed, 19 insertions(+), 0 deletions(-)

diff --git a/src/main/java/com/zy/ai/utils/AiPromptUtils.java b/src/main/java/com/zy/ai/utils/AiPromptUtils.java
index 71b4a3c..3fed8c0 100644
--- a/src/main/java/com/zy/ai/utils/AiPromptUtils.java
+++ b/src/main/java/com/zy/ai/utils/AiPromptUtils.java
@@ -1,10 +1,29 @@
 package com.zy.ai.utils;
 
+import com.zy.ai.enums.AiPromptScene;
 import org.springframework.stereotype.Component;
 
 @Component
 public class AiPromptUtils {
 
+    public String getDefaultPrompt(String sceneCode) {
+        AiPromptScene scene = AiPromptScene.ofCode(sceneCode);
+        if (scene == null) {
+            throw new IllegalArgumentException("涓嶆敮鎸佺殑 Prompt 鍦烘櫙: " + sceneCode);
+        }
+        return getDefaultPrompt(scene);
+    }
+
+    public String getDefaultPrompt(AiPromptScene scene) {
+        if (scene == AiPromptScene.DIAGNOSE_STREAM) {
+            return getAiDiagnosePromptMcp();
+        }
+        if (scene == AiPromptScene.SENSOR_CHAT) {
+            return getWcsSensorPromptMcp();
+        }
+        throw new IllegalArgumentException("涓嶆敮鎸佺殑 Prompt 鍦烘櫙: " + scene.getCode());
+    }
+
     //AI璇婃柇绯荤粺Prompt
     public String getAiDiagnosePromptMcp() {
         String prompt = "浣犳槸涓�鍚嶈祫娣� WCS锛堜粨鍌ㄦ帶鍒剁郴缁燂級涓庤嚜鍔ㄥ寲绔嬪簱涓撳锛岀啛鎮夛細鍫嗗灈鏈恒�佽緭閫佺嚎銆佹彁鍗囨満銆佺┛姊溅绛夎澶囩殑浠诲姟鍒嗛厤鍜岃繍琛岄�昏緫锛屼篃鐔熸倝甯歌鐨勭郴缁熷崱姝汇�佷换鍔′笉鎵ц銆佽澶囩┖闂蹭絾鏃犱换鍔$瓑闂妯″紡銆俓n\n" +

--
Gitblit v1.9.1