init

ageer

2024-01-16 1f7f97e86a79ee4f1515d753e84a1e20a3ec41fb

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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
package com.xmzs.common.chat.entity.chat;
 
import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
import com.fasterxml.jackson.annotation.JsonInclude;
import com.fasterxml.jackson.annotation.JsonProperty;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.Getter;
 
import java.io.Serializable;
import java.util.List;
 
/**
 * 描述：
 *
 * @author https:www.unfbx.com
 * @since 2023-03-02
 */
@Data
@JsonInclude(JsonInclude.Include.NON_NULL)
@JsonIgnoreProperties(ignoreUnknown = true)
public class Message implements Serializable {
 
    /**
     * 目前支持四个中角色参考官网，进行情景输入：
     * https://platform.openai.com/docs/guides/chat/introduction
     */
    private String role;
 
    private String content;
 
    private String name;
 
    @JsonProperty("function_call")
    private FunctionCall functionCall;
 
    public static Builder builder() {
        return new Builder();
    }
 
    /**
     * 构造函数
     *
     * @param role         角色
     * @param content      描述主题信息
     * @param name         name
     * @param functionCall functionCall
     */
    public Message(String role, String content, String name, FunctionCall functionCall) {
        this.role = role;
        this.content = content;
        this.name = name;
        this.functionCall = functionCall;
    }
 
    public Message() {
    }
 
    private Message(Builder builder) {
        setRole(builder.role);
        setContent(builder.content);
        setName(builder.name);
        setFunctionCall(builder.functionCall);
    }
 
 
    @Getter
    @AllArgsConstructor
    public enum Role {
 
        SYSTEM("system"),
        USER("user"),
        ASSISTANT("assistant"),
        FUNCTION("function"),
        ;
        private String name;
    }
 
    public static final class Builder {
        private String role;
        private String content;
        private String name;
        private FunctionCall functionCall;
 
        public Builder() {
        }
 
        public Builder role(Role role) {
            this.role = role.getName();
            return this;
        }
 
        public Builder role(String role) {
            this.role = role;
            return this;
        }
 
        public Builder content(String content) {
            this.content = content;
            return this;
        }
 
        public Builder name(String name) {
            this.name = name;
            return this;
        }
 
        public Builder functionCall(FunctionCall functionCall) {
            this.functionCall = functionCall;
            return this;
        }
 
        public Message build() {
            return new Message(this);
        }
    }
}