multiagent

type AgentMessage struct {
	ID           string                 `json:"id"`
	From         string                 `json:"from"`
	To           string                 `json:"to"`
	Topic        string                 `json:"topic"`
	Type         MessageType            `json:"type"`
	Payload      interface{}            `json:"payload"`
	Metadata     map[string]string      `json:"metadata"`
	Timestamp    time.Time              `json:"timestamp"`
	TraceContext propagation.MapCarrier `json:"trace_context,omitempty"` // 追踪上下文
}

type AgentSession struct {
	ID           string
	Participants []string
	Messages     []*AgentMessage
	State        map[string]interface{}
	CreatedAt    string
	UpdatedAt    string
}

type Assignment struct {
	AgentID   string      `json:"agent_id"`
	Role      Role        `json:"role"`
	Subtask   interface{} `json:"subtask"`
	Status    TaskStatus  `json:"status"`
	Result    interface{} `json:"result,omitempty"`
	StartTime time.Time   `json:"start_time,omitempty"`
	EndTime   time.Time   `json:"end_time,omitempty"`
}

type BaseCollaborativeAgent struct {
	*core.BaseAgent
	// contains filtered or unexported fields
}

type ChannelStore interface {
	// GetOrCreateChannel 获取或创建指定 agent 的消息 channel
	GetOrCreateChannel(agentID string) chan *AgentMessage

	// GetChannel 获取指定 agent 的消息 channel，如果不存在返回 nil
	GetChannel(agentID string) chan *AgentMessage

	// ListChannels 列出所有 channel 的 agent ID
	ListChannels() []string

	// Subscribe 订阅主题，返回接收消息的 channel
	Subscribe(topic string) <-chan *AgentMessage

	// Unsubscribe 取消订阅主题
	Unsubscribe(topic string)

	// GetSubscribers 获取指定主题的所有订阅者 channel
	GetSubscribers(topic string) []chan *AgentMessage

	// Close 关闭存储（可选的清理操作）
	Close() error
}

type CollaborationType string

type CollaborativeAgent interface {
	core.Agent

	// GetRole returns the agent's role
	GetRole() Role

	// SetRole sets the agent's role
	SetRole(role Role)

	// ReceiveMessage handles incoming messages
	ReceiveMessage(ctx context.Context, message Message) error

	// SendMessage sends a message to another agent
	SendMessage(ctx context.Context, message Message) error

	// Collaborate participates in a collaborative task
	Collaborate(ctx context.Context, task *CollaborativeTask) (*Assignment, error)

	// Vote participates in consensus decision making
	Vote(ctx context.Context, proposal interface{}) (bool, error)
}

type CollaborativeTask struct {
	ID          string                 `json:"id"`
	Name        string                 `json:"name"`
	Description string                 `json:"description"`
	Type        CollaborationType      `json:"type"`
	Input       interface{}            `json:"input"`
	Output      interface{}            `json:"output,omitempty"`
	Status      TaskStatus             `json:"status"`
	Assignments map[string]Assignment  `json:"assignments"`
	Results     map[string]interface{} `json:"results"`
	StartTime   time.Time              `json:"start_time"`
	EndTime     time.Time              `json:"end_time,omitempty"`
	Metadata    map[string]interface{} `json:"metadata,omitempty"`
}

type Communicator interface {
	// Send 发送消息
	Send(ctx context.Context, to string, message *AgentMessage) error

	// Receive 接收消息
	Receive(ctx context.Context) (*AgentMessage, error)

	// Broadcast 广播消息
	Broadcast(ctx context.Context, message *AgentMessage) error

	// Subscribe 订阅主题
	Subscribe(ctx context.Context, topic string) (<-chan *AgentMessage, error)

	// Unsubscribe 取消订阅
	Unsubscribe(ctx context.Context, topic string) error

	// Close 关闭
	Close() error
}

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

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

type Message struct {
	ID        string                 `json:"id"`
	From      string                 `json:"from"`
	To        string                 `json:"to"`
	Type      MessageType            `json:"type"`
	Content   interface{}            `json:"content"`
	Priority  int                    `json:"priority"`
	Timestamp time.Time              `json:"timestamp"`
	ReplyTo   string                 `json:"reply_to,omitempty"`
	Metadata  map[string]interface{} `json:"metadata,omitempty"`
}

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

type MessageType string

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

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

type NATSConfig struct {
	URL         string
	ClusterID   string
	Credentials string
	TLS         *tls.Config
}

type NegotiatingAgent struct {
	*BaseCollaborativeAgent
	// contains filtered or unexported fields
}

type NegotiationRound struct {
	Round    int                    `json:"round"`
	Proposal interface{}            `json:"proposal"`
	Offers   map[string]interface{} `json:"offers"`
	Accepted bool                   `json:"accepted"`
}

type Role string

type RouteHandler func(ctx context.Context, message *AgentMessage) (*AgentMessage, error)

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

type SpecializedAgent struct {
	*BaseCollaborativeAgent
	// contains filtered or unexported fields
}

type SystemOption func(*MultiAgentSystem)

type TaskStatus string

type Team struct {
	ID           string                 `json:"id"`
	Name         string                 `json:"name"`
	Leader       string                 `json:"leader"`
	Members      []string               `json:"members"`
	Purpose      string                 `json:"purpose"`
	Capabilities []string               `json:"capabilities"`
	Metadata     map[string]interface{} `json:"metadata,omitempty"`
}