Method GetRemainingResults

agent/tools_executor.go:196–226

(ctx context.Context)

func (e *StreamingToolExecutor) WaitForActivity(ctx context.Context) {
	if !e.HasPendingWork() {
		return
	}
	if !e.hasRunningWork() {
		e.maybeDrain()
		if !e.hasRunningWork() {
			return
		}
	}
	select {
	case <-ctx.Done():
	case <-e.activity:
	case <-e.cancelCtx.Done():
	}
}

func (e *StreamingToolExecutor) GetRemainingResults(ctx context.Context) []map[string]any {
	for e.HasPendingWork() {
		if ctx != nil && ctx.Err() != nil {
			break
		}
		if e.Cancelled() && !e.hasRunningWork() {
			break
		}
		e.WaitForActivity(ctx)
		e.maybeDrain()
		if !e.hasRunningWork() {
			break
		}
	}
	e.mu.Lock()
	defer e.mu.Unlock()
	var results []map[string]any
	for _, tid := range e.toolOrder {
		slot := e.slots[tid]