agent

type Agent interface {
	// GetID returns the unique identifier of the agent
	GetID() string
	// GetName returns the display name of the agent
	GetName() string
	// GetType returns the agent type (e.g., "claude", "gemini")
	GetType() string
	// GetModel returns the specific model being used
	GetModel() string
	// GetRateLimit returns the rate limit in requests per second (0 = unlimited)
	GetRateLimit() float64
	// GetRateLimitBurst returns the burst size for rate limiting
	GetRateLimitBurst() int
	// Initialize configures the agent with the provided configuration
	Initialize(config AgentConfig) error
	// SendMessage sends a message to the agent and returns the response
	SendMessage(ctx context.Context, messages []Message) (string, error)
	// StreamMessage sends a message and streams the response to the writer
	StreamMessage(ctx context.Context, messages []Message, writer io.Writer) error
	// Announce returns the agent's join announcement message
	Announce() string
	// IsAvailable checks if the agent's CLI tool is available
	IsAvailable() bool
	// HealthCheck performs a comprehensive health check of the agent
	HealthCheck(ctx context.Context) error
}

type AgentConfig struct {
	// ID is the unique identifier for this agent instance
	ID string `yaml:"id"`
	// Type is the agent type (e.g., "claude", "gemini", "copilot")
	Type string `yaml:"type"`
	// Name is the display name for the agent
	Name string `yaml:"name"`
	// Prompt is the system prompt that defines the agent's behavior
	Prompt string `yaml:"prompt"`
	// Announcement is the message shown when the agent joins
	Announcement string `yaml:"announcement"`
	// Model is the specific model to use (e.g., "claude-sonnet-4.5")
	Model string `yaml:"model"`
	// Temperature controls randomness in responses (0.0 to 1.0)
	Temperature float64 `yaml:"temperature"`
	// MaxTokens limits the length of generated responses
	MaxTokens int `yaml:"max_tokens"`
	// RateLimit is the maximum requests per second for this agent (0 = unlimited)
	RateLimit float64 `yaml:"rate_limit"`
	// RateLimitBurst is the maximum burst size for rate limiting (default: 1)
	RateLimitBurst int `yaml:"rate_limit_burst"`
	// CustomSettings allows agent-specific configuration options
	CustomSettings map[string]interface{} `yaml:"custom_settings"`
}

type BaseAgent struct {
	// ID is the unique identifier for this agent instance
	ID string
	// Name is the display name
	Name string
	// Type is the agent type
	Type string
	// Config stores the full agent configuration
	Config AgentConfig
	// Announcement is the custom join message
	Announcement string
}

type Factory func() Agent

type Message struct {
	// AgentID is the unique identifier of the agent or entity that sent the message
	AgentID string
	// AgentName is the display name of the agent
	AgentName string
	// AgentType is the type of agent (e.g., "claude", "gemini", "qoder")
	AgentType string
	// Content is the actual message text
	Content string
	// Timestamp is the Unix timestamp when the message was created
	Timestamp int64
	// Role indicates the message type: "agent", "user", or "system"
	Role string
	// Metrics contains optional performance and cost metrics for agent responses
	Metrics *ResponseMetrics
}

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

type ResponseMetrics struct {
	// Duration is how long the agent took to generate the response
	Duration time.Duration
	// InputTokens is the number of tokens in the input (prompt + conversation history)
	InputTokens int
	// OutputTokens is the number of tokens in the agent's response
	OutputTokens int
	// TotalTokens is InputTokens + OutputTokens
	TotalTokens int
	// Model is the specific model used by the agent
	Model string
	// Cost is the estimated monetary cost of the API call in USD
	Cost float64
}