feat: collect calls as they come for easy processing

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

Author: thedadams

run.go

@@ -8,6 +8,7 @@ import (
 	"fmt"
 	"io"
 	"log/slog"
+	"maps"
 	"net/http"
 	"os/exec"
 	"strconv"
@@ -26,10 +27,13 @@ type Run struct {
 	wait                                func()
 	basicCommand                        bool
 
-	rawOutput      map[string]any
-	output, errput string
-	events         chan Frame
-	lock           sync.Mutex
+	callsLock         sync.RWMutex
+	calls             map[string]CallFrame
+	parentCallFrameID string
+	rawOutput         map[string]any
+	output, errput    string
+	events            chan Frame
+	lock              sync.Mutex
 }
 
 // Text returns the text output of the gptscript. It blocks until the output is ready.
@@ -59,6 +63,25 @@ func (r *Run) Err() error {
 	return r.err
 }
 
+// Calls will return a flattened array of the calls for this run.
+func (r *Run) Calls() map[string]CallFrame {
+	r.callsLock.RLock()
+	defer r.callsLock.RUnlock()
+	return maps.Clone(r.calls)
+}
+
+// ParentCallFrame returns the CallFrame for the top-level or "parent" call. The boolean indicates whether there is a parent CallFrame.
+func (r *Run) ParentCallFrame() (CallFrame, bool) {
+	r.callsLock.RLock()
+	defer r.callsLock.RUnlock()
+
+	if r.parentCallFrameID == "" {
+		return CallFrame{}, false
+	}
+
+	return r.calls[r.parentCallFrameID], true
+}
+
 // ErrorOutput returns the stderr output of the gptscript.
 // Should only be called after Bytes or Text has returned an error.
 func (r *Run) ErrorOutput() string {
@@ -287,6 +310,15 @@ func (r *Run) request(ctx context.Context, payload any) (err error) {
 						return
 					}
 
+					if event.Call != nil {
+						r.callsLock.Lock()
+						r.calls[event.Call.ID] = *event.Call
+						if r.parentCallFrameID == "" && event.Call.ParentID == "" {
+							r.parentCallFrameID = event.Call.ID
+						}
+						r.callsLock.Unlock()
+					}
+
 					if r.opts.IncludeEvents {
 						r.events <- event
 					}