Package com.bytedesk.ai.service

Class PromptHelper

java.lang.Object

com.bytedesk.ai.service.PromptHelper

@Component
public class PromptHelper
extends Object

Field Summary

Fields

Modifier and Type	Field	Description
private MessageRestService	messageRestService

Constructor Summary

Constructors

Constructor	Description
PromptHelper()

Method Summary

Modifier and Type	Method	Description
String	buildContextFromFaqs(List<FaqProtobuf> searchResultList)
List<org.springframework.ai.chat.messages.Message>	buildMessagesForSse(String query, String context, RobotProtobuf robot, MessageProtobuf messageProtobufQuery)
List<org.springframework.ai.chat.messages.Message>	buildMessagesForSync(String query, String context, RobotProtobuf robot, MessageProtobuf messageProtobufQuery)
String	createRobotStreamContentAnswer(String question, String answer, List<RobotContent.SourceReference> sourceReferences, RobotProtobuf robot)
String	extractFullPromptContent(List<org.springframework.ai.chat.messages.Message> messages)
String	extractTextFromResponse(Object response)
private String	stripThinkTags(String content)
org.springframework.ai.chat.prompt.Prompt	toPrompt(List<org.springframework.ai.chat.messages.Message> messages)

Methods inherited from class java.lang.Object

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

Field Details

messageRestService

@Autowired
private MessageRestService messageRestService

Constructor Details

PromptHelper

public PromptHelper()

Method Details

buildMessagesForSse

public List<org.springframework.ai.chat.messages.Message> buildMessagesForSse(String query,
 String context,
 RobotProtobuf robot,
 MessageProtobuf messageProtobufQuery)

buildMessagesForSync

public List<org.springframework.ai.chat.messages.Message> buildMessagesForSync(String query,
 String context,
 RobotProtobuf robot,
 MessageProtobuf messageProtobufQuery)

toPrompt

public org.springframework.ai.chat.prompt.Prompt toPrompt(List<org.springframework.ai.chat.messages.Message> messages)

createRobotStreamContentAnswer

public String createRobotStreamContentAnswer(String question,
 String answer,
 List<RobotContent.SourceReference> sourceReferences,
 RobotProtobuf robot)

extractFullPromptContent

public String extractFullPromptContent(List<org.springframework.ai.chat.messages.Message> messages)

extractTextFromResponse

public String extractTextFromResponse(Object response)

stripThinkTags

private String stripThinkTags(String content)

buildContextFromFaqs

public String buildContextFromFaqs(List<FaqProtobuf> searchResultList)