From 1202543787fe8cef7aff6bc48cdfa1ad13700c1e Mon Sep 17 00:00:00 2001
From: Donnie Adams <donnie@acorn.io>
Date: Tue, 20 Aug 2024 21:46:30 -0400
Subject: [PATCH] fix: include proper input on call events

Signed-off-by: Donnie Adams <donnie@acorn.io>
---
 pkg/runner/runner.go     | 5 +++++
 pkg/sdkserver/monitor.go | 5 +----
 pkg/sdkserver/types.go   | 7 +++++--
 3 files changed, 11 insertions(+), 6 deletions(-)

diff --git a/pkg/runner/runner.go b/pkg/runner/runner.go
index f92b0705..93e40670 100644
--- a/pkg/runner/runner.go
+++ b/pkg/runner/runner.go
@@ -628,11 +628,16 @@ func (r *Runner) resume(callCtx engine.Context, monitor Monitor, env []string, s
 			}
 		}
 
+		var content string
+		if state.ResumeInput != nil {
+			content = *state.ResumeInput
+		}
 		monitor.Event(Event{
 			Time:        time.Now(),
 			CallContext: callCtx.GetCallContext(),
 			Type:        EventTypeCallContinue,
 			ToolResults: len(callResults),
+			Content:     content,
 		})
 
 		e := engine.Engine{
diff --git a/pkg/sdkserver/monitor.go b/pkg/sdkserver/monitor.go
index a5b0236b..bdd88c67 100644
--- a/pkg/sdkserver/monitor.go
+++ b/pkg/sdkserver/monitor.go
@@ -33,6 +33,7 @@ func (s SessionFactory) Start(ctx context.Context, prg *types.Program, env []str
 					Time: time.Now(),
 					Type: runner.EventTypeRunStart,
 				},
+				Input:   input,
 				RunID:   id,
 				Program: prg,
 			},
@@ -43,7 +44,6 @@ func (s SessionFactory) Start(ctx context.Context, prg *types.Program, env []str
 		id:     id,
 		prj:    prg,
 		env:    env,
-		input:  input,
 		events: s.events,
 	}, nil
 }
@@ -56,7 +56,6 @@ type Session struct {
 	id      string
 	prj     *types.Program
 	env     []string
-	input   string
 	events  *broadcaster.Broadcaster[event]
 	runLock sync.Mutex
 }
@@ -68,7 +67,6 @@ func (s *Session) Event(e runner.Event) {
 		Event: gserver.Event{
 			Event: e,
 			RunID: s.id,
-			Input: s.input,
 		},
 	}
 }
@@ -87,7 +85,6 @@ func (s *Session) Stop(ctx context.Context, output string, err error) {
 				Type: runner.EventTypeRunFinish,
 			},
 			RunID:  s.id,
-			Input:  s.input,
 			Output: output,
 		},
 	}
diff --git a/pkg/sdkserver/types.go b/pkg/sdkserver/types.go
index e26bbba5..2889626b 100644
--- a/pkg/sdkserver/types.go
+++ b/pkg/sdkserver/types.go
@@ -144,6 +144,7 @@ func (r *runInfo) process(e event) map[string]any {
 		r.Start = e.Time
 		r.Program = *e.Program
 		r.State = Running
+		r.Input = e.Input
 	case runner.EventTypeRunFinish:
 		r.End = e.Time
 		r.Output = e.Output
@@ -167,9 +168,11 @@ func (r *runInfo) process(e event) map[string]any {
 	call.Type = e.Type
 
 	switch e.Type {
-	case runner.EventTypeCallStart:
+	case runner.EventTypeCallStart, runner.EventTypeCallContinue:
 		call.Start = e.Time
-		call.Input = e.Content
+		if e.Content != "" {
+			call.Input = e.Content
+		}
 
 	case runner.EventTypeCallSubCalls:
 		call.setSubCalls(e.ToolSubCalls)