From 825813e2dd90cf8bdc48acbb6eee85159bc33b4d Mon Sep 17 00:00:00 2001
From: Junjie <fallin.jie@qq.com>
Date: 星期二, 03 三月 2026 13:04:28 +0800
Subject: [PATCH] #AI LLM路由
---
src/main/java/com/zy/ai/service/WcsDiagnosisService.java | 19 ++++++-------------
1 files changed, 6 insertions(+), 13 deletions(-)
diff --git a/src/main/java/com/zy/ai/service/WcsDiagnosisService.java b/src/main/java/com/zy/ai/service/WcsDiagnosisService.java
index 189e57d..45dea58 100644
--- a/src/main/java/com/zy/ai/service/WcsDiagnosisService.java
+++ b/src/main/java/com/zy/ai/service/WcsDiagnosisService.java
@@ -11,7 +11,6 @@
import com.zy.common.utils.RedisUtil;
import com.zy.core.enums.RedisKeyType;
import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
@@ -28,8 +27,6 @@
private static final long CHAT_TTL_SECONDS = 7L * 24 * 3600;
- @Value("${llm.platform}")
- private String platform;
@Autowired
private LlmChatService llmChatService;
@Autowired
@@ -40,8 +37,6 @@
private AiUtils aiUtils;
@Autowired(required = false)
private McpController mcpController;
- @Autowired
- private PythonService pythonService;
public void diagnoseStream(WcsDiagnosisRequest request, SseEmitter emitter) {
List<ChatCompletionRequest.Message> messages = new ArrayList<>();
@@ -85,7 +80,7 @@
try {
try { emitter.send(SseEmitter.event().data("銆怉I銆戣繍琛屽凡鍋滄锛堝紓甯革級")); } catch (Exception ignore) {}
log.error("AI diagnose stream stopped: error", e);
- emitter.completeWithError(e);
+ emitter.complete();
} catch (Exception ignore) {}
});
}
@@ -95,11 +90,6 @@
String chatId,
boolean reset,
SseEmitter emitter) {
- if (platform.equals("python")) {
- pythonService.runPython(prompt, chatId, emitter);
- return;
- }
-
List<ChatCompletionRequest.Message> messages = new ArrayList<>();
List<ChatCompletionRequest.Message> history = null;
@@ -187,7 +177,10 @@
emitter.complete();
} catch (Exception ignore) {}
}, e -> {
- try { emitter.completeWithError(e); } catch (Exception ignore) {}
+ try {
+ try { emitter.send(SseEmitter.event().data("銆怉I銆戣繍琛屽凡鍋滄锛堝紓甯革級")); } catch (Exception ignore) {}
+ emitter.complete();
+ } catch (Exception ignore) {}
});
}
@@ -380,7 +373,7 @@
try {
sse(emitter, "\\n\\n銆怉I銆戣繍琛屽凡鍋滄锛堝紓甯革級\\n\\n");
log.error("AI MCP diagnose stopped: error", e);
- emitter.completeWithError(e);
+ emitter.complete();
} catch (Exception ignore) {}
return true;
}
--
Gitblit v1.9.1