xieyinlu/ruoyi-ai
1
0
Fork 0
mirror of https://gitcode.com/ageerle/ruoyi-ai.git synced 2026-04-17 13:53:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: 代码优化

Browse Source
This commit is contained in:
ageer
2025-05-25 14:42:22 +08:00
parent 1bff2791a6
commit 5d972d66f4
3 changed files with 2 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/StreamAssistant.java
View File

@@ -1,7 +0,0 @@
package org.ruoyi.chat.service.chat;
import dev.langchain4j.service.TokenStream;
public interface StreamAssistant {
TokenStream chat(String message);
}

10
ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/QianWenAiChatServiceImpl.java
View File

@@ -1,7 +1,6 @@
package org.ruoyi.chat.service.chat.impl; package org.ruoyi.chat.service.chat.impl;
import dev.langchain4j.community.model.dashscope.QwenStreamingChatModel; import dev.langchain4j.community.model.dashscope.QwenStreamingChatModel;
import dev.langchain4j.community.model.zhipu.ZhipuAiStreamingChatModel;
import dev.langchain4j.model.chat.StreamingChatModel; import dev.langchain4j.model.chat.StreamingChatModel;
import dev.langchain4j.model.chat.response.ChatResponse; import dev.langchain4j.model.chat.response.ChatResponse;
import dev.langchain4j.model.chat.response.StreamingChatResponseHandler; import dev.langchain4j.model.chat.response.StreamingChatResponseHandler;
@@ -10,21 +9,12 @@ import lombok.extern.slf4j.Slf4j;
import org.ruoyi.chat.enums.ChatModeType; import org.ruoyi.chat.enums.ChatModeType;
import org.ruoyi.chat.service.chat.IChatService; import org.ruoyi.chat.service.chat.IChatService;
import org.ruoyi.common.chat.request.ChatRequest; import org.ruoyi.common.chat.request.ChatRequest;
import org.ruoyi.domain.ChatMessage;
import org.ruoyi.domain.vo.ChatModelVo; import org.ruoyi.domain.vo.ChatModelVo;
import org.ruoyi.service.IChatModelService; import org.ruoyi.service.IChatModelService;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter; import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
import java.util.List;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.TimeoutException;
import static dev.langchain4j.data.message.SystemMessage.systemMessage;
import static dev.langchain4j.data.message.UserMessage.userMessage;
import static java.util.Arrays.asList;
/** /**
* 阿里通义千问 * 阿里通义千问

6
ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/chat/impl/ZhipuAiChatServiceImpl.java
View File

@@ -16,8 +16,6 @@ import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter; import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;
import java.util.concurrent.ExecutionException;
import java.util.concurrent.TimeoutException;
/** /**
@@ -46,14 +44,14 @@ public class ZhipuAiChatServiceImpl implements IChatService {
@SneakyThrows @SneakyThrows
@Override @Override
public void onPartialResponse(String token) { public void onPartialResponse(String token) {
System.out.println(token); //System.out.println(token);
emitter.send(token); emitter.send(token);
} }
@SneakyThrows @SneakyThrows
@Override @Override
public void onError(Throwable error) { public void onError(Throwable error) {
System.out.println(error.getMessage()); // System.out.println(error.getMessage());
emitter.send(error.getMessage()); emitter.send(error.getMessage());
} }