xieyinlu/ruoyi-ai
1
0
Fork 0
mirror of https://gitcode.com/ageerle/ruoyi-ai.git synced 2026-04-16 21:33:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: 修复ollama调用失败

Browse Source
This commit is contained in:
ageer
2025-03-12 23:01:39 +08:00
parent d8fda15593
commit 15fd124da3
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

BIN
image/12.png
View File

Binary file not shown.
Side by Side Swipe Overlay

Before

Width:  |  Height:  |  Size: 81 KiB

After

Width:  |  Height:  |  Size: 85 KiB

4
ruoyi-modules/ruoyi-system/src/main/java/org/ruoyi/system/service/impl/SseServiceImpl.java
View File

@@ -609,14 +609,14 @@ public class SseServiceImpl implements ISseService {
@Override @Override
public SseEmitter ollamaChat(ChatRequest chatRequest) { public SseEmitter ollamaChat(ChatRequest chatRequest) {
String[] parts = chatRequest.getModel().split("ollama-"); String[] parts = chatRequest.getModel().split("ollama-");
SysModel sysModel = sysModelService.selectModelByName(parts[1]); SysModel sysModel = sysModelService.selectModelByName(chatRequest.getModel());
final SseEmitter emitter = new SseEmitter(); final SseEmitter emitter = new SseEmitter();
String host = sysModel.getApiHost(); String host = sysModel.getApiHost();
List<Message> msgList = chatRequest.getMessages(); List<Message> msgList = chatRequest.getMessages();
Message message = msgList.get(msgList.size() - 1); Message message = msgList.get(msgList.size() - 1);
OllamaAPI api = new OllamaAPI(host); OllamaAPI api = new OllamaAPI(host);
api.setRequestTimeoutSeconds(100); api.setRequestTimeoutSeconds(100);
OllamaChatRequestBuilder builder = OllamaChatRequestBuilder.getInstance(sysModel.getModelName()); OllamaChatRequestBuilder builder = OllamaChatRequestBuilder.getInstance(parts[1]);
OllamaChatRequestModel requestModel = builder OllamaChatRequestModel requestModel = builder
.withMessage(OllamaChatMessageRole.USER, .withMessage(OllamaChatMessageRole.USER,
message.getContent().toString()) message.getContent().toString())