From 59b45c1c24ad7ca63902813b86705d7c460fa505 Mon Sep 17 00:00:00 2001
From: Junjie <fallin.jie@qq.com>
Date: 星期一, 16 三月 2026 08:51:34 +0800
Subject: [PATCH] #
---
src/main/java/com/zy/ai/controller/WcsDiagnosisController.java | 12 ++++++------
1 files changed, 6 insertions(+), 6 deletions(-)
diff --git a/src/main/java/com/zy/ai/controller/WcsDiagnosisController.java b/src/main/java/com/zy/ai/controller/WcsDiagnosisController.java
index 15857a0..fde25a1 100644
--- a/src/main/java/com/zy/ai/controller/WcsDiagnosisController.java
+++ b/src/main/java/com/zy/ai/controller/WcsDiagnosisController.java
@@ -25,15 +25,16 @@
private AiUtils aiUtils;
@GetMapping("/runAiStream")
- public SseEmitter runAiStream() {
+ public SseEmitter runAiStream(@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 = aiUtils.makeAiRequest(1000, "瀵瑰綋鍓嶇郴缁熻繘琛屽贰妫�锛屽鏋滄湁寮傚父鎯呭喌灏辫繘琛岃缁嗙殑鍒嗘瀽锛屽鏋滄病鏈夊紓甯告儏鍐靛垯褰撴垚涓�娆℃鏌n\n");
- wcsDiagnosisService.diagnoseStream(request, emitter);
+ wcsDiagnosisService.diagnoseStream(request, chatId, reset, emitter);
} catch (Exception e) {
- emitter.completeWithError(e);
+ try { emitter.complete(); } catch (Exception ignore) {}
}
}).start();
@@ -47,10 +48,9 @@
SseEmitter emitter = new SseEmitter(0L);
new Thread(() -> {
try {
- WcsDiagnosisRequest request = aiUtils.makeAiRequest(100, null);
- wcsDiagnosisService.askStream(request, prompt, chatId, reset, emitter);
+ wcsDiagnosisService.askStream(prompt, chatId, reset, emitter);
} catch (Exception e) {
- emitter.completeWithError(e);
+ try { emitter.complete(); } catch (Exception ignore) {}
}
}).start();
return emitter;
--
Gitblit v1.9.1