Add multi-agent config and registry runtime flow

pkg/tools/subagent_router.go

@@ -0,0 +1,121 @@
+package tools
+
+import (
+	"context"
+	"fmt"
+	"strings"
+	"time"
+)
+
+type RouterDispatchRequest struct {
+	Task           string
+	Label          string
+	Role           string
+	AgentID        string
+	ThreadID       string
+	CorrelationID  string
+	ParentRunID    string
+	OriginChannel  string
+	OriginChatID   string
+	MaxRetries     int
+	RetryBackoff   int
+	TimeoutSec     int
+	MaxTaskChars   int
+	MaxResultChars int
+}
+
+type RouterReply struct {
+	TaskID        string
+	ThreadID      string
+	CorrelationID string
+	AgentID       string
+	Status        string
+	Result        string
+}
+
+type SubagentRouter struct {
+	manager *SubagentManager
+}
+
+func NewSubagentRouter(manager *SubagentManager) *SubagentRouter {
+	return &SubagentRouter{manager: manager}
+}
+
+func (r *SubagentRouter) DispatchTask(ctx context.Context, req RouterDispatchRequest) (*SubagentTask, error) {
+	if r == nil || r.manager == nil {
+		return nil, fmt.Errorf("subagent router is not configured")
+	}
+	task, err := r.manager.SpawnTask(ctx, SubagentSpawnOptions{
+		Task:           req.Task,
+		Label:          req.Label,
+		Role:           req.Role,
+		AgentID:        req.AgentID,
+		ThreadID:       req.ThreadID,
+		CorrelationID:  req.CorrelationID,
+		ParentRunID:    req.ParentRunID,
+		OriginChannel:  req.OriginChannel,
+		OriginChatID:   req.OriginChatID,
+		MaxRetries:     req.MaxRetries,
+		RetryBackoff:   req.RetryBackoff,
+		TimeoutSec:     req.TimeoutSec,
+		MaxTaskChars:   req.MaxTaskChars,
+		MaxResultChars: req.MaxResultChars,
+	})
+	if err != nil {
+		return nil, err
+	}
+	return task, nil
+}
+
+func (r *SubagentRouter) WaitReply(ctx context.Context, taskID string, interval time.Duration) (*RouterReply, error) {
+	if r == nil || r.manager == nil {
+		return nil, fmt.Errorf("subagent router is not configured")
+	}
+	if interval <= 0 {
+		interval = 100 * time.Millisecond
+	}
+	taskID = strings.TrimSpace(taskID)
+	if taskID == "" {
+		return nil, fmt.Errorf("task id is required")
+	}
+	ticker := time.NewTicker(interval)
+	defer ticker.Stop()
+	for {
+		task, ok := r.manager.GetTask(taskID)
+		if ok && task != nil && task.Status != "running" {
+			return &RouterReply{
+				TaskID:        task.ID,
+				ThreadID:      task.ThreadID,
+				CorrelationID: task.CorrelationID,
+				AgentID:       task.AgentID,
+				Status:        task.Status,
+				Result:        strings.TrimSpace(task.Result),
+			}, nil
+		}
+		select {
+		case <-ctx.Done():
+			return nil, ctx.Err()
+		case <-ticker.C:
+		}
+	}
+}
+
+func (r *SubagentRouter) MergeResults(replies []*RouterReply) string {
+	if len(replies) == 0 {
+		return ""
+	}
+	var sb strings.Builder
+	for _, reply := range replies {
+		if reply == nil {
+			continue
+		}
+		sb.WriteString(fmt.Sprintf("[%s] agent=%s status=%s\n", reply.TaskID, reply.AgentID, reply.Status))
+		if txt := strings.TrimSpace(reply.Result); txt != "" {
+			sb.WriteString(txt)
+		} else {
+			sb.WriteString("(empty result)")
+		}
+		sb.WriteString("\n\n")
+	}
+	return strings.TrimSpace(sb.String())
+}