mirror of
https://github.com/ccmjga/zhilu-admin
synced 2026-04-08 14:37:38 +00:00
新增 ChatDto 数据传输对象,更新聊天接口以支持知识库功能,优化聊天服务逻辑,调整前端组件以提升用户体验。
This commit is contained in:
@@ -2,6 +2,7 @@ package com.zl.mjga.service;
|
||||
|
||||
import com.zl.mjga.config.ai.AiChatAssistant;
|
||||
import com.zl.mjga.config.ai.SystemToolAssistant;
|
||||
import com.zl.mjga.dto.ai.ChatDto;
|
||||
import com.zl.mjga.exception.BusinessException;
|
||||
import dev.langchain4j.service.TokenStream;
|
||||
import java.util.Optional;
|
||||
@@ -39,8 +40,20 @@ public class AiChatService {
|
||||
};
|
||||
}
|
||||
|
||||
public TokenStream chatPrecedenceLlmWith(String sessionIdentifier, String userMessage) {
|
||||
public TokenStream chat(String sessionIdentifier, ChatDto chatDto) {
|
||||
return switch (chatDto.mode()) {
|
||||
case NORMAL -> chatWithPrecedenceLlm(sessionIdentifier, chatDto);
|
||||
case WITH_LIBRARY -> chatWithLibrary(chatDto.libraryId(), chatDto);
|
||||
};
|
||||
}
|
||||
|
||||
public TokenStream chatWithLibrary(Long libraryId, ChatDto chatDto) {
|
||||
return zhiPuChatAssistant.chat(String.valueOf(libraryId), chatDto.message());
|
||||
}
|
||||
|
||||
public TokenStream chatWithPrecedenceLlm(String sessionIdentifier, ChatDto chatDto) {
|
||||
LlmCodeEnum code = getPrecedenceLlmCode();
|
||||
String userMessage = chatDto.message();
|
||||
return switch (code) {
|
||||
case ZHI_PU -> zhiPuChatAssistant.chat(sessionIdentifier, userMessage);
|
||||
case DEEP_SEEK -> deepSeekChatAssistant.chat(sessionIdentifier, userMessage);
|
||||
|
||||
Reference in New Issue
Block a user