conversation

type Branch struct {
	ID          string               `json:"id"`
	Name        string               `json:"name"`
	Description string               `json:"description,omitempty"`
	States      []*ConversationState `json:"states"`
	CreatedAt   time.Time            `json:"created_at"`
	UpdatedAt   time.Time            `json:"updated_at"`
	IsActive    bool                 `json:"is_active"`
}

type ChatMessage struct {
	ID        string         `json:"id"`
	Role      string         `json:"role"` // user, assistant, system
	SenderID  string         `json:"sender_id"`
	Content   string         `json:"content"`
	Timestamp time.Time      `json:"timestamp"`
	Metadata  map[string]any `json:"metadata,omitempty"`
}

type Conversation struct {
	ID       string
	Mode     ConversationMode
	Agents   []ConversationAgent
	Messages []ChatMessage
	Config   ConversationConfig
	Selector SpeakerSelector
	// contains filtered or unexported fields
}

type ConversationAgent interface {
	ID() string
	Name() string
	SystemPrompt() string
	Reply(ctx context.Context, messages []ChatMessage) (*ChatMessage, error)
	ShouldTerminate(messages []ChatMessage) bool
}

type ConversationConfig struct {
	MaxRounds        int           `json:"max_rounds"`
	MaxMessages      int           `json:"max_messages"`
	Timeout          time.Duration `json:"timeout"`
	AllowInterrupts  bool          `json:"allow_interrupts"`
	TerminationWords []string      `json:"termination_words"`
}

type ConversationMode string

type ConversationResult struct {
	ConversationID    string        `json:"conversation_id"`
	Messages          []ChatMessage `json:"messages"`
	TotalRounds       int           `json:"total_rounds"`
	StartTime         time.Time     `json:"start_time"`
	EndTime           time.Time     `json:"end_time"`
	TerminationReason string        `json:"termination_reason"`
}

type ConversationState struct {
	ID        string         `json:"id"`
	ParentID  string         `json:"parent_id,omitempty"`
	Messages  []llm.Message  `json:"messages"`
	Metadata  map[string]any `json:"metadata,omitempty"`
	CreatedAt time.Time      `json:"created_at"`
	Label     string         `json:"label,omitempty"`
}

type ConversationTree struct {
	ID           string             `json:"id"`
	RootState    *ConversationState `json:"root_state"`
	Branches     map[string]*Branch `json:"branches"`
	ActiveBranch string             `json:"active_branch"`
	// contains filtered or unexported fields
}

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

type LLMClient interface {
	Complete(ctx context.Context, prompt string) (string, error)
}

type LLMSelector struct {
	LLM LLMClient
}

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

type SpeakerSelector interface {
	SelectNext(ctx context.Context, agents []ConversationAgent, messages []ChatMessage) (ConversationAgent, error)
}