From 88fa84a5be51d5e586234a84f64723c4bb51b08f Mon Sep 17 00:00:00 2001
From: Junjie <DELL@qq.com>
Date: 星期五, 12 十二月 2025 14:16:30 +0800
Subject: [PATCH] #AI

---
 src/main/java/com/zy/ai/controller/WcsDiagnosisController.java |   90 ++++++++++++++++++++++++++++++++++++++++++++
 1 files changed, 89 insertions(+), 1 deletions(-)

diff --git a/src/main/java/com/zy/ai/controller/WcsDiagnosisController.java b/src/main/java/com/zy/ai/controller/WcsDiagnosisController.java
index 45c90e3..07eff9f 100644
--- a/src/main/java/com/zy/ai/controller/WcsDiagnosisController.java
+++ b/src/main/java/com/zy/ai/controller/WcsDiagnosisController.java
@@ -3,10 +3,13 @@
 import com.baomidou.mybatisplus.mapper.EntityWrapper;
 import com.zy.ai.entity.DeviceConfigsData;
 import com.zy.ai.entity.DeviceRealTimeData;
+import com.zy.ai.entity.ChatCompletionRequest;
 import com.zy.ai.entity.WcsDiagnosisRequest;
 import com.zy.ai.entity.WcsDiagnosisResponse;
 import com.zy.ai.log.AiLogAppender;
 import com.zy.ai.service.WcsDiagnosisService;
+import com.core.annotations.ManagerAuth;
+import com.zy.common.web.BaseController;
 import com.zy.asrs.entity.BasCrnp;
 import com.zy.asrs.entity.WrkMast;
 import com.zy.asrs.service.BasCrnpService;
@@ -33,7 +36,7 @@
 @RestController
 @RequestMapping("/ai/diagnose")
 @RequiredArgsConstructor
-public class WcsDiagnosisController {
+public class WcsDiagnosisController extends BaseController {
 
     @Autowired
     private WcsDiagnosisService wcsDiagnosisService;
@@ -177,6 +180,91 @@
         return emitter;
     }
 
+    @GetMapping("/askStream")
+    public SseEmitter askStream(@RequestParam("prompt") String prompt,
+                                @RequestParam(value = "chatId", required = false) String chatId,
+                                @RequestParam(value = "reset", required = false, defaultValue = "false") boolean reset) {
+        SseEmitter emitter = new SseEmitter(0L);
+        new Thread(() -> {
+            try {
+                WcsDiagnosisRequest request = new WcsDiagnosisRequest();
+
+                List<String> logs = AiLogAppender.getRecentLogs(100);
+                request.setLogs(logs);
+
+                List<WrkMast> wrkMasts = wrkMastService.selectList(new EntityWrapper<>());
+                request.setTasks(wrkMasts);
+
+                List<DeviceRealTimeData> deviceRealTimeDataList = new ArrayList<>();
+                List<DeviceConfigsData> deviceConfigsDataList = new ArrayList<>();
+
+                List<BasCrnp> basCrnps = basCrnpService.selectList(new EntityWrapper<>());
+                for (BasCrnp basCrnp : basCrnps) {
+                    CrnThread crnThread = (CrnThread) SlaveConnection.get(SlaveType.Crn, basCrnp.getCrnNo());
+                    if (crnThread == null) {
+                        continue;
+                    }
+
+                    CrnProtocol protocol = crnThread.getStatus();
+
+                    for (StationObjModel stationObjModel : basCrnp.getInStationList$()) {
+                        StationThread stationThread = (StationThread) SlaveConnection.get(SlaveType.Devp, stationObjModel.getDeviceNo());
+                        if (stationThread == null) {
+                            continue;
+                        }
+
+                        Map<Integer, StationProtocol> map = stationThread.getStatusMap();
+                        StationProtocol stationProtocol = map.get(stationObjModel.getStationId());
+                        if (stationProtocol == null) {
+                            continue;
+                        }
+
+                        DeviceRealTimeData stationData = new DeviceRealTimeData();
+                        stationData.setDeviceNo(stationObjModel.getDeviceNo());
+                        stationData.setDeviceType(String.valueOf(SlaveType.Devp));
+                        stationData.setDeviceData(stationProtocol);
+                        deviceRealTimeDataList.add(stationData);
+                    }
+
+                    DeviceRealTimeData deviceRealTimeData = new DeviceRealTimeData();
+                    deviceRealTimeData.setDeviceNo(basCrnp.getCrnNo());
+                    deviceRealTimeData.setDeviceType(String.valueOf(SlaveType.Crn));
+                    deviceRealTimeData.setDeviceData(protocol);
+                    deviceRealTimeDataList.add(deviceRealTimeData);
+
+                    DeviceConfigsData deviceConfigsData = new DeviceConfigsData();
+                    deviceConfigsData.setDeviceNo(basCrnp.getCrnNo());
+                    deviceConfigsData.setDeviceType(String.valueOf(SlaveType.Crn));
+                    deviceConfigsData.setDeviceData(basCrnp);
+                    deviceConfigsDataList.add(deviceConfigsData);
+                }
+
+                request.setDeviceRealtimeData(deviceRealTimeDataList);
+                request.setDeviceConfigs(deviceConfigsDataList);
+
+                wcsDiagnosisService.askStream(request, prompt, chatId, reset, emitter);
+            } catch (Exception e) {
+                emitter.completeWithError(e);
+            }
+        }).start();
+        return emitter;
+    }
+
+    @GetMapping("/chats")
+    public List<Map<String, Object>> listChats() {
+        return wcsDiagnosisService.listChats();
+    }
+
+    @DeleteMapping("/chats/{chatId}")
+    public Boolean deleteChat(@PathVariable("chatId") String chatId) {
+        return wcsDiagnosisService.deleteChat(chatId);
+    }
+
+    @GetMapping("/chats/{chatId}/history")
+    public List<ChatCompletionRequest.Message> getChatHistory(@PathVariable("chatId") String chatId) {
+        return wcsDiagnosisService.getChatHistory(chatId);
+    }
+
     /**
      * POST /api/ai/diagnose/wcs
      */

--
Gitblit v1.9.1