浏览代码

fix: 修复ollama调用失败

ageer 2 月之前
父节点
当前提交
15fd124da3
共有 2 个文件被更改,包括 2 次插入2 次删除
  1. 二进制
      image/12.png
  2. 2 2
      ruoyi-modules/ruoyi-system/src/main/java/org/ruoyi/system/service/impl/SseServiceImpl.java

二进制
image/12.png


+ 2 - 2
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<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())