handoff

type AgentCapability struct {
	Name        string   `json:"name"`
	Description string   `json:"description"`
	TaskTypes   []string `json:"task_types"`
	Priority    int      `json:"priority"`
}

type Handoff struct {
	ID              string         `json:"id"`
	FromAgentID     string         `json:"from_agent_id"`
	ToAgentID       string         `json:"to_agent_id"`
	ToolName        string         `json:"tool_name,omitempty"`
	ToolDescription string         `json:"tool_description,omitempty"`
	TransferMessage string         `json:"transfer_message,omitempty"`
	Task            Task           `json:"task"`
	Status          HandoffStatus  `json:"status"`
	Context         HandoffContext `json:"context"`
	Result          *HandoffResult `json:"result,omitempty"`
	CreatedAt       time.Time      `json:"created_at"`
	AcceptedAt      *time.Time     `json:"accepted_at,omitempty"`
	CompletedAt     *time.Time     `json:"completed_at,omitempty"`
	Timeout         time.Duration  `json:"timeout"`
	RetryCount      int            `json:"retry_count"`
	MaxRetries      int            `json:"max_retries"`
	// contains filtered or unexported fields
}

type HandoffAgent interface {
	ID() string
	Capabilities() []AgentCapability
	CanHandle(task Task) bool
	AcceptHandoff(ctx context.Context, handoff *Handoff) error
	ExecuteHandoff(ctx context.Context, handoff *Handoff) (*HandoffResult, error)
}

type HandoffContext struct {
	ConversationID string          `json:"conversation_id,omitempty"`
	Messages       []types.Message `json:"messages,omitempty"`
	Variables      map[string]any  `json:"variables,omitempty"`
	ParentHandoff  string          `json:"parent_handoff,omitempty"`
}

type HandoffEnabledFunc func(ctx context.Context, opts HandoffOptions) (bool, error)

type HandoffHistoryMapper func(history []types.Message) []types.Message

type HandoffHook func(ctx context.Context, handoff *Handoff) error

type HandoffInputData struct {
	InputHistory  []types.Message `json:"input_history,omitempty"`
	PreHandoff    []types.Message `json:"pre_handoff,omitempty"`
	NewMessages   []types.Message `json:"new_messages,omitempty"`
	InputMessages []types.Message `json:"input_messages,omitempty"`
	Context       HandoffContext  `json:"context"`
}

type HandoffInputFilter func(ctx context.Context, data HandoffInputData) (HandoffInputData, error)

type HandoffManager struct {
	// contains filtered or unexported fields
}

type HandoffOptions struct {
	FromAgentID             string
	ToAgentID               string
	Task                    Task
	Context                 HandoffContext
	Timeout                 time.Duration
	MaxRetries              int
	Wait                    bool
	OnHandoff               HandoffHook
	InputFilter             HandoffInputFilter
	Enabled                 *bool
	EnableFunc              HandoffEnabledFunc
	ToolNameOverride        string
	ToolDescriptionOverride string
	NestHistory             *bool
	HistoryMapper           HandoffHistoryMapper
}

type HandoffResult struct {
	Output   any    `json:"output"`
	Error    string `json:"error,omitempty"`
	Duration int64  `json:"duration_ms"`
}

type HandoffStatus string

type Task struct {
	Type        string         `json:"type"`
	Description string         `json:"description"`
	Input       any            `json:"input"`
	Priority    int            `json:"priority"`
	Metadata    map[string]any `json:"metadata,omitempty"`
}