Package com.bytedesk.ai.springai.providers.custom

Class SpringAICustomService

java.lang.Object

com.bytedesk.ai.springai.service.BaseSpringAIService

com.bytedesk.ai.springai.providers.custom.SpringAICustomService

All Implemented Interfaces:

SpringAIService

@Service
@ConditionalOnProperty(name="spring.ai.custom.chat.enabled",
                       havingValue="true",
                       matchIfMissing=false)
public class SpringAICustomService
extends BaseSpringAIService

Field Summary

Fields

Modifier and Type	Field	Description
private org.springframework.ai.openai.OpenAiChatModel	customChatModel

Fields inherited from class com.bytedesk.ai.springai.service.BaseSpringAIService

faqService, messagePersistCache, messageSendService, robotMessageCache, robotRestService, springAIVectorService, threadRestService, uidUtils

Constructor Summary

Constructors

Constructor	Description
SpringAICustomService()

Method Summary

Modifier and Type	Method	Description
private org.springframework.ai.openai.OpenAiChatOptions	createDynamicOptions(RobotLlm llm)	根据机器人配置创建动态的OpenAiChatOptions
protected String	generateFaqPairs(String prompt)
org.springframework.ai.openai.OpenAiChatModel	getChatModel()
boolean	isServiceHealthy()
protected void	processPrompt(org.springframework.ai.chat.prompt.Prompt prompt, RobotProtobuf robot, MessageProtobuf messageProtobufQuery, MessageProtobuf messageProtobufReply)
protected void	processPromptSSE(org.springframework.ai.chat.prompt.Prompt prompt, RobotProtobuf robot, MessageProtobuf messageProtobufQuery, MessageProtobuf messageProtobufReply, org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter)
protected String	processPromptSync(String message)

Methods inherited from class com.bytedesk.ai.springai.service.BaseSpringAIService

buildKbPrompt, createDynamicOptions, generateFaqPairsAsync, generateFaqPairsSync, handleSseError, isEmitterCompleted, persistMessage, sendMessage, sendSseMessage, sendStreamEndMessage, sendStreamMessage, sendStreamStartMessage, sendWebsocketMessage

Methods inherited from class java.lang.Object

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

Field Details

customChatModel

@Autowired(required=false)
private org.springframework.ai.openai.OpenAiChatModel customChatModel

Constructor Details

SpringAICustomService

public SpringAICustomService()

Method Details

createDynamicOptions

private org.springframework.ai.openai.OpenAiChatOptions createDynamicOptions(RobotLlm llm)

根据机器人配置创建动态的OpenAiChatOptions

Parameters:

llm - 机器人LLM配置

Returns:

根据机器人配置创建的选项

processPrompt

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

Specified by:

processPrompt in class BaseSpringAIService

generateFaqPairs

protected String generateFaqPairs(String prompt)

Specified by:

generateFaqPairs in class BaseSpringAIService

processPromptSync

protected String processPromptSync(String message)

Specified by:

processPromptSync in class BaseSpringAIService

processPromptSSE

protected void processPromptSSE(org.springframework.ai.chat.prompt.Prompt prompt,
 RobotProtobuf robot,
 MessageProtobuf messageProtobufQuery,
 MessageProtobuf messageProtobufReply,
 org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter)

Specified by:

processPromptSSE in class BaseSpringAIService

getChatModel

public org.springframework.ai.openai.OpenAiChatModel getChatModel()

isServiceHealthy

public boolean isServiceHealthy()