multiagent

type ActionType string

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

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

type ConversationContext struct {
	IsIndependentTopic bool              `json:"isIndependentTopic"`
	UserIntent         string            `json:"userIntent"`
	RelevantHistory    []*schema.Message `json:"relevantHistory"`
	KeyTopics          []string          `json:"keyTopics"`
	ContextSummary     string            `json:"contextSummary"`
	Complexity         TaskComplexity    `json:"complexity"`
	Metadata           map[string]any    `json:"metadata,omitempty"`
}

type ExecutionRecord struct {
	StepID    string            `json:"stepID"`
	Action    ActionType        `json:"action"`
	Input     []*schema.Message `json:"input"`
	Output    *schema.Message   `json:"output"`
	StartTime time.Time         `json:"startTime"`
	EndTime   time.Time         `json:"endTime"`
	Duration  time.Duration     `json:"duration"`
	Status    ExecutionStatus   `json:"status"`
	Error     string            `json:"error,omitempty"`
	Metadata  map[string]any    `json:"metadata,omitempty"`
}

type ExecutionStatus string

type Feedback struct {
	ExecutionCompleted bool     `json:"execution_completed"`
	OverallQuality     float64  `json:"overall_quality"`
	PlanNeedsUpdate    bool     `json:"plan_needs_update"`
	Issues             []string `json:"issues"`
	Suggestions        []string `json:"suggestions"`
	Confidence         float64  `json:"confidence"`
	NextActionReason   string   `json:"next_action_reason"`
}

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

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

type Host struct {
	Model        model.ToolCallingChatModel `yaml:"model" json:"model"`
	ReactAgent   *react.ReactAgent          `yaml:"react_agent" json:"react_agent"`
	SystemPrompt string                     `yaml:"system_prompt" json:"system_prompt"`
	Prompts      map[string]string          `yaml:"prompts,omitempty" json:"prompts,omitempty"`
	Thinking     ThinkingConfig             `yaml:"thinking" json:"thinking"`
	Planning     PlanningConfig             `yaml:"planning" json:"planning"`
}

type MessageHandler interface {
	OnMessage(ctx context.Context, message *schema.Message) (context.Context, error)
	OnStreamMessage(ctx context.Context, message *schema.StreamReader[*schema.Message]) (context.Context, error)
}

type MultiAgent struct {
	Runnable         compose.Runnable[[]*schema.Message, *schema.Message]
	Graph            *compose.Graph[[]*schema.Message, *schema.Message]
	GraphAddNodeOpts []compose.GraphAddNodeOpt
	AgentOptions     []base.AgentOption
	Config           *MultiAgentConfig
}

type MultiAgentConfig struct {
	Name            string            `yaml:"name" json:"name"`
	Description     string            `yaml:"description,omitempty" json:"description,omitempty"`
	Host            Host              `yaml:"host" json:"host"`
	Specialists     []*Specialist     `yaml:"specialists" json:"specialists"`
	PromptTemplates map[string]string `yaml:"prompt_templates,omitempty" json:"prompt_templates,omitempty"`
	Session         SessionConfig     `yaml:"session" json:"session"`
	MaxRounds       int               `yaml:"max_rounds" json:"max_rounds"`
}

type MultiAgentState struct {
	RoundNumber int       `json:"round_number"`
	StartTime   time.Time `json:"start_time"`

	// Conversation Context
	ConversationContext *ConversationContext `json:"conversation_context,omitempty"`
	OriginalMessages    []*schema.Message    `json:"original_messages"`

	// Task Planning
	CurrentPlan *TaskPlan   `json:"current_plan,omitempty"`
	PlanHistory []*TaskPlan `json:"plan_history,omitempty"`

	// Execution Status
	ExecutionStatus  ExecutionStatus    `json:"execution_status"`
	CurrentStep      string             `json:"current_step,omitempty"`
	ExecutionHistory []*ExecutionRecord `json:"execution_history,omitempty"`

	// Specialist Results
	SpecialistResults map[string]*StepResult `json:"specialist_results,omitempty"`

	// Collected Results
	CollectedResults []*schema.Message `json:"collected_results,omitempty"`

	// Feedback and Reflection
	FeedbackHistory []*Feedback `json:"feedback_history,omitempty"`
	ReflectionCount int         `json:"reflection_count"`

	// Execution Control
	MaxRounds      int  `json:"max_rounds"`
	ShouldContinue bool `json:"should_continue"`
	IsCompleted    bool `json:"is_completed"`

	// Final Answer
	FinalAnswer *schema.Message `json:"final_answer,omitempty"`

	// Metadata
	Metadata map[string]any `json:"metadata,omitempty"`
}

type OperationData struct {
	Type     string    `json:"type"`
	StepID   string    `json:"stepID"`
	StepData *StepData `json:"step_data,omitempty"`
	Position string    `json:"position,omitempty"`
	Reason   string    `json:"reason,omitempty"`
}

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

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

type PlanHandler interface {
	OnPlanStepCreate(ctx context.Context, plan *TaskPlan, step *PlanStep) (context.Context, error)
	OnPlanStepUpdate(ctx context.Context, plan *TaskPlan, step *PlanStep) (context.Context, error)
	OnPlanStepStatusUpdate(ctx context.Context, plan *TaskPlan, step *PlanStep) (context.Context, error)
	OnPlanStepDelete(ctx context.Context, plan *TaskPlan, step *PlanStep) (context.Context, error)
	OnPlanOpEnd(ctx context.Context, plan *TaskPlan) (context.Context, error)
}

type PlanStep struct {
	ID                 string         `json:"id"`
	Name               string         `json:"name"`
	Description        string         `json:"description"`
	AssignedSpecialist string         `json:"assignedSpecialist"`
	Priority           int            `json:"priority"`
	Status             StepStatus     `json:"status"`
	Dependencies       []string       `json:"dependencies,omitempty"`
	Parameters         map[string]any `json:"parameters,omitempty"`
	Result             *StepResult    `json:"result,omitempty"`
	Metadata           map[string]any `json:"metadata,omitempty"`
}

type PlanUpdate struct {
	UpdateReason string           `json:"update_reason"`
	Operations   []*OperationData `json:"operations"`
	PlanMetadata *struct {
		Name        string `json:"name,omitempty"`
		Description string `json:"description,omitempty"`
	} `json:"plan_metadata,omitempty"`
}

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

type PlanUpdateType string

type PlanningConfig struct {
	PlanningPrompt string `yaml:"planning_prompt" json:"planning_prompt"`
}

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

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

type SessionConfig struct {
	HistoryLength     int            `yaml:"history_length" json:"history_length"`
	ContextWindow     int            `yaml:"context_window" json:"context_window"`
	ContextProcessing map[string]any `yaml:"context_processing,omitempty" json:"context_processing,omitempty"`
	IntentAnalysis    map[string]any `yaml:"intent_analysis,omitempty" json:"intent_analysis,omitempty"`
	Persistence       map[string]any `yaml:"persistence,omitempty" json:"persistence,omitempty"`
}

type Specialist struct {
	Name         string `yaml:"name" json:"name"`
	IntendedUse  string `yaml:"intended_use" json:"intended_use"`
	ChatModel    model.BaseChatModel
	SystemPrompt string `yaml:"system_prompt" json:"system_prompt"`
	Invokable    compose.Invoke[[]*schema.Message, *schema.Message, base.AgentOption]
	Streamable   compose.Stream[[]*schema.Message, *schema.Message, base.AgentOption]
	ReactAgent   *react.ReactAgent
}

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

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

type StatePostHandler[O any] func(ctx context.Context, output O, state *MultiAgentState) (O, error)

type StatePreHandler[I any] func(ctx context.Context, input I, state *MultiAgentState) (I, error)

type StepData struct {
	ID                 string         `json:"id"`
	Name               string         `json:"name"`
	Description        string         `json:"description"`
	AssignedSpecialist string         `json:"assignedSpecialist"`
	Priority           int            `json:"priority"`
	Dependencies       []string       `json:"dependencies,omitempty"`
	Parameters         map[string]any `json:"parameters,omitempty"`
}

type StepResult struct {
	Success      bool            `json:"success"`
	Target       string          `json:"target,omitempty"`
	Output       *schema.Message `json:"output"`
	Error        string          `json:"error,omitempty"`
	Confidence   float64         `json:"confidence"`
	QualityScore float64         `json:"qualityScore"`
	Metadata     map[string]any  `json:"metadata,omitempty"`
}

type StepStatus string

type TaskComplexity string

type TaskPlan struct {
	ID          string          `json:"id"`
	Version     int             `json:"version"`
	Name        string          `json:"name"`
	Description string          `json:"description"`
	Status      ExecutionStatus `json:"status"`
	CreatedAt   time.Time       `json:"created_at"`
	UpdatedAt   time.Time       `json:"updated_at"`
	Steps       []*PlanStep     `json:"steps"`
	PlanUpdate  *PlanUpdate     `json:"plan_update,omitempty"`
	Metadata    map[string]any  `json:"metadata,omitempty"`
}

type ThinkingConfig struct {
	MaxSteps       int    `yaml:"max_steps" json:"max_steps"`
	ThinkingPrompt string `yaml:"thinking_prompt" json:"thinking_prompt"`
}