fix: allow retrying of failed runs

Signed-off-by: Donnie Adams <[email protected]>

Author: thedadams

client.go

@@ -119,7 +119,6 @@ func (c *client) Evaluate(ctx context.Context, opts Options, tools ...fmt.String
 		state:       Creating,
 		opts:        opts,
 		content:     concatTools(tools),
-		chatState:   opts.ChatState,
 	}).NextChat(ctx, opts.Input)
 }
 
@@ -130,7 +129,6 @@ func (c *client) Run(ctx context.Context, toolPath string, opts Options) (*Run,
 		state:       Creating,
 		opts:        opts,
 		toolPath:    toolPath,
-		chatState:   opts.ChatState,
 	}).NextChat(ctx, opts.Input)
 }
 

run.go

@@ -106,22 +106,24 @@ func (r *Run) ChatState() string {
 // NextChat will pass input and create the next run in a chat.
 // The new Run will be returned.
 func (r *Run) NextChat(ctx context.Context, input string) (*Run, error) {
-	if r.state != Creating && r.state != Continue {
-		return nil, fmt.Errorf("run must be in creating or continue state not %q", r.state)
+	if r.state != Creating && r.state != Continue && r.state != Error {
+		return nil, fmt.Errorf("run must be in creating, continue, or error state not %q", r.state)
 	}
 
 	run := &Run{
 		url:         r.url,
 		requestPath: r.requestPath,
 		state:       Creating,
-		chatState:   r.chatState,
 		toolPath:    r.toolPath,
 		content:     r.content,
 		opts:        r.opts,
 	}
+
 	run.opts.Input = input
-	if run.chatState != "" {
-		run.opts.ChatState = run.chatState
+	if r.chatState != "" && r.state != Error {
+		// If the previous run errored, then don't update the chat state.
+		// opts.ChatState will be the last chat state where an error did not occur.
+		run.opts.ChatState = r.chatState
 	}
 
 	var payload any
@@ -287,6 +289,11 @@ func (r *Run) request(ctx context.Context, payload any) (err error) {
 						return
 					}
 
+					if event.Run != nil && event.Run.Type == EventTypeRunFinish && event.Run.Error != "" {
+						r.state = Error
+						r.err = fmt.Errorf(event.Run.Error)
+					}
+
 					if r.opts.IncludeEvents {
 						r.events <- event
 					}

run_test.go

@@ -0,0 +1,34 @@
+package gptscript
+
+import (
+	"context"
+	"testing"
+)
+
+func TestRestartingErrorRun(t *testing.T) {
+	tool := &ToolDef{
+		Instructions: "Say hello",
+	}
+	run, err := c.Evaluate(context.Background(), Options{}, tool)
+	if err != nil {
+		t.Errorf("Error executing tool: %v", err)
+	}
+
+	// Wait for the run to complete
+	_, err = run.Text()
+	if err != nil {
+		t.Fatalf("run returned error: %v", err)
+	}
+
+	// Set the run to error, so we can test the retry.
+	run.state = Error
+	run, err = run.NextChat(context.Background(), "")
+	if err != nil {
+		t.Errorf("Error executing next run: %v", err)
+	}
+
+	_, err = run.Text()
+	if err != nil {
+		t.Errorf("executing run with input of 0 should not fail: %v", err)
+	}
+}