Package com.bytedesk.ai.springai.providers.dify

Class SpringAIDifyService

java.lang.Object

com.bytedesk.ai.springai.service.BaseSpringAIService

com.bytedesk.ai.springai.providers.dify.SpringAIDifyService

All Implemented Interfaces:

SpringAIService

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

SpringAI Dify Service - 基于 Dify 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 DifyChatService	difyChatService

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

Method Summary

Modifier and Type	Method	Description
private String	getConversationId(RobotProtobuf robot, MessageProtobuf messageProtobuf)	获取或生成会话ID
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, 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

difyChatService

@Autowired
private DifyChatService difyChatService

Constructor Details

SpringAIDifyService

public SpringAIDifyService()

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

getConversationId

private String getConversationId(RobotProtobuf robot,
 MessageProtobuf messageProtobuf)

获取或生成会话ID

Parameters:

robot - 机器人信息

messageProtobuf - 消息信息

Returns:

会话ID

recordTokenUsage

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

记录token使用情况

Parameters:

robot - 机器人信息

inputLength - 输入长度

outputLength - 输出长度