forked from u2takey/chatgpt-go
-
Notifications
You must be signed in to change notification settings - Fork 0
/
types.go
71 lines (61 loc) · 1.67 KB
/
types.go
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
package chatgpt_go
import "encoding/json"
type Content struct {
ContentType string `json:"content_type"`
Parts []string `json:"parts"`
}
type MessageMeta struct {
ModelSlug string `json:"model_slug"`
}
type Message struct {
Id string `json:"id"`
Role string `json:"role,omitempty"`
Content *Content `json:"content,omitempty"`
MetaData *MessageMeta `json:"meta_data,omitempty"`
}
type ConversationData struct {
Action string `json:"action"`
Messages []Message `json:"messages"`
ConversationId string `json:"conversation_id,omitempty"`
ParentMessageId string `json:"parent_message_id,omitempty"`
Model string `json:"model"`
}
type RetMessage struct {
Message string `json:"message"`
ConversationId string `json:"conversation_id"`
ParentId string `json:"parent_id"`
Model string `json:"model"`
}
type LineWithContent struct {
Message Message `json:"message"`
Detail Detail `json:"detail"`
ConversationId string `json:"conversation_id"`
}
type Detail struct {
Str string `json:"str"`
Code string `json:"code"`
Message string `json:"message"`
}
func (d *Detail) UnmarshalJSON(data []byte) error {
if len(data) > 1 && data[0] == '{' {
a := struct {
Code string `json:"code"`
Message string `json:"message"`
}{}
err := json.Unmarshal(data, &a)
d.Code = a.Code
d.Message = a.Message
return err
}
if len(data) >= 2 {
d.Str = string(data[1 : len(data)-1])
}
return nil
}
type ConversationResponse struct {
Items []Conversation `json:"items"`
}
type Conversation struct {
Id string `json:"id"`
}
type MsgHistory map[string]interface{}