feat: add ChatState for runs

thedadams · thedadams · commit c12f828b5436 · 2024-05-16T11:30:10.000-04:00
Signed-off-by: Donnie Adams &lt;donnie@acorn.io&gt;
diff --git a/run.go b/run.go
@@ -80,7 +80,7 @@ func (r *Run) Close() error {
 	}
 
 	// If the command has already exited, then nothing to do.
-	if r.cmd.ProcessState == nil {
+	if r.cmd.ProcessState != nil {
 		return nil
 	}
 
@@ -99,6 +99,11 @@ func (r *Run) RawOutput() (map[string]any, error) {
 	return r.rawOutput, nil
 }
 
+// ChatState returns the current chat state of the Run.
+func (r *Run) ChatState() string {
+	return r.chatState
+}
+
 // 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) {

Original file line number	Diff line number	Diff line change
`@@ -80,7 +80,7 @@ func (r *Run) Close() error {`
`80`	`80`	`}`
`81`	`81`
`82`	`82`	`// If the command has already exited, then nothing to do.`
`83`		`- if r.cmd.ProcessState == nil {`
	`83`	`+ if r.cmd.ProcessState != nil {`
`84`	`84`	`return nil`
`85`	`85`	`}`
`86`	`86`
`@@ -99,6 +99,11 @@ func (r *Run) RawOutput() (map[string]any, error) {`
`99`	`99`	`return r.rawOutput, nil`
`100`	`100`	`}`
`101`	`101`
	`102`	`+// ChatState returns the current chat state of the Run.`
	`103`	`+func (r *Run) ChatState() string {`
	`104`	`+ return r.chatState`
	`105`	`+}`
	`106`	`+`
`102`	`107`	`// NextChat will pass input and create the next run in a chat.`
`103`	`108`	`// The new Run will be returned.`
`104`	`109`	`func (r Run) NextChat(ctx context.Context, input string) (Run, error) {`