From 995507e7574a68afd598f8ac610d1bc810a0210e Mon Sep 17 00:00:00 2001
From: stageluo <979175267@qq.com>
Date: Tue, 25 Nov 2025 09:28:14 +0800
Subject: [PATCH] =?UTF-8?q?=E6=94=AF=E6=8C=81=E6=9C=AC=E5=9C=B0ollama?=
 =?UTF-8?q?=E9=9B=86=E6=88=90LLM=E9=97=AE=E7=AD=94=E5=88=86=E6=94=AF?=
 =?UTF-8?q?=E5=B7=A5=E4=BD=9C=E6=B5=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../service/chat/impl/OllamaServiceImpl.java  | 32 +++++++++++++++++--
 1 file changed, 29 insertions(+), 3 deletions(-)

diff --git a/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/OllamaServiceImpl.java b/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/OllamaServiceImpl.java
index 7ce42215..2956e555 100644
--- a/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/OllamaServiceImpl.java
+++ b/ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/OllamaServiceImpl.java
@@ -1,5 +1,8 @@
 package org.ruoyi.chat.service.chat.impl;
 
+import dev.langchain4j.model.chat.StreamingChatModel;
+import dev.langchain4j.model.chat.response.StreamingChatResponseHandler;
+import dev.langchain4j.model.ollama.OllamaStreamingChatModel;
 import io.github.ollama4j.OllamaAPI;
 import io.github.ollama4j.models.chat.OllamaChatMessage;
 import io.github.ollama4j.models.chat.OllamaChatMessageRole;
@@ -9,7 +12,8 @@ import io.github.ollama4j.models.generate.OllamaStreamHandler;
 import lombok.extern.slf4j.Slf4j;
 import org.ruoyi.chat.enums.ChatModeType;
 import org.ruoyi.chat.service.chat.IChatService;
-import org.ruoyi.chat.util.SSEUtil;
+import org.ruoyi.chat.support.ChatServiceHelper;
+import org.ruoyi.chat.support.RetryNotifier;
 import org.ruoyi.common.chat.entity.chat.Message;
 import org.ruoyi.common.chat.request.ChatRequest;
 import org.ruoyi.domain.vo.ChatModelVo;
@@ -22,8 +26,6 @@ import java.io.IOException;
 import java.util.ArrayList;
 import java.util.List;
 import java.util.concurrent.CompletableFuture;
-import org.ruoyi.chat.support.RetryNotifier;
-import org.ruoyi.chat.support.ChatServiceHelper;
 
 
 /**
@@ -81,6 +83,30 @@ public class OllamaServiceImpl implements IChatService {
         return emitter;
     }
 
+    /**
+     * 工作流场景：支持 langchain4j handler
+     */
+    @Override
+    public void chat(ChatRequest request, StreamingChatResponseHandler handler) {
+        log.info("workflow chat, model: {}", request.getModel());
+
+        ChatModelVo chatModelVo = chatModelService.selectModelByName(request.getModel());
+
+        StreamingChatModel model = OllamaStreamingChatModel.builder()
+                .baseUrl(chatModelVo.getApiHost() != null ? chatModelVo.getApiHost() : "http://localhost:11434")
+                .modelName(chatModelVo.getModelName())
+                .build();
+
+        try {
+            // 将 ruoyi-ai 的 ChatRequest 转换为 langchain4j 的格式
+            dev.langchain4j.model.chat.request.ChatRequest chatRequest = convertToLangchainRequest(request);
+            model.chat(chatRequest, handler);
+        } catch (Exception e) {
+            log.error("workflow ollama请求失败：{}", e.getMessage(), e);
+            throw new RuntimeException("ollama workflow chat failed: " + e.getMessage(), e);
+        }
+    }
+
     @Override
     public String getCategory() {
         return ChatModeType.OLLAMA.getCode();