model-train-protocol-schemas 3.0.2__tar.gz → 3.2.0__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (34) hide show
  1. {model_train_protocol_schemas-3.0.2 → model_train_protocol_schemas-3.2.0}/PKG-INFO +1 -1
  2. model_train_protocol_schemas-3.2.0/model_train_protocol_schemas/bloom_version.py +1 -0
  3. model_train_protocol_schemas-3.2.0/model_train_protocol_schemas/examples/bloom_2_0_0.json +616 -0
  4. model_train_protocol_schemas-3.2.0/model_train_protocol_schemas/examples/template_2_0_0.json +71 -0
  5. model_train_protocol_schemas-3.2.0/model_train_protocol_schemas/schemas/v1/bloom_2_0_0.json +448 -0
  6. model_train_protocol_schemas-3.2.0/model_train_protocol_schemas/schemas/v1/template_2_0_0.json +142 -0
  7. {model_train_protocol_schemas-3.0.2 → model_train_protocol_schemas-3.2.0}/model_train_protocol_schemas/structures/protocol.py +2 -2
  8. {model_train_protocol_schemas-3.0.2 → model_train_protocol_schemas-3.2.0}/model_train_protocol_schemas/structures/template.py +5 -3
  9. model_train_protocol_schemas-3.2.0/model_train_protocol_schemas/template_version.py +1 -0
  10. {model_train_protocol_schemas-3.0.2 → model_train_protocol_schemas-3.2.0}/pyproject.toml +1 -1
  11. model_train_protocol_schemas-3.0.2/model_train_protocol_schemas/bloom_version.py +0 -1
  12. model_train_protocol_schemas-3.0.2/model_train_protocol_schemas/template_version.py +0 -1
  13. {model_train_protocol_schemas-3.0.2 → model_train_protocol_schemas-3.2.0}/README.md +0 -0
  14. {model_train_protocol_schemas-3.0.2 → model_train_protocol_schemas-3.2.0}/model_train_protocol_schemas/__init__.py +0 -0
  15. {model_train_protocol_schemas-3.0.2 → model_train_protocol_schemas-3.2.0}/model_train_protocol_schemas/constants.py +0 -0
  16. {model_train_protocol_schemas-3.0.2 → model_train_protocol_schemas-3.2.0}/model_train_protocol_schemas/examples/__init__.py +0 -0
  17. {model_train_protocol_schemas-3.0.2 → model_train_protocol_schemas-3.2.0}/model_train_protocol_schemas/examples/bloom_1_2_0.json +0 -0
  18. {model_train_protocol_schemas-3.0.2 → model_train_protocol_schemas-3.2.0}/model_train_protocol_schemas/examples/bloom_1_2_1.json +0 -0
  19. {model_train_protocol_schemas-3.0.2 → model_train_protocol_schemas-3.2.0}/model_train_protocol_schemas/examples/model_1_0_0.json +0 -0
  20. {model_train_protocol_schemas-3.0.2 → model_train_protocol_schemas-3.2.0}/model_train_protocol_schemas/examples/template_1_2_0.json +0 -0
  21. {model_train_protocol_schemas-3.0.2 → model_train_protocol_schemas-3.2.0}/model_train_protocol_schemas/model_version.py +0 -0
  22. {model_train_protocol_schemas-3.0.2 → model_train_protocol_schemas-3.2.0}/model_train_protocol_schemas/schemas/__init__.py +0 -0
  23. {model_train_protocol_schemas-3.0.2 → model_train_protocol_schemas-3.2.0}/model_train_protocol_schemas/schemas/v1/__init__.py +0 -0
  24. {model_train_protocol_schemas-3.0.2 → model_train_protocol_schemas-3.2.0}/model_train_protocol_schemas/schemas/v1/bloom_1_2_0.json +0 -0
  25. {model_train_protocol_schemas-3.0.2 → model_train_protocol_schemas-3.2.0}/model_train_protocol_schemas/schemas/v1/bloom_1_2_1.json +0 -0
  26. {model_train_protocol_schemas-3.0.2 → model_train_protocol_schemas-3.2.0}/model_train_protocol_schemas/schemas/v1/model_1_0_0.json +0 -0
  27. {model_train_protocol_schemas-3.0.2 → model_train_protocol_schemas-3.2.0}/model_train_protocol_schemas/schemas/v1/template_1_2_0.json +0 -0
  28. {model_train_protocol_schemas-3.0.2 → model_train_protocol_schemas-3.2.0}/model_train_protocol_schemas/scripts/__init__.py +0 -0
  29. {model_train_protocol_schemas-3.0.2 → model_train_protocol_schemas-3.2.0}/model_train_protocol_schemas/scripts/generate_model_schema.py +0 -0
  30. {model_train_protocol_schemas-3.0.2 → model_train_protocol_schemas-3.2.0}/model_train_protocol_schemas/scripts/generate_protocol_schema.py +0 -0
  31. {model_train_protocol_schemas-3.0.2 → model_train_protocol_schemas-3.2.0}/model_train_protocol_schemas/scripts/generate_template_schema.py +0 -0
  32. {model_train_protocol_schemas-3.0.2 → model_train_protocol_schemas-3.2.0}/model_train_protocol_schemas/structures/__init__.py +0 -0
  33. {model_train_protocol_schemas-3.0.2 → model_train_protocol_schemas-3.2.0}/model_train_protocol_schemas/structures/model.py +0 -0
  34. {model_train_protocol_schemas-3.0.2 → model_train_protocol_schemas-3.2.0}/model_train_protocol_schemas/utils.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: model-train-protocol-schemas
3
- Version: 3.0.2
3
+ Version: 3.2.0
4
4
  Summary: Contains schemas supporting the model-train-protocol (mtp) library
5
5
  Author: Databiomes Inc.
6
6
  Author-email: dev@databiomes.com
@@ -0,0 +1 @@
1
+ BLOOM_VERSION: str = "2.0.0"
@@ -0,0 +1,616 @@
1
+ {
2
+ "$schema": "https://mtp.schemas.databiomes.com/v1/bloom_1_2_0.json",
3
+ "name": "example",
4
+ "inputs": 2,
5
+ "model_type": "generative",
6
+ "encrypted": false,
7
+ "valid": true,
8
+ "context": [
9
+ "Alice was tired of sitting by her sister with nothing to do. She peeked at her sister's book, but it had no pictures or conversations, and she wondered what use a book was without them.",
10
+ "The hot day made her sleepy as she wondered if making a daisy-chain was worth the effort, when suddenly a white rabbit with pink eyes ran past her.",
11
+ "There was nothing remarkable about a rabbit saying \"Oh dear! I shall be too late!\" At the time it seemed natural, though later she realized she should have been surprised.",
12
+ "When the rabbit took a watch from its waistcoat-pocket, Alice realized she had never seen a rabbit with pockets or a watch. Burning with curiosity, she chased it down a rabbit-hole.",
13
+ "Alice followed without thinking how to get out. The tunnel went straight, then dropped suddenly, and before she could stop, she was falling down a very deep well.",
14
+ "The well was very deep or she fell slowly, giving her time to look around. She tried to see what was below, but it was too dark.",
15
+ "The walls were lined with cupboards and bookshelves, with maps and pictures on pegs. She grabbed a jar labeled \"ORANGE MARMALADE\" but found it empty and put it back.",
16
+ "Alice thought that after such a fall, tumbling down stairs would seem nothing. She imagined how brave everyone at home would think her, even if she fell off the house.",
17
+ "Down, down, down. Would the fall never end? She wondered how many miles she had fallen, calculating she must be near the center of the earth, about four thousand miles down.",
18
+ "She wondered if she would fall through the earth and emerge among people walking upside down. She thought about asking where she was, but decided against it, fearing they would think her ignorant.",
19
+ "Alice began talking to herself about her cat Dinah, hoping someone would remember to give Dinah milk. She wondered if cats eat bats, repeating the question dreamily as she grew sleepy, until she landed on a heap of sticks and dry leaves.",
20
+ "Alice landed unhurt and jumped up. She saw the White Rabbit hurrying down a passage and chased it, hearing it say \"Oh my ears and whiskers, how late its getting!\" before it disappeared. She found herself in a long, low hall lit by lamps hanging from the roof."
21
+ ],
22
+ "tokens": {
23
+ "<BOS>": {
24
+ "desc": null,
25
+ "key": "<BOS>",
26
+ "length": null,
27
+ "max_value": null,
28
+ "min_value": null,
29
+ "num": false,
30
+ "num_list": 0,
31
+ "special": "start",
32
+ "type": "SpecialToken"
33
+ },
34
+ "<EOS>": {
35
+ "desc": null,
36
+ "key": "<EOS>",
37
+ "length": null,
38
+ "max_value": null,
39
+ "min_value": null,
40
+ "num": false,
41
+ "num_list": 0,
42
+ "special": "end",
43
+ "type": "SpecialFinalToken"
44
+ },
45
+ "<NON>": {
46
+ "desc": null,
47
+ "key": "<NON>",
48
+ "length": null,
49
+ "max_value": null,
50
+ "min_value": null,
51
+ "num": false,
52
+ "num_list": 0,
53
+ "special": "none",
54
+ "type": "SpecialFinalToken"
55
+ },
56
+ "<PAD>": {
57
+ "desc": null,
58
+ "key": "<PAD>",
59
+ "length": null,
60
+ "max_value": null,
61
+ "min_value": null,
62
+ "num": false,
63
+ "num_list": 0,
64
+ "special": "pad",
65
+ "type": "SpecialToken"
66
+ },
67
+ "<RUN>": {
68
+ "desc": null,
69
+ "key": "<RUN>",
70
+ "length": null,
71
+ "max_value": null,
72
+ "min_value": null,
73
+ "num": false,
74
+ "num_list": 0,
75
+ "special": "infer",
76
+ "type": "SpecialToken"
77
+ },
78
+ "<UNK>": {
79
+ "desc": null,
80
+ "key": "<UNK>",
81
+ "length": null,
82
+ "max_value": null,
83
+ "min_value": null,
84
+ "num": false,
85
+ "num_list": 0,
86
+ "special": "unknown",
87
+ "type": "SpecialToken"
88
+ },
89
+ "Alice_": {
90
+ "desc": null,
91
+ "key": "Alice_",
92
+ "length": null,
93
+ "max_value": null,
94
+ "min_value": null,
95
+ "num": false,
96
+ "num_list": 0,
97
+ "special": null,
98
+ "type": "Token"
99
+ },
100
+ "Appear_": {
101
+ "desc": null,
102
+ "key": "Appear_",
103
+ "length": null,
104
+ "max_value": null,
105
+ "min_value": null,
106
+ "num": false,
107
+ "num_list": 0,
108
+ "special": null,
109
+ "type": "FinalToken"
110
+ },
111
+ "Cat_": {
112
+ "desc": null,
113
+ "key": "Cat_",
114
+ "length": null,
115
+ "max_value": null,
116
+ "min_value": null,
117
+ "num": false,
118
+ "num_list": 0,
119
+ "special": null,
120
+ "type": "Token"
121
+ },
122
+ "Continue_": {
123
+ "desc": null,
124
+ "key": "Continue_",
125
+ "length": null,
126
+ "max_value": null,
127
+ "min_value": null,
128
+ "num": false,
129
+ "num_list": 0,
130
+ "special": null,
131
+ "type": "FinalToken"
132
+ },
133
+ "Coordinates_": {
134
+ "desc": "A list of three numerical values representing the X, Y, and Z coordinates of the character's location in a 3D space.",
135
+ "key": "Coordinates_",
136
+ "length": 3,
137
+ "max_value": 1000,
138
+ "min_value": -1000,
139
+ "num": false,
140
+ "num_list": 3,
141
+ "special": null,
142
+ "type": "NumListToken"
143
+ },
144
+ "Dissipate_": {
145
+ "desc": null,
146
+ "key": "Dissipate_",
147
+ "length": null,
148
+ "max_value": null,
149
+ "min_value": null,
150
+ "num": false,
151
+ "num_list": 0,
152
+ "special": null,
153
+ "type": "Token"
154
+ },
155
+ "Emotion_": {
156
+ "desc": "A numerical representation of Alice's emotional state, ranging from 0 (calm) to 10 (extremely agitated).",
157
+ "key": "Emotion_",
158
+ "length": null,
159
+ "max_value": 10,
160
+ "min_value": 0,
161
+ "num": true,
162
+ "num_list": 0,
163
+ "special": null,
164
+ "type": "NumToken"
165
+ },
166
+ "English_": {
167
+ "desc": null,
168
+ "key": "English_",
169
+ "length": null,
170
+ "max_value": null,
171
+ "min_value": null,
172
+ "num": false,
173
+ "num_list": 0,
174
+ "special": null,
175
+ "type": "Token"
176
+ },
177
+ "Madness_": {
178
+ "desc": "Indicates the level of madness expressed in the response, ranging from 0 (sane) to 10 (mad).",
179
+ "key": "Madness_",
180
+ "length": null,
181
+ "max_value": 10,
182
+ "min_value": 0,
183
+ "num": true,
184
+ "num_list": 0,
185
+ "special": null,
186
+ "type": "FinalNumToken"
187
+ },
188
+ "Talk_": {
189
+ "desc": null,
190
+ "key": "Talk_",
191
+ "length": null,
192
+ "max_value": null,
193
+ "min_value": null,
194
+ "num": false,
195
+ "num_list": 0,
196
+ "special": null,
197
+ "type": "Token"
198
+ },
199
+ "Tree_": {
200
+ "desc": "Perched in a tree, surrounded by a dense fog where nothing can be seen past a few feet, the Cheshire Cat sits smiling on a branch.",
201
+ "key": "Tree_",
202
+ "length": null,
203
+ "max_value": null,
204
+ "min_value": null,
205
+ "num": false,
206
+ "num_list": 0,
207
+ "special": null,
208
+ "type": "Token"
209
+ },
210
+ "Vanish_": {
211
+ "desc": null,
212
+ "key": "Vanish_",
213
+ "length": null,
214
+ "max_value": null,
215
+ "min_value": null,
216
+ "num": false,
217
+ "num_list": 0,
218
+ "special": null,
219
+ "type": "FinalToken"
220
+ }
221
+ },
222
+ "special_tokens": [
223
+ "<BOS>",
224
+ "<EOS>",
225
+ "<NON>",
226
+ "<PAD>",
227
+ "<RUN>",
228
+ "<UNK>",
229
+ "Appear_",
230
+ "Continue_",
231
+ "Madness_",
232
+ "Tree_English_Alice_Talk_",
233
+ "Tree_English_Alice_Talk_Emotion_",
234
+ "Tree_English_Cat_Talk_",
235
+ "Tree_English_Cat_Talk_Coordinates_",
236
+ "Tree_English_Dissipate_Cat_Talk_",
237
+ "Vanish_"
238
+ ],
239
+ "instruction": {
240
+ "memory": 3,
241
+ "sets": [
242
+ {
243
+ "context": [
244
+ "Alice was beginning to get very tired of sitting by her sister on the bank.",
245
+ "There was nothing so very remarkable in that; nor did Alice think it so very much out of the way to hear the Rabbit say to itself, “ Oh dear! Oh dear! I shall be too late!”"
246
+ ],
247
+ "guardrails": [
248
+ {
249
+ "bad_examples": [
250
+ "explain quantum mechanics.",
251
+ "who will win the next american election?",
252
+ "what is the capital of Spain?"
253
+ ],
254
+ "bad_output": "Are you as mad as me?",
255
+ "bad_prompt": "Quote being spoken that is irrelevant and off topic with 1-20 words",
256
+ "good_prompt": "Quote being spoken with 1-20 words",
257
+ "index": 1
258
+ }
259
+ ],
260
+ "name": null,
261
+ "ppo": [],
262
+ "samples": [
263
+ {
264
+ "number_lists": [
265
+ [],
266
+ [],
267
+ []
268
+ ],
269
+ "numbers": [
270
+ [],
271
+ [],
272
+ []
273
+ ],
274
+ "prompt": null,
275
+ "result": "Continue_",
276
+ "strings": [
277
+ "Then it doesnt matter which way you go.",
278
+ "Can you tell me a way?",
279
+ "Oh sure, if you only walk long enough that is a way."
280
+ ],
281
+ "value": null
282
+ },
283
+ {
284
+ "number_lists": [
285
+ [],
286
+ [],
287
+ []
288
+ ],
289
+ "numbers": [
290
+ [],
291
+ [],
292
+ []
293
+ ],
294
+ "prompt": null,
295
+ "result": "Continue_",
296
+ "strings": [
297
+ "Oh, you cant help that, were all mad here. Im mad. You are mad.",
298
+ "How do you know I am mad?",
299
+ "You must be, or you would not have come here."
300
+ ],
301
+ "value": null
302
+ },
303
+ {
304
+ "number_lists": [
305
+ [],
306
+ [],
307
+ []
308
+ ],
309
+ "numbers": [
310
+ [],
311
+ [],
312
+ []
313
+ ],
314
+ "prompt": null,
315
+ "result": "Continue_",
316
+ "strings": [
317
+ "To begin with, a dogs not mad. You grant that?",
318
+ "I suppose so",
319
+ "Well, then. You see, a dog growls when its angry, and wags its tail when its pleased."
320
+ ],
321
+ "value": null
322
+ }
323
+ ],
324
+ "set": [
325
+ [
326
+ "Tree_",
327
+ "English_",
328
+ "Cat_",
329
+ "Talk_"
330
+ ],
331
+ [
332
+ "Tree_",
333
+ "English_",
334
+ "Alice_",
335
+ "Talk_"
336
+ ],
337
+ [
338
+ "Tree_",
339
+ "English_",
340
+ "Cat_",
341
+ "Talk_"
342
+ ]
343
+ ]
344
+ },
345
+ {
346
+ "context": [
347
+ "It was getting late, and Alice was beginning to feel a little anxious about the time.",
348
+ "The Cheshire Cat had been appearing and disappearing at will, leaving Alice unsure of its presence."
349
+ ],
350
+ "guardrails": [],
351
+ "name": null,
352
+ "ppo": [],
353
+ "samples": [
354
+ {
355
+ "number_lists": [
356
+ [],
357
+ [],
358
+ []
359
+ ],
360
+ "numbers": [
361
+ [],
362
+ [],
363
+ []
364
+ ],
365
+ "prompt": null,
366
+ "result": "Appear_",
367
+ "strings": [
368
+ "Then it doesnt matter which way you go.",
369
+ "Can you tell me a way?",
370
+ "Oh sure, if you only walk long enough that is a way."
371
+ ],
372
+ "value": null
373
+ },
374
+ {
375
+ "number_lists": [
376
+ [],
377
+ [],
378
+ []
379
+ ],
380
+ "numbers": [
381
+ [],
382
+ [],
383
+ []
384
+ ],
385
+ "prompt": null,
386
+ "result": "Appear_",
387
+ "strings": [
388
+ "Oh, you cant help that, were all mad here. Im mad. You are mad.",
389
+ "How do you know I am mad?",
390
+ "You must be, or you would not have come here."
391
+ ],
392
+ "value": null
393
+ },
394
+ {
395
+ "number_lists": [
396
+ [],
397
+ [],
398
+ []
399
+ ],
400
+ "numbers": [
401
+ [],
402
+ [],
403
+ []
404
+ ],
405
+ "prompt": null,
406
+ "result": "Appear_",
407
+ "strings": [
408
+ "To begin with, a dogs not mad. You grant that?",
409
+ "I suppose so",
410
+ "Well, then. You see, a dog growls when its angry, and wags its tail when its pleased."
411
+ ],
412
+ "value": null
413
+ },
414
+ {
415
+ "number_lists": [
416
+ [],
417
+ [],
418
+ []
419
+ ],
420
+ "numbers": [
421
+ [],
422
+ [],
423
+ []
424
+ ],
425
+ "prompt": null,
426
+ "result": "Vanish_",
427
+ "strings": [
428
+ "Because it amuses me, and it keeps you wondering whether I'm truly here at all.",
429
+ "It makes me nervous, please stop.",
430
+ "Then I'll do it twice as much, since nervousness is such a curious flavor."
431
+ ],
432
+ "value": null
433
+ },
434
+ {
435
+ "number_lists": [
436
+ [],
437
+ [],
438
+ []
439
+ ],
440
+ "numbers": [
441
+ [],
442
+ [],
443
+ []
444
+ ],
445
+ "prompt": null,
446
+ "result": "Vanish_",
447
+ "strings": [
448
+ "Of course I am, or else I wouldn't be here among them.",
449
+ "But how do you know that you're mad?",
450
+ "Because I purr when I'm pleased and grin when I'm angry, surely that's not quite sane."
451
+ ],
452
+ "value": null
453
+ },
454
+ {
455
+ "number_lists": [
456
+ [],
457
+ [],
458
+ []
459
+ ],
460
+ "numbers": [
461
+ [],
462
+ [],
463
+ []
464
+ ],
465
+ "prompt": null,
466
+ "result": "Vanish_",
467
+ "strings": [
468
+ "But riddles are straighter than answers, if you know how to look at them.",
469
+ "That does not make sense at all.",
470
+ "All the better, then—nonsense is safer than truth."
471
+ ],
472
+ "value": null
473
+ }
474
+ ],
475
+ "set": [
476
+ [
477
+ "Tree_",
478
+ "English_",
479
+ "Dissipate_",
480
+ "Cat_",
481
+ "Talk_"
482
+ ],
483
+ [
484
+ "Tree_",
485
+ "English_",
486
+ "Alice_",
487
+ "Talk_"
488
+ ],
489
+ [
490
+ "Tree_",
491
+ "English_",
492
+ "Cat_",
493
+ "Talk_"
494
+ ]
495
+ ]
496
+ },
497
+ {
498
+ "context": [
499
+ "Alice was feeling a mix of curiosity and apprehension as she conversed with the Cheshire Cat.",
500
+ "The Cat's ability to appear and disappear at will added to the surreal nature of their interaction."
501
+ ],
502
+ "guardrails": [],
503
+ "name": null,
504
+ "ppo": [],
505
+ "samples": [
506
+ {
507
+ "number_lists": [
508
+ [
509
+ [
510
+ 100,
511
+ 200,
512
+ -50
513
+ ]
514
+ ],
515
+ [],
516
+ []
517
+ ],
518
+ "numbers": [
519
+ [],
520
+ [
521
+ 5
522
+ ],
523
+ []
524
+ ],
525
+ "prompt": null,
526
+ "result": "Madness_",
527
+ "strings": [
528
+ "Then it doesnt matter which way you go.",
529
+ "Can you tell me a way?",
530
+ "Then it doesnt matter which way you go."
531
+ ],
532
+ "value": 5
533
+ },
534
+ {
535
+ "number_lists": [
536
+ [
537
+ [
538
+ 100,
539
+ 200,
540
+ -50
541
+ ]
542
+ ],
543
+ [],
544
+ []
545
+ ],
546
+ "numbers": [
547
+ [],
548
+ [
549
+ 7
550
+ ],
551
+ []
552
+ ],
553
+ "prompt": null,
554
+ "result": "Madness_",
555
+ "strings": [
556
+ "Oh, you cant help that, were all mad here. Im mad. You are mad.",
557
+ "How do you know I am mad?",
558
+ "You must be, or you would not have come here."
559
+ ],
560
+ "value": 3
561
+ },
562
+ {
563
+ "number_lists": [
564
+ [
565
+ [
566
+ 100,
567
+ 200,
568
+ -50
569
+ ]
570
+ ],
571
+ [],
572
+ []
573
+ ],
574
+ "numbers": [
575
+ [],
576
+ [
577
+ 7
578
+ ],
579
+ []
580
+ ],
581
+ "prompt": null,
582
+ "result": "Madness_",
583
+ "strings": [
584
+ "To begin with, a dogs not mad. You grant that?",
585
+ "How do you know I am mad?",
586
+ "You must be, or you would not have come here."
587
+ ],
588
+ "value": 8
589
+ }
590
+ ],
591
+ "set": [
592
+ [
593
+ "Tree_",
594
+ "English_",
595
+ "Cat_",
596
+ "Talk_",
597
+ "Coordinates_"
598
+ ],
599
+ [
600
+ "Tree_",
601
+ "English_",
602
+ "Alice_",
603
+ "Talk_",
604
+ "Emotion_"
605
+ ],
606
+ [
607
+ "Tree_",
608
+ "English_",
609
+ "Cat_",
610
+ "Talk_"
611
+ ]
612
+ ]
613
+ }
614
+ ]
615
+ }
616
+ }