fix: datasets: handle error responses

Signed-off-by: Grant Linville <[email protected]>
gptscript-ai · Oct 14, 2024 · b109bdf · b109bdf
1 parent 96d3724
commit b109bdf
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/src/gptscript.ts b/src/gptscript.ts
@@ -401,6 +401,9 @@ export class GPTScript {
         const r: Run = new RunSubcommand("datasets", "", {URL: this.opts.URL, Token: this.opts.Token})
         r.request({input: "{}", workspace: workspace, datasetToolRepo: this.opts.DatasetToolRepo ?? ""})
         const result = await r.text()
+        if (result.startsWith("ERROR:")) {
+            throw new Error(result)
+        }
         return JSON.parse(result) as Array<DatasetMeta>
     }
 
@@ -416,6 +419,9 @@ export class GPTScript {
             datasetToolRepo: this.opts.DatasetToolRepo ?? ""
         })
         const result = await r.text()
+        if (result.startsWith("ERROR:")) {
+            throw new Error(result)
+        }
         return JSON.parse(result) as Dataset
     }
 
@@ -436,6 +442,9 @@ export class GPTScript {
             datasetToolRepo: this.opts.DatasetToolRepo ?? ""
         })
         const result = await r.text()
+        if (result.startsWith("ERROR:")) {
+            throw new Error(result)
+        }
         return JSON.parse(result) as DatasetElementMeta
     }
 
@@ -451,6 +460,9 @@ export class GPTScript {
             datasetToolRepo: this.opts.DatasetToolRepo ?? ""
         })
         const result = await r.text()
+        if (result.startsWith("ERROR:")) {
+            throw new Error(result)
+        }
         return JSON.parse(result) as Array<DatasetElementMeta>
     }
 
@@ -466,6 +478,9 @@ export class GPTScript {
             datasetToolRepo: this.opts.DatasetToolRepo ?? ""
         })
         const result = await r.text()
+        if (result.startsWith("ERROR:")) {
+            throw new Error(result)
+        }
         return JSON.parse(result) as DatasetElement
     }