Selaa lähdekoodia

fix: 修复ollama调用失败

ageer 2 kuukautta sitten
vanhempi
commit
15fd124da3

BIN
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())