Package com.bytedesk.ai.springai.base

Class BaseSpringAIService

java.lang.Object

com.bytedesk.ai.springai.base.BaseSpringAIService

All Implemented Interfaces:

SpringAIService

Direct Known Subclasses:

SpringAIDashscopeService, SpringAIDeepseekService, SpringAIGiteeService, SpringAIOllamaService, SpringAISiliconFlowService, SpringAIZhipuaiService

public abstract class BaseSpringAIService
extends Object
implements SpringAIService

Field Summary

Fields

Modifier and Type	Field	Description
protected final MessagePersistCache	messagePersistCache
protected final IMessageSendService	messageSendService
protected final RobotRestService	robotRestService
protected final Optional<SpringAIVectorService>	springAIVectorService
protected final ThreadRestService	threadRestService
protected final UidUtils	uidUtils

Constructor Summary

Constructors

Modifier	Constructor	Description
protected	BaseSpringAIService(Optional<SpringAIVectorService> springAIVectorService, IMessageSendService messageSendService, UidUtils uidUtils, RobotRestService robotRestService, ThreadRestService threadRestService, MessagePersistCache messagePersistCache)

Method Summary

Modifier and Type	Method	Description
String	buildKbPrompt(String systemPrompt, String query, String context)
protected abstract String	generateFaqPairs(String prompt)
String	generateFaqPairsAsync(String chunk)	异步生成FAQ对
void	generateFaqPairsSync(String chunk)	同步生成FAQ对
void	persistMessage(String messageJson)
protected abstract void	processPrompt(org.springframework.ai.chat.prompt.Prompt prompt, MessageProtobuf messageProtobuf)
protected abstract void	processPromptSSE(org.springframework.ai.chat.prompt.Prompt prompt, MessageProtobuf messageProtobuf, org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter)
protected abstract String	processPromptSync(String message)
void	sendSseMessage(String query, RobotProtobuf robot, MessageProtobuf messageProtobuf, org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter)	发送Sse消息
void	sendWebsocketMessage(String query, RobotEntity robot, MessageProtobuf messageProtobuf)	发送websocket消息

Methods inherited from class java.lang.Object

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

Field Details

springAIVectorService

protected final Optional<SpringAIVectorService> springAIVectorService

messageSendService

protected final IMessageSendService messageSendService

uidUtils

protected final UidUtils uidUtils

robotRestService

protected final RobotRestService robotRestService

threadRestService

protected final ThreadRestService threadRestService

messagePersistCache

protected final MessagePersistCache messagePersistCache

Constructor Details

BaseSpringAIService

protected BaseSpringAIService(Optional<SpringAIVectorService> springAIVectorService,
 IMessageSendService messageSendService,
 UidUtils uidUtils,
 RobotRestService robotRestService,
 ThreadRestService threadRestService,
 MessagePersistCache messagePersistCache)

Method Details

sendWebsocketMessage

public void sendWebsocketMessage(String query,
 RobotEntity robot,
 MessageProtobuf messageProtobuf)

Description copied from interface: SpringAIService

发送websocket消息

Specified by:

sendWebsocketMessage in interface SpringAIService

Parameters:

query - 用户查询

robot - 机器人实体

messageProtobuf - 消息协议

sendSseMessage

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

Description copied from interface: SpringAIService

发送Sse消息

Specified by:

sendSseMessage in interface SpringAIService

emitter - SseEmitter

generateFaqPairsAsync

public String generateFaqPairsAsync(String chunk)

Description copied from interface: SpringAIService

异步生成FAQ对

Specified by:

generateFaqPairsAsync in interface SpringAIService

Parameters:

chunk - 文本块

Returns:

生成的FAQ对

generateFaqPairsSync

public void generateFaqPairsSync(String chunk)

Description copied from interface: SpringAIService

同步生成FAQ对

Specified by:

generateFaqPairsSync in interface SpringAIService

Parameters:

chunk - 文本块

persistMessage

public void persistMessage(String messageJson)

Specified by:

persistMessage in interface SpringAIService

buildKbPrompt

public String buildKbPrompt(String systemPrompt,
 String query,
 String context)

processPrompt

protected abstract void processPrompt(org.springframework.ai.chat.prompt.Prompt prompt,
 MessageProtobuf messageProtobuf)

processPromptSync

protected abstract String processPromptSync(String message)

processPromptSSE

protected abstract void processPromptSSE(org.springframework.ai.chat.prompt.Prompt prompt,
 MessageProtobuf messageProtobuf,
 org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter)

generateFaqPairs

protected abstract String generateFaqPairs(String prompt)