agent

type AgentEvent struct {
	Type    AgentEventType
	Message message.Message
	Error   error

	// When has structured output
	StructOutput *message.ToolResult

	// When summarizing
	SessionID string
	Progress  string
	Done      bool

	// FlowStepID is set when event originates from a Flow step
	FlowStepID string
}

type AgentEventType string

type AgentFactory interface {
	NewAgent(ctx context.Context, agentID string, outputSchema map[string]any, stepID string) (Service, error)
	InitPrimaryAgents(ctx context.Context, outputSchema map[string]any) ([]Service, error)
}

type MCPClient interface {
	Initialize(
		ctx context.Context,
		request mcp.InitializeRequest,
	) (*mcp.InitializeResult, error)
	ListTools(ctx context.Context, request mcp.ListToolsRequest) (*mcp.ListToolsResult, error)
	CallTool(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error)
	Close() error
}

type MCPRegistry interface {
	// GetTools return tools matching filter, if registry is not loaded it will begin loading
	LoadTools(ctx context.Context, filter *MCPRegistryFiler) <-chan tools.BaseTool
	// StartClient starts a new MCPClient, caller have to properly close when done
	StartClient(ctx context.Context, name string) (c *client.Client, err error)
}

type MCPRegistryFiler struct {
	ToolNames   []string
	ServerNames []string
}

type Service interface {
	pubsub.Subscriber[AgentEvent]
	AgentID() config.AgentName
	Model() models.Model
	Run(ctx context.Context, sessionID string, content string, attachments ...message.Attachment) (<-chan AgentEvent, error)
	Cancel(sessionID string)
	IsSessionBusy(sessionID string) bool
	IsBusy() bool
	Update(agentName config.AgentName, modelID models.ModelID) (models.Model, error)
	Summarize(ctx context.Context, sessionID string) error
}

type TaskParams struct {
	Prompt       string `json:"prompt"`
	SubagentType string `json:"subagent_type"`
	TaskID       string `json:"task_id,omitempty"`
	TaskTitle    string `json:"task_title,omitempty"`
}

type TaskResponseMetadata struct {
	TaskID         string `json:"task_id"`
	SubagentType   string `json:"subagent_type"`
	SubagentName   string `json:"subagent_name"`
	IsResumed      bool   `json:"is_resumed"`
	IsStructOutput bool   `json:"is_struct_output"`
}