swarm

type BudgetState struct {
	CallsAfterSoft int
	TotalCalls     int
	Remaining      float64
	Elapsed        time.Duration
}

type BudgetStatus int

type BudgetStrategy interface {
	// InitialBudget returns the total budget for a task.
	InitialBudget() float64

	// BeforeToolCall returns the budget cost to consume before executing a tool.
	BeforeToolCall(ctx ToolCallContext) float64

	// AfterToolCall returns additional budget cost based on the tool execution result.
	AfterToolCall(ctx ToolCallContext) float64

	// ShouldExit determines whether tool execution must be forcibly stopped.
	// Called after the soft limit has been hit to decide if the task should be terminated.
	ShouldExit(state BudgetState) bool
}

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

type DefaultBudgetStrategy struct{}

type Option func(*SwarmChat)

type PlanResult struct {
	Message string     `json:"message"`
	Tasks   []TaskPlan `json:"tasks"`
}

type ReplanResult struct {
	Message  string     `json:"message,omitempty"`
	Tasks    []TaskPlan `json:"tasks"`
	Question string     `json:"question,omitempty"`
}

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

type TaskPlan struct {
	ID                 string   `json:"id"`
	Title              string   `json:"title"`
	Description        string   `json:"description"`
	AcceptanceCriteria string   `json:"acceptance_criteria"`
	Tools              []string `json:"tools"`
	SubAgents          []string `json:"sub_agents"`
}

type TaskResult struct {
	TaskID         string
	Title          string
	Result         string
	Error          error
	BudgetExceeded bool // true if the task was terminated due to budget exhaustion
}

type ToolCallContext struct {
	ToolName    string
	Elapsed     time.Duration
	PrevElapsed time.Duration // Elapsed time at the previous tool call (for time cost delta)
	CallCount   int
	Result      map[string]any
	Error       error
}

type ToolRecord struct {
	Name      string
	Cost      float64
	Timestamp time.Time
}