Package com.bytedesk.ai.springai.providers.maxkb

Class SpringAIMaxkbService

java.lang.Object

com.bytedesk.ai.springai.service.BaseSpringAIService

com.bytedesk.ai.springai.providers.maxkb.SpringAIMaxkbService

All Implemented Interfaces:

SpringAIService

@Service
@ConditionalOnProperty(name="bytedesk.maxkb.enabled",
                       havingValue="true",
                       matchIfMissing=false)
public class SpringAIMaxkbService
extends BaseSpringAIService

SpringAI MaxKB Service - 基于 MaxKB API 的SpringAI服务实现 继承BaseSpringAIService，提供统一的AI服务接口

Nested Class Summary

Nested classes/interfaces inherited from class com.bytedesk.ai.springai.service.BaseSpringAIService

BaseSpringAIService.PromptResult

Field Summary

Fields

Modifier and Type	Field	Description
private MaxkbChatService	maxkbChatService

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

applicationEventPublisher, chunkElasticService, chunkVectorService, faqElasticService, faqVectorService, messagePersistCache, messageRestService, messageSendService, robotMessageCache, robotRestService, textElasticService, textVectorService, threadRestService, uidUtils

Constructor Summary

Constructors

Constructor	Description
SpringAIMaxkbService()

Method Summary

Modifier and Type	Method	Description
private String	getApplicationId(RobotProtobuf robot)	获取应用ID 可以从robot配置中获取，或者使用默认值
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, String fullPromptContent)
protected String	processPromptSync(String message, RobotProtobuf robot, String fullPromptContent)
protected void	processPromptWebsocket(org.springframework.ai.chat.prompt.Prompt prompt, RobotProtobuf robot, MessageProtobuf messageProtobufQuery, MessageProtobuf messageProtobufReply, String fullPromptContent)	Token usage data class
private void	recordTokenUsage(RobotProtobuf robot, String response, int inputLength, int outputLength)	记录token使用情况

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

createDynamicOptions, extractFullPromptContent, extractTextFromResponse, extractTokenUsage, getTokenUnitPrice, handleSseError, isEmitterCompleted, persistMessage, persistMessage, persistMessage, persistMessage, processDirectLlmRequest, processLlmResponseWebsocket, publishAiTokenUsageEvent, recordAiTokenUsage, searchKnowledgeBase, sendMessageWebsocket, sendSseMessage, sendStreamEndMessage, sendStreamMessage, sendStreamStartMessage, sendSyncMessage, sendWebsocketMessage

Methods inherited from class java.lang.Object

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

Field Details

maxkbChatService

@Autowired
private MaxkbChatService maxkbChatService

Constructor Details

SpringAIMaxkbService

public SpringAIMaxkbService()

Method Details

processPromptWebsocket

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

Description copied from class: BaseSpringAIService

Token usage data class

Specified by:

processPromptWebsocket in class BaseSpringAIService

processPromptSync

protected String processPromptSync(String message,
 RobotProtobuf robot,
 String fullPromptContent)

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,
 String fullPromptContent)

Specified by:

processPromptSse in class BaseSpringAIService

getApplicationId

private String getApplicationId(RobotProtobuf robot)

获取应用ID 可以从robot配置中获取，或者使用默认值

Parameters:

robot - 机器人信息

Returns:

应用ID

recordTokenUsage

private void recordTokenUsage(RobotProtobuf robot,
 String response,
 int inputLength,
 int outputLength)

记录token使用情况

Parameters:

robot - 机器人信息

response - API响应

inputLength - 输入长度

outputLength - 输出长度