Commit 05a09c72 authored by alex yao's avatar alex yao

style: 大模型content 多模态

parent 6fc892b9
......@@ -596,7 +596,7 @@ public class AgentApplicationInfoServiceImpl implements AgentApplicationInfoServ
* @param promptTemplate
* @return
*/
private static Message[] buildMessages(List<Message> messages, Integer communicationTurn, String promptTemplate) {
private Message[] buildMessages(List<Message> messages, Integer communicationTurn, String promptTemplate) {
int messLength = messages.size() - 1;
int skip = communicationTurn * 2;
if (skip < messLength) {
......@@ -609,10 +609,21 @@ public class AgentApplicationInfoServiceImpl implements AgentApplicationInfoServ
systemMessage.setContent(promptTemplate);
messages.add(0, systemMessage);
}
return messages.toArray(new Message[0]);
// 转换消息体
Message[] messageArray = new Message[messages.size()];
int index = 0;
for (Message message : messages) {
if (message.getContent() instanceof List) {
List<HashMap<String,Object>> content = (List<HashMap<String,Object>>) message.getContent();
if (content.get(0).get("type").equals("text")) {
message.setContent(content.get(0).get("text").toString());
}
}
messageArray[index++] = message;
}
return messageArray;
}
/**
* 获取模型
*
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment