fix: allow retry of failed runs

Author: thedadams

src/gptscript.ts

@@ -238,25 +238,25 @@ export class Run {
 	}
 
 	nextChat(input: string = ""): Run {
-		if (this.state === RunState.Finished || this.state === RunState.Error) {
-			throw (new Error("Run already finished"))
+		if (this.state !== RunState.Continue && this.state !== RunState.Creating && this.state !== RunState.Error) {
+			throw (new Error(`Run must in creating, continue or error state, not ${this.state}`))
 		}
 
 		let run = this
 		if (run.state !== RunState.Creating) {
 			run = new (this.constructor as any)(this.requestPath, this.filePath, this.content, this.opts, this.gptscriptURL)
 		}
 
-		if (this.chatState) {
-			run.chatState = this.chatState
-		} else if (this.opts.chatState) {
-			run.chatState = this.opts.chatState
+		if (this.chatState && this.state === RunState.Continue) {
+			// Only update the chat state if the previous run didn't error.
+			// The chat state on opts will be the chat state for the last successful run.
+			this.opts.chatState = this.chatState
 		}
 		run.opts.input = input
 		if (run.content !== "") {
-			run.request({content: this.content, chatState: run.chatState})
+			run.request({content: this.content, ...this.opts})
 		} else {
-			run.request({file: this.filePath, chatState: run.chatState})
+			run.request({file: this.filePath, ...this.opts})
 		}
 
 		return run

tests/gptscript.test.ts

@@ -1,4 +1,5 @@
 import * as gptscript from "../src/gptscript"
+import {RunState} from "../src/gptscript"
 import path from "path"
 import {fileURLToPath} from "url"
 
@@ -470,4 +471,23 @@ describe("gptscript module", () => {
 		expect(run.err).toContain("prompt occurred")
 		expect(promptFound).toBeFalsy()
 	})
+
+	test("retry failed run", async () => {
+		const t = {
+			instructions: "say hello"
+		}
+
+		let run = await client.evaluate(t as any)
+		await run.text()
+
+		expect(run.err).toEqual("")
+
+		// Set the run to an error state, so we can retry it.
+		run.state = RunState.Error
+		run = run.nextChat()
+
+		await run.text()
+
+		expect(run.err).toEqual("")
+	})
 })