Package com.bytedesk.ai.service

Class SseMessageHelper

java.lang.Object

com.bytedesk.ai.service.SseMessageHelper

@Component
public class SseMessageHelper
extends Object

Field Summary

Fields

Modifier and Type	Field	Description
private MessagePersistenceHelper	messagePersistenceHelper
private PromptHelper	promptHelper

Constructor Summary

Constructors

Constructor	Description
SseMessageHelper()

Method Summary

Modifier and Type	Method	Description
void	handleSseError(Throwable error, MessageProtobuf messageProtobufQuery, MessageProtobuf messageProtobufReply, org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter)
boolean	isEmitterCompleted(org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter)
private boolean	looksLikeRobotContentJson(String content)
void	processPromptSseWithContext(BaseSpringAIService service, String query, String context, RobotProtobuf robot, MessageProtobuf messageProtobufQuery, MessageProtobuf messageProtobufReply, List<RobotContent.SourceReference> sourceReferences, org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter, String errorLogTag)
void	sendDefaultReplySse(String query, RobotProtobuf robot, MessageProtobuf messageProtobufQuery, MessageProtobuf messageProtobufReply, org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter)
void	sendMessageWebsocket(MessageTypeEnum type, String content, MessageProtobuf messageProtobufReply)
void	sendStreamEndMessage(MessageProtobuf messageProtobufQuery, MessageProtobuf messageProtobufReply, org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter, long promptTokens, long completionTokens, long totalTokens, org.springframework.ai.chat.prompt.Prompt prompt, String aiProvider, String aiModel)
void	sendStreamEndMessage(MessageProtobuf messageProtobufQuery, MessageProtobuf messageProtobufReply, org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter, long promptTokens, long completionTokens, long totalTokens, org.springframework.ai.chat.prompt.Prompt prompt, String aiProvider, String aiModel, boolean persistMessage)
void	sendStreamMessage(MessageProtobuf messageProtobufQuery, MessageProtobuf messageProtobufReply, org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter, String content, String reasonContent, List<RobotContent.SourceReference> sourceReferences)
void	sendStreamMessage(MessageProtobuf messageProtobufQuery, MessageProtobuf messageProtobufReply, org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter, String content, String reasonContent, List<RobotContent.SourceReference> sourceReferences, Boolean isUnanswered, boolean completeAfterSend, boolean contentIsStreamContentJson)
void	sendStreamStartMessage(MessageProtobuf messageProtobufQuery, MessageProtobuf messageProtobufReply, org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter, String initialContent)

Methods inherited from class java.lang.Object

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

Field Details

messagePersistenceHelper

@Autowired
private MessagePersistenceHelper messagePersistenceHelper

promptHelper

@Autowired
private PromptHelper promptHelper

Constructor Details

SseMessageHelper

public SseMessageHelper()

Method Details

sendStreamStartMessage

public void sendStreamStartMessage(MessageProtobuf messageProtobufQuery,
 MessageProtobuf messageProtobufReply,
 org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter,
 String initialContent)

sendStreamMessage

public void sendStreamMessage(MessageProtobuf messageProtobufQuery,
 MessageProtobuf messageProtobufReply,
 org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter,
 String content,
 String reasonContent,
 List<RobotContent.SourceReference> sourceReferences)

sendStreamMessage

public void sendStreamMessage(MessageProtobuf messageProtobufQuery,
 MessageProtobuf messageProtobufReply,
 org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter,
 String content,
 String reasonContent,
 List<RobotContent.SourceReference> sourceReferences,
 Boolean isUnanswered,
 boolean completeAfterSend,
 boolean contentIsStreamContentJson)

looksLikeRobotContentJson

private boolean looksLikeRobotContentJson(String content)

sendStreamEndMessage

public void sendStreamEndMessage(MessageProtobuf messageProtobufQuery,
 MessageProtobuf messageProtobufReply,
 org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter,
 long promptTokens,
 long completionTokens,
 long totalTokens,
 org.springframework.ai.chat.prompt.Prompt prompt,
 String aiProvider,
 String aiModel)

sendStreamEndMessage

public void sendStreamEndMessage(MessageProtobuf messageProtobufQuery,
 MessageProtobuf messageProtobufReply,
 org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter,
 long promptTokens,
 long completionTokens,
 long totalTokens,
 org.springframework.ai.chat.prompt.Prompt prompt,
 String aiProvider,
 String aiModel,
 boolean persistMessage)

sendDefaultReplySse

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

sendMessageWebsocket

public void sendMessageWebsocket(MessageTypeEnum type,
 String content,
 MessageProtobuf messageProtobufReply)

processPromptSseWithContext

public void processPromptSseWithContext(BaseSpringAIService service,
 String query,
 String context,
 RobotProtobuf robot,
 MessageProtobuf messageProtobufQuery,
 MessageProtobuf messageProtobufReply,
 List<RobotContent.SourceReference> sourceReferences,
 org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter,
 String errorLogTag)

handleSseError

public void handleSseError(Throwable error,
 MessageProtobuf messageProtobufQuery,
 MessageProtobuf messageProtobufReply,
 org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter)

isEmitterCompleted

public boolean isEmitterCompleted(org.springframework.web.servlet.mvc.method.annotation.SseEmitter emitter)