From 15fd124da390fba0da2caba0ec9162993204eba3 Mon Sep 17 00:00:00 2001
From: ageer <ageerle@163.com>
Date: 星期三, 12 三月 2025 23:01:39 +0800
Subject: [PATCH] fix: 修复ollama调用失败
---
ruoyi-modules/ruoyi-system/src/main/java/org/ruoyi/system/service/impl/SseServiceImpl.java | 4 ++--
image/12.png | 0
2 files changed, 2 insertions(+), 2 deletions(-)
diff --git a/image/12.png b/image/12.png
index 58b5697..68c016b 100644
--- a/image/12.png
+++ b/image/12.png
Binary files 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 c822e3d..634cb02 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 @@
@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<Message> 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())
--
Gitblit v1.9.3