forked from gptscript-ai/go-gptscript
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdatasets.go
50 lines (41 loc) · 1.25 KB
/
datasets.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
package gptscript
type DatasetElementMeta struct {
Name string `json:"name"`
Description string `json:"description"`
}
type DatasetElement struct {
DatasetElementMeta `json:",inline"`
Contents string `json:"contents"`
}
type DatasetMeta struct {
ID string `json:"id"`
Name string `json:"name"`
Description string `json:"description"`
}
type Dataset struct {
DatasetMeta `json:",inline"`
BaseDir string `json:"baseDir,omitempty"`
Elements map[string]DatasetElementMeta `json:"elements"`
}
type datasetRequest struct {
Input string `json:"input"`
Workspace string `json:"workspace"`
DatasetToolRepo string `json:"datasetToolRepo"`
}
type createDatasetArgs struct {
Name string `json:"datasetName"`
Description string `json:"datasetDescription"`
}
type addDatasetElementArgs struct {
DatasetID string `json:"datasetID"`
ElementName string `json:"elementName"`
ElementDescription string `json:"elementDescription"`
ElementContent string `json:"elementContent"`
}
type listDatasetElementArgs struct {
DatasetID string `json:"datasetID"`
}
type getDatasetElementArgs struct {
DatasetID string `json:"datasetID"`
Element string `json:"element"`
}