@@ -887,12 +887,17 @@ describe("gptscript module", () => {
887
887
} , 20000 )
888
888
889
889
test ( "dataset operations" , async ( ) => {
890
- process . env . GPTSCRIPT_WORKSPACE_ID = await g . createWorkspace ( "directory" )
890
+ const workspaceID = await g . createWorkspace ( "directory" )
891
+ const client = new gptscript . GPTScript ( {
892
+ APIKey : process . env . OPENAI_API_KEY ,
893
+ Env : [ "GPTSCRIPT_WORKSPACE_ID=" + workspaceID ]
894
+ } )
895
+
891
896
let datasetID : string
892
897
893
898
// Create and add two elements
894
899
try {
895
- datasetID = await g . addDatasetElements ( [
900
+ datasetID = await client . addDatasetElements ( [
896
901
{
897
902
name : "element1" ,
898
903
description : "" ,
@@ -903,37 +908,37 @@ describe("gptscript module", () => {
903
908
description : "a description" ,
904
909
binaryContents : Buffer . from ( "this is element 2 contents" )
905
910
}
906
- ] )
911
+ ] , { name : "test-dataset" , description : "a test dataset" } )
907
912
} catch ( e ) {
908
913
throw new Error ( "failed to create dataset: " + e )
909
914
}
910
915
911
916
// Add another element
912
917
try {
913
- await g . addDatasetElements ( [
918
+ await client . addDatasetElements ( [
914
919
{
915
920
name : "element3" ,
916
921
description : "a description" ,
917
922
contents : "this is element 3 contents"
918
923
}
919
- ] , datasetID )
924
+ ] , { datasetID : datasetID } )
920
925
} catch ( e ) {
921
926
throw new Error ( "failed to add elements: " + e )
922
927
}
923
928
924
929
// Get elements
925
930
try {
926
- const e1 = await g . getDatasetElement ( datasetID , "element1" )
931
+ const e1 = await client . getDatasetElement ( datasetID , "element1" )
927
932
expect ( e1 . name ) . toEqual ( "element1" )
928
933
expect ( e1 . description ) . toEqual ( "" )
929
934
expect ( e1 . contents ) . toEqual ( "this is element 1 contents" )
930
935
931
- const e2 = await g . getDatasetElement ( datasetID , "element2" )
936
+ const e2 = await client . getDatasetElement ( datasetID , "element2" )
932
937
expect ( e2 . name ) . toEqual ( "element2" )
933
938
expect ( e2 . description ) . toEqual ( "a description" )
934
939
expect ( e2 . binaryContents ) . toEqual ( Buffer . from ( "this is element 2 contents" ) )
935
940
936
- const e3 = await g . getDatasetElement ( datasetID , "element3" )
941
+ const e3 = await client . getDatasetElement ( datasetID , "element3" )
937
942
expect ( e3 . name ) . toEqual ( "element3" )
938
943
expect ( e3 . description ) . toEqual ( "a description" )
939
944
expect ( e3 . contents ) . toEqual ( "this is element 3 contents" )
@@ -943,7 +948,7 @@ describe("gptscript module", () => {
943
948
944
949
// List the elements in the dataset
945
950
try {
946
- const elements = await g . listDatasetElements ( datasetID )
951
+ const elements = await client . listDatasetElements ( datasetID )
947
952
expect ( elements . length ) . toEqual ( 3 )
948
953
expect ( elements . map ( e => e . name ) ) . toContain ( "element1" )
949
954
expect ( elements . map ( e => e . name ) ) . toContain ( "element2" )
@@ -954,9 +959,11 @@ describe("gptscript module", () => {
954
959
955
960
// List datasets
956
961
try {
957
- const datasets = await g . listDatasets ( )
962
+ const datasets = await client . listDatasets ( )
958
963
expect ( datasets . length ) . toBeGreaterThan ( 0 )
959
- expect ( datasets ) . toContain ( datasetID )
964
+ expect ( datasets [ 0 ] . id ) . toEqual ( datasetID )
965
+ expect ( datasets [ 0 ] . name ) . toEqual ( "test-dataset" )
966
+ expect ( datasets [ 0 ] . description ) . toEqual ( "a test dataset" )
960
967
} catch ( e ) {
961
968
throw new Error ( "failed to list datasets: " + e )
962
969
}
0 commit comments