0606

admin

2025-06-06 9cd825aea53fa5ba0cda1485464af027e27f0ce4

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
package tech.aiflowy.ai.entity;
 
import tech.aiflowy.ai.entity.base.AiBotMessageBase;
import tech.aiflowy.common.util.StringUtil;
import com.agentsflex.core.llm.functions.Function;
import com.agentsflex.core.message.AiMessage;
import com.agentsflex.core.message.HumanMessage;
import com.agentsflex.core.message.Message;
import com.agentsflex.core.message.SystemMessage;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson.parser.Feature;
import com.mybatisflex.annotation.Table;
 
import java.util.List;
 
 
/**
 * Bot 消息记录表 实体类。
 *
 * @author michael
 * @since 2024-11-04
 */
 
@Table(value = "tb_ai_bot_message", comment = "Bot 消息记录表")
public class AiBotMessage extends AiBotMessageBase {
 
    public Message toMessage() {
        String role = getRole();
        if ("user".equalsIgnoreCase(role)) {
            HumanMessage humanMessage = new HumanMessage();
            humanMessage.setContent(getContent());
            String functionsJson = getFunctions();
            if (StringUtil.hasText(functionsJson)) {
                List<Function> functions = JSON.parseArray(functionsJson, Function.class, Feature.SupportAutoType);
                if (functions != null && !functions.isEmpty()) {
                    humanMessage.addFunctions(functions);
                }
            }
            return humanMessage;
        } else if ("assistant".equalsIgnoreCase(role)) {
            AiMessage aiMessage = new AiMessage();
            aiMessage.setFullContent(getContent());
            aiMessage.setTotalTokens(getTotalTokens());
            return aiMessage;
        } else if ("system".equalsIgnoreCase(role)) {
            SystemMessage systemMessage = new SystemMessage();
            systemMessage.setContent(getContent());
            return systemMessage;
        }
        return null;
 
    }
}