Package com.bytedesk.ai.service

Class BaseSpringAIService

java.lang.Object

com.bytedesk.ai.service.BaseSpringAIService

All Implemented Interfaces:

SpringAIService

Direct Known Subclasses:

SpringAIBaiduChatService, SpringAIBaiduService, SpringAICozeService, SpringAICustomChatService, SpringAICustomService, SpringAIDashscopeChatService, SpringAIDashscopeService, SpringAIDeepseekChatService, SpringAIDeepseekService, SpringAIDifyService, SpringAIFastGptService, SpringAIGeminiChatService, SpringAIGeminiService, SpringAIGiteeChatService, SpringAIGiteeService, SpringAIMaxkbService, SpringAIMinimaxChatService, SpringAIMinimaxService, SpringAIN8nService, SpringAIOllamaChatService, SpringAIOllamaService, SpringAIOpenaiChatService, SpringAIOpenaiService, SpringAIOpenrouterChatService, SpringAIOpenrouterService, SpringAIRagflowService, SpringAISiliconFlowChatService, SpringAISiliconFlowService, SpringAITencentChatService, SpringAITencentService, SpringAIVolcengineChatService, SpringAIVolcengineService, SpringAIWeknoraService, SpringAIZhipuaiChatService, ZhipuaiMultiModelService

public abstract class BaseSpringAIService
extends Object
implements SpringAIService

Field Summary

Fields

Modifier and Type	Field	Description
protected org.springframework.context.ApplicationEventPublisher	applicationEventPublisher
protected ChunkElasticService	chunkElasticService
protected ChunkVectorService	chunkVectorService
protected FaqElasticService	faqElasticService
protected FaqVectorService	faqVectorService
protected KnowledgeBaseSearchHelper	knowledgeBaseSearchHelper
protected MessagePersistCache	messagePersistCache
protected MessagePersistenceHelper	messagePersistenceHelper
protected MessageRestService	messageRestService
protected IMessageSendService	messageSendService
protected PromptHelper	promptHelper
protected RobotMessageCache	robotMessageCache
protected RobotRestService	robotRestService
protected SseMessageHelper	sseMessageHelper
protected TextElasticService	textElasticService
protected TextVectorService	textVectorService
protected ThreadRestService	threadRestService
protected UidUtils	uidUtils
protected WebpageElasticService	webpageElasticService
protected WebpageVectorService	webpageVectorService

Constructor Summary

Constructors

Modifier	Constructor	Description
protected	BaseSpringAIService()

Method Summary

Modifier and Type	Method	Description
protected abstract void	processPromptSse(org.springframework.ai.chat.prompt.Prompt prompt, RobotProtobuf robot, MessageProtobuf messageProtobufQuery, MessageProtobuf messageProtobufReply, List<RobotContent.SourceReference> sourceReferences, org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter)
protected abstract String	processPromptSync(String message, RobotProtobuf robot)
protected abstract void	processPromptWebsocket(org.springframework.ai.chat.prompt.Prompt prompt, RobotProtobuf robot, MessageProtobuf messageProtobufQuery, MessageProtobuf messageProtobufReply)
String	processSyncRequest(String query, RobotProtobuf robot, boolean searchKnowledgeBase)	直接处理LLM请求并同步返回结果，支持控制是否查询知识库 主要用于RobotAgentService中的processLlmRequest方法
void	sendSseMessage(String query, RobotProtobuf robot, MessageProtobuf messageProtobufQuery, MessageProtobuf messageProtobufReply, org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter)	发送Sse消息
String	sendSyncMessage(String query, RobotProtobuf robot, MessageProtobuf messageProtobufQuery, MessageProtobuf messageProtobufReply)	发送同步消息并返回回复内容
void	sendWebsocketMessage(String query, RobotProtobuf robot, MessageProtobuf messageProtobufQuery, MessageProtobuf messageProtobufReply)	发送websocket消息

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Details

faqElasticService

@Autowired
protected FaqElasticService faqElasticService

faqVectorService

@Autowired(required=false)
protected FaqVectorService faqVectorService

textElasticService

@Autowired
protected TextElasticService textElasticService

textVectorService

@Autowired(required=false)
protected TextVectorService textVectorService

chunkElasticService

@Autowired
protected ChunkElasticService chunkElasticService

chunkVectorService

@Autowired(required=false)
protected ChunkVectorService chunkVectorService

webpageElasticService

@Autowired
protected WebpageElasticService webpageElasticService

webpageVectorService

@Autowired(required=false)
protected WebpageVectorService webpageVectorService

messageSendService

@Autowired
protected IMessageSendService messageSendService

uidUtils

@Autowired
protected UidUtils uidUtils

robotRestService

@Autowired
protected RobotRestService robotRestService

threadRestService

@Autowired
protected ThreadRestService threadRestService

messagePersistCache

@Autowired
protected MessagePersistCache messagePersistCache

robotMessageCache

@Autowired
protected RobotMessageCache robotMessageCache

messageRestService

@Autowired
protected MessageRestService messageRestService

applicationEventPublisher

@Autowired
protected org.springframework.context.ApplicationEventPublisher applicationEventPublisher

knowledgeBaseSearchHelper

@Autowired
protected KnowledgeBaseSearchHelper knowledgeBaseSearchHelper

promptHelper

@Autowired
protected PromptHelper promptHelper

messagePersistenceHelper

@Autowired
protected MessagePersistenceHelper messagePersistenceHelper

sseMessageHelper

@Autowired
protected SseMessageHelper sseMessageHelper

Constructor Details

BaseSpringAIService

protected BaseSpringAIService()

Method Details

sendWebsocketMessage

public void sendWebsocketMessage(String query,
 RobotProtobuf robot,
 MessageProtobuf messageProtobufQuery,
 MessageProtobuf messageProtobufReply)

Description copied from interface: SpringAIService

发送websocket消息

Specified by:

sendWebsocketMessage in interface SpringAIService

Parameters:

query - 用户查询

robot - 机器人实体

sendSseMessage

public void sendSseMessage(String query,
 RobotProtobuf robot,
 MessageProtobuf messageProtobufQuery,
 MessageProtobuf messageProtobufReply,
 org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter)

Description copied from interface: SpringAIService

发送Sse消息

Specified by:

sendSseMessage in interface SpringAIService

emitter - SseEmitter

sendSyncMessage

public String sendSyncMessage(String query,
 RobotProtobuf robot,
 MessageProtobuf messageProtobufQuery,
 MessageProtobuf messageProtobufReply)

Description copied from interface: SpringAIService

发送同步消息并返回回复内容

Specified by:

sendSyncMessage in interface SpringAIService

Parameters:

query - 用户查询

robot - 机器人实体

messageProtobufQuery - 查询消息

messageProtobufReply - 回复消息

Returns:

回复内容

processSyncRequest

public String processSyncRequest(String query,
 RobotProtobuf robot,
 boolean searchKnowledgeBase)

Description copied from interface: SpringAIService

直接处理LLM请求并同步返回结果，支持控制是否查询知识库 主要用于RobotAgentService中的processLlmRequest方法

Specified by:

processSyncRequest in interface SpringAIService

Parameters:

query - 用户查询

robot - 机器人配置

searchKnowledgeBase - 是否需要查询知识库

Returns:

大模型生成的回复内容

processPromptWebsocket

protected abstract void processPromptWebsocket(org.springframework.ai.chat.prompt.Prompt prompt,
 RobotProtobuf robot,
 MessageProtobuf messageProtobufQuery,
 MessageProtobuf messageProtobufReply)

processPromptSse

protected abstract void processPromptSse(org.springframework.ai.chat.prompt.Prompt prompt,
 RobotProtobuf robot,
 MessageProtobuf messageProtobufQuery,
 MessageProtobuf messageProtobufReply,
 List<RobotContent.SourceReference> sourceReferences,
 org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter)

processPromptSync

protected abstract String processPromptSync(String message,
 RobotProtobuf robot)