| | |
|---|
| | | String question = dto.getQuestion(); |
|---|
| | | log.info("----- standard request -----"); |
|---|
| | | |
|---|
| | | |
|---|
| | | List<ChatMessage> messages = new ArrayList<>(); |
|---|
| | | final ChatMessage systemMessage = ChatMessage.builder() |
|---|
| | | .role(ChatMessageRole.SYSTEM) |
|---|
| | | .content("你是豆包，是由字节跳动开发的 AI 人工智能助手") |
|---|
| | | .build(); |
|---|
| | | |
|---|
| | | List<ChatMessage> messages = Arrays.asList(systemMessage); |
|---|
| | | messages.add(systemMessage); |
|---|
| | | //获取消息记录 |
|---|
| | | if (StrUtil.isNotEmpty(dto.getTalkId())){ |
|---|
| | | List<AiTalkItem> aiTalkItems = aiTalkItemService.getListByTalkId(dto.getTalkId()); |
|---|
| | | if(CollUtil.isNotEmpty(aiTalkItems)){ |
|---|
| | | for (AiTalkItem aiTalkItem : aiTalkItems){ |
|---|
| | | if (aiTalkItem.getType() == 1){ |
|---|
| | | ChatMessage userMessage = ChatMessage.builder() |
|---|
| | | ChatMessage memberMessage = ChatMessage.builder() |
|---|
| | | .role(ChatMessageRole.USER) |
|---|
| | | .content(aiTalkItem.getContext()) |
|---|
| | | .build(); |
|---|
| | | messages.add(userMessage); |
|---|
| | | messages.add(memberMessage); |
|---|
| | | } |
|---|
| | | if (aiTalkItem.getType() == 2){ |
|---|
| | | ChatMessage assistantMessage = ChatMessage.builder() |
|---|
