-
Notifications
You must be signed in to change notification settings - Fork 2
/
schema.py
128 lines (123 loc) · 3.28 KB
/
schema.py
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
118
119
120
121
122
123
124
125
126
127
128
chatSchema = {
"$schema": "http://json-schema.org/draft-04/schema#",
"type": "object",
"properties": {
"userName": {
"type": "string"
},
"genParams": {
"type": "object",
"properties": {
"min_length": {
"type": "integer"
},
"max_length": {
"type": "integer"
},
"top_p": {
"type": "number"
},
"min_p": {
"type": "number"
},
"top_k": {
"type": "number"
},
"temperature": {
"type": "number"
},
"token_repetition_penalty": {
"type": "number"
},
"token_repetition_range": {
"type": "integer"
},
"token_repetition_decay": {
"type": "integer"
},
"max_new_tokens": {
"type": "integer"
},
},
"required": [
"max_new_tokens"
]
},
"character": {
"type": "object",
"properties": {
"charName": {
"type": "string"
},
"persona": {
"type": "string"
},
"greeting": {
"type": "string"
},
"emotions": {
"type": "array",
"items":
{
"type": "string"
},
},
"chatExample": {
"type": "array",
"items":
{
"type": "object",
"properties": {
"chatType": {
"type": "string"
},
"message": {
"type": "string"
}
},
"required": [
"chatType",
"message"
]
}
}
},
"required": [
"charName",
"persona",
"greeting",
"chatExample"
]
},
"promptTemplate": {
"type": "object",
"properties": {
"prompt": {
"type": "string"
},
"character": {
"type": "string"
},
"user": {
"type": "string"
}
},
"required": [
"prompt",
"character",
"user"
]
},
"chatHistory": {
"type": "array",
"items": {}
}
},
"required": [
"userName",
"genParams",
"character",
"promptTemplate",
"chatHistory"
]
}