agent

type Agent interface {
	GetID() string
	GetName() string
	GetType() string
	GetModel() string
	Initialize(config AgentConfig) error
	SendMessage(ctx context.Context, messages []Message) (string, error)
	StreamMessage(ctx context.Context, messages []Message, writer io.Writer) error
	Announce() string
	IsAvailable() bool
	HealthCheck(ctx context.Context) error
}

type AgentConfig struct {
	ID             string                 `yaml:"id"`
	Type           string                 `yaml:"type"`
	Name           string                 `yaml:"name"`
	Prompt         string                 `yaml:"prompt"`
	Announcement   string                 `yaml:"announcement"`
	Model          string                 `yaml:"model"`
	Temperature    float64                `yaml:"temperature"`
	MaxTokens      int                    `yaml:"max_tokens"`
	CustomSettings map[string]interface{} `yaml:"custom_settings"`
}

type BaseAgent struct {
	ID           string
	Name         string
	Type         string
	Config       AgentConfig
	Announcement string
}

type Factory func() Agent

type Message struct {
	AgentID   string
	AgentName string
	Content   string
	Timestamp int64
	Role      string // "agent" or "system"
	Metrics   *ResponseMetrics
}

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

type ResponseMetrics struct {
	Duration     time.Duration
	InputTokens  int
	OutputTokens int
	TotalTokens  int
	Model        string
	Cost         float64
}