types

type Plan struct {
	ID           string    `json:"id"`
	StrategyName string    `json:"strategy_name"`
	Description  string    `json:"description,omitempty"`
	Steps        []*Step   `json:"steps"`
	CreatedAt    time.Time `json:"created_at"`
	Confidence   float64   `json:"confidence,omitempty"`
}

type Query struct {
	Text    string                 `json:"text"`
	Context map[string]interface{} `json:"context,omitempty"`
	Type    string                 `json:"type,omitempty"`
}

type Result struct {
	PlanID      string                 `json:"plan_id"`
	Success     bool                   `json:"success"`
	StartTime   time.Time              `json:"start_time"`
	EndTime     time.Time              `json:"end_time"`
	Duration    time.Duration          `json:"duration"`
	Steps       []*StepResult          `json:"steps"`
	FinalOutput map[string]interface{} `json:"final_output"`
	Error       string                 `json:"error,omitempty"`
}

type Step struct {
	ID          string                 `json:"id"`
	Type        string                 `json:"type"` // Use constants: StepTypeMCPToolCall, StepTypeLLMProcessing
	ToolName    string                 `json:"tool_name"`
	Description string                 `json:"description"`
	Params      map[string]interface{} `json:"params"`
	Context     map[string]interface{} `json:"context"`              // Context data (dynamically filled during execution)
	DependsOn   []string               `json:"depends_on,omitempty"` // Dependent step IDs (future expansion)
	Timeout     time.Duration          `json:"timeout,omitempty"`    // Step timeout duration
}

type StepResult struct {
	StepID    string                 `json:"step_id"`
	StepIndex int                    `json:"step_index"`
	Success   bool                   `json:"success"`
	StartTime time.Time              `json:"start_time"`
	EndTime   time.Time              `json:"end_time"`
	Duration  time.Duration          `json:"duration"`
	Output    map[string]interface{} `json:"output"`
	Error     string                 `json:"error,omitempty"`
}

type Strategy interface {
	// Basic information
	Name() string
	Type() string
	Version() string
	Description() string

	// Planning and execution - simplified to only Execute method
	Plan(ctx context.Context, query *Query, mcpClient *mcp.MCPClientWrapper) (*Plan, error)
	Execute(ctx context.Context, plan *Plan, mcpClient *mcp.MCPClientWrapper, llmClient *llm.ChatModelWrapper, progressCallback StrategyProgressCallback) (*Result, error)

	// Configuration
	Configure(config interface{}) error
	Validate() error
}

type StrategyProgress struct {
	Type       string                 `json:"type"`        // strategy, step, error
	Stage      string                 `json:"stage"`       // Execution stage
	StepIndex  int                    `json:"step_index"`  // Step index (starting from 0, -1 indicates strategy level)
	TotalSteps int                    `json:"total_steps"` // Total number of steps
	Content    string                 `json:"content"`     // Progress description
	Data       map[string]interface{} `json:"data"`        // Additional data
	Error      error                  `json:"error"`       // Error information
	Timestamp  time.Time              `json:"timestamp"`   // Timestamp
	Complete   bool                   `json:"complete"`    // Whether completed
}

type StrategyProgressCallback func(progress *StrategyProgress)

type StrategyRegistry interface {
	Register(strategy Strategy) error
	Get(strategyName string) (Strategy, bool)
	List() []Strategy
	ListByType(strategyType string) []Strategy
}