diff --git a/image/12.png b/image/12.png index 58b56977..68c016b9 100644 Binary files a/image/12.png and b/image/12.png differ diff --git a/ruoyi-modules/ruoyi-system/src/main/java/org/ruoyi/system/service/impl/SseServiceImpl.java b/ruoyi-modules/ruoyi-system/src/main/java/org/ruoyi/system/service/impl/SseServiceImpl.java index c822e3d2..634cb025 100644 --- a/ruoyi-modules/ruoyi-system/src/main/java/org/ruoyi/system/service/impl/SseServiceImpl.java +++ b/ruoyi-modules/ruoyi-system/src/main/java/org/ruoyi/system/service/impl/SseServiceImpl.java @@ -609,14 +609,14 @@ public class SseServiceImpl implements ISseService { @Override public SseEmitter ollamaChat(ChatRequest chatRequest) { String[] parts = chatRequest.getModel().split("ollama-"); - SysModel sysModel = sysModelService.selectModelByName(parts[1]); + SysModel sysModel = sysModelService.selectModelByName(chatRequest.getModel()); final SseEmitter emitter = new SseEmitter(); String host = sysModel.getApiHost(); List msgList = chatRequest.getMessages(); Message message = msgList.get(msgList.size() - 1); OllamaAPI api = new OllamaAPI(host); api.setRequestTimeoutSeconds(100); - OllamaChatRequestBuilder builder = OllamaChatRequestBuilder.getInstance(sysModel.getModelName()); + OllamaChatRequestBuilder builder = OllamaChatRequestBuilder.getInstance(parts[1]); OllamaChatRequestModel requestModel = builder .withMessage(OllamaChatMessageRole.USER, message.getContent().toString())