Package com.bytedesk.ai.springai.providers.ragflow

Class SpringAIRagflowService

java.lang.Object

com.bytedesk.ai.springai.service.BaseSpringAIService

com.bytedesk.ai.springai.providers.ragflow.SpringAIRagflowService

All Implemented Interfaces:

SpringAIService

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

SpringAI RAGFlow Service - 基于 RAGFlow 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 RagflowChatService	ragflowChatService

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
SpringAIRagflowService()

Method Summary

Modifier and Type	Method	Description
private String	getChatId(RobotProtobuf robot, MessageProtobuf messageProtobuf)	获取聊天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

ragflowChatService

@Autowired
private RagflowChatService ragflowChatService

Constructor Details

SpringAIRagflowService

public SpringAIRagflowService()

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

getChatId

private String getChatId(RobotProtobuf robot,
 MessageProtobuf messageProtobuf)

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

Parameters:

robot - 机器人信息

messageProtobuf - 消息信息

Returns:

聊天ID

recordTokenUsage

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

记录token使用情况

Parameters:

robot - 机器人信息

response - API响应

inputLength - 输入长度

outputLength - 输出长度