agents

type Agent interface {
	// Name returns the agent's name
	Name() string

	// Execute processes input and returns output
	Execute(ctx context.Context, input interface{}) (interface{}, error)
}

type AgentConfig struct {
	LLM           LLMConfig           `yaml:"llm"`
	SchemaAgent   SchemaAgentConfig   `yaml:"schema_agent"`
	ChunkingAgent ChunkingAgentConfig `yaml:"chunking_agent"`
	QueryAgent    QueryAgentConfig    `yaml:"query_agent"`
	PlanningAgent PlanningAgentConfig `yaml:"planning_agent"`
	ReportAgent   ReportAgentConfig   `yaml:"report_agent"`
	EvalAgent     EvalAgentConfig     `yaml:"eval_agent"`
	Output        AgentOutputConfig   `yaml:"output"`
	Cache         CacheConfig         `yaml:"cache"`
	Performance   PerformanceConfig   `yaml:"performance"`
	Features      FeaturesConfig      `yaml:"features"`
}

type AgentOutputConfig struct {
	DefaultFormat  string `yaml:"default_format"`
	Verbosity      string `yaml:"verbosity"`
	ShowConfidence bool   `yaml:"show_confidence"`
	ShowReasoning  bool   `yaml:"show_reasoning"`
	ShowWarnings   bool   `yaml:"show_warnings"`
}

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

type BashCommand struct {
	Command     string            `json:"command"`
	Args        []string          `json:"args"`
	WorkingDir  string            `json:"working_dir,omitempty"`
	Timeout     time.Duration     `json:"timeout,omitempty"`
	Environment map[string]string `json:"environment,omitempty"`
}

type BashResult struct {
	Success  bool          `json:"success"`
	Stdout   string        `json:"stdout"`
	Stderr   string        `json:"stderr"`
	ExitCode int           `json:"exit_code"`
	Duration time.Duration `json:"duration"`
}

type CacheConfig struct {
	Enabled    bool `yaml:"enabled"`
	TTLSeconds int  `yaml:"ttl_seconds"`
	MaxSizeMB  int  `yaml:"max_size_mb"`
}

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

type ChunkingAgentConfig struct {
	Enabled               bool    `yaml:"enabled"`
	Model                 string  `yaml:"model"`
	Temperature           float64 `yaml:"temperature"`
	MaxTokens             int     `yaml:"max_tokens"`
	DefaultChunkSize      int     `yaml:"default_chunk_size"`
	MinChunkSize          int     `yaml:"min_chunk_size"`
	MaxChunkSize          int     `yaml:"max_chunk_size"`
	DefaultOverlapPercent int     `yaml:"default_overlap_percent"`
	MinOverlapPercent     int     `yaml:"min_overlap_percent"`
	MaxOverlapPercent     int     `yaml:"max_overlap_percent"`
	TechnicalChunkSize    int     `yaml:"technical_chunk_size"`
	NarrativeChunkSize    int     `yaml:"narrative_chunk_size"`
	MixedChunkSize        int     `yaml:"mixed_chunk_size"`
}

type ChunkingAnalysisInput struct {
	SampleFiles    []string // Paths to sample documents
	CollectionName string   // Target collection name
	VDBType        string   // Target VDB type
	Requirements   string   // User requirements (optional)
	MaxSamples     int      // Max files to analyze (default: 50)
}

type ChunkingAnalysisOutput struct {
	Recommendation ChunkingRecommendation `json:"chunking_advice"`
	Confidence     float64                `json:"confidence"`
	Warnings       []string               `json:"warnings,omitempty"`
	Metrics        ChunkingMetrics        `json:"metrics,omitempty"`
}

type ChunkingMetrics struct {
	AvgContentLength int      `json:"avg_content_length" yaml:"avg_content_length"`
	AvgParagraphs    int      `json:"avg_paragraphs" yaml:"avg_paragraphs"`
	AvgSections      int      `json:"avg_sections" yaml:"avg_sections"`
	AvgParagraphLen  int      `json:"avg_paragraph_len" yaml:"avg_paragraph_len"`
	ContentDensity   string   `json:"content_density" yaml:"content_density"`
	FileTypes        []string `json:"file_types" yaml:"file_types"`
}

type ChunkingRecommendation struct {
	RecommendedSize int      `json:"recommended_size" yaml:"recommended_size"`
	MinSize         int      `json:"min_size" yaml:"min_size"`
	MaxSize         int      `json:"max_size" yaml:"max_size"`
	OverlapSize     int      `json:"overlap_size" yaml:"overlap_size"`
	Reasoning       string   `json:"reasoning" yaml:"reasoning"`
	DocumentType    string   `json:"document_type" yaml:"document_type"`
	Considerations  []string `json:"considerations,omitempty" yaml:"considerations,omitempty"`
}

type CommandReport struct {
	Type     string        `json:"type"` // "bash", "weave"
	Command  string        `json:"command"`
	Success  bool          `json:"success"`
	Output   string        `json:"output"`
	Error    string        `json:"error,omitempty"`
	Duration time.Duration `json:"duration"`
}

type DocumentSample struct {
	Path    string
	Type    string
	Size    int64
	Fields  map[string]interface{}
	Preview string // First 1000 chars
}

type DocumentStructure struct {
	Samples        []DocumentSample
	FileTypes      []string
	CommonFields   map[string]int    // field -> frequency
	FieldTypes     map[string]string // field -> inferred type
	ContentLengths []int
	Languages      []string
	// Chunking analysis fields
	ParagraphCounts []int  // paragraph count per document
	SectionCounts   []int  // section count per document
	AvgParagraphLen int    // average paragraph length
	ContentDensity  string // "sparse", "medium", "dense"
}

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

type EvalAgentConfig struct {
	Enabled     bool    `yaml:"enabled"`
	Temperature float64 `yaml:"temperature"`
	MaxTokens   int     `yaml:"max_tokens"`
}

type EvaluationMetrics struct {
	QueryID          string        `json:"query_id"`
	Success          bool          `json:"success"`
	IntentMatched    bool          `json:"intent_matched"`
	LLMInvocations   int           `json:"llm_invocations"`
	TotalTokens      int           `json:"total_tokens"`
	PromptTokens     int           `json:"prompt_tokens"`
	CompletionTokens int           `json:"completion_tokens"`
	TotalCost        float64       `json:"total_cost"`
	Latency          time.Duration `json:"latency"`
	ErrorRate        float64       `json:"error_rate"`
	UserSatisfaction *float64      `json:"user_satisfaction,omitempty"`
}

type ExecutionPlan struct {
	Steps       []ExecutionStep `json:"steps"`
	Summary     string          `json:"summary"`
	Warnings    []string        `json:"warnings,omitempty"`
	Estimations struct {
		Duration string `json:"duration"`
		Risk     string `json:"risk"` // "low", "medium", "high"
	} `json:"estimations"`
}

type ExecutionStep struct {
	Type        string                 `json:"type"` // "bash", "weave", "confirm"
	Command     string                 `json:"command"`
	Description string                 `json:"description"`
	Args        []string               `json:"args,omitempty"`
	Params      map[string]interface{} `json:"params,omitempty"`
	DependsOn   []int                  `json:"depends_on,omitempty"` // Step indices
	Optional    bool                   `json:"optional"`
	Destructive bool                   `json:"destructive"`
}

type FeaturesConfig struct {
	Experimental bool `yaml:"experimental"`
	Debug        bool `yaml:"debug"`
	Telemetry    bool `yaml:"telemetry"`
}

type FieldConfig struct {
	Name        string `yaml:"name" json:"name"`
	Type        string `yaml:"type" json:"type"`
	Description string `yaml:"description,omitempty" json:"description,omitempty"`
	Indexed     bool   `yaml:"indexed" json:"indexed"`
	Filterable  bool   `yaml:"filterable" json:"filterable"`
	Required    bool   `yaml:"required" json:"required"`
}

type FieldSuggestion struct {
	Name        string        `json:"name"`
	Type        string        `json:"type"`
	Indexed     bool          `json:"indexed"`
	Filterable  bool          `json:"filterable"`
	Required    bool          `json:"required"`
	Examples    []interface{} `json:"examples"`
	Frequency   float64       `json:"frequency"`
	Cardinality int           `json:"cardinality"`
	Reasoning   string        `json:"reasoning"`
}

type IndexConfig struct {
	Name   string   `yaml:"name" json:"name"`
	Fields []string `yaml:"fields" json:"fields"`
	Type   string   `yaml:"type" json:"type"` // "vector", "bm25", "composite"
}

type LLMConfig struct {
	Provider           string  `yaml:"provider"`
	APIKeyEnv          string  `yaml:"api_key_env"`
	DefaultModel       string  `yaml:"default_model"`
	DefaultTemperature float64 `yaml:"default_temperature"`
	DefaultMaxTokens   int     `yaml:"default_max_tokens"`
}

type OperationReport struct {
	QueryIntent     string          `json:"query_intent"`
	ExecutedSteps   int             `json:"executed_steps"`
	SuccessfulSteps int             `json:"successful_steps"`
	FailedSteps     int             `json:"failed_steps"`
	StartTime       time.Time       `json:"start_time"`
	EndTime         time.Time       `json:"end_time"`
	Duration        time.Duration   `json:"duration"`
	Commands        []CommandReport `json:"commands"`
	Summary         string          `json:"summary"`
	Recommendations []string        `json:"recommendations,omitempty"`
	NextSteps       []string        `json:"next_steps,omitempty"`
}

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

type OutputConfig struct {
	NoColor      bool
	Verbose      bool
	Quiet        bool
	OutputFormat string // "text", "json", "yaml"
}

type PerformanceConfig struct {
	LLMTimeout         int `yaml:"llm_timeout"`
	MaxConcurrentFiles int `yaml:"max_concurrent_files"`
	MaxRetries         int `yaml:"max_retries"`
	RetryDelayMS       int `yaml:"retry_delay_ms"`
}

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

type PlanningAgentConfig struct {
	Enabled     bool    `yaml:"enabled"`
	Model       string  `yaml:"model"`
	Temperature float64 `yaml:"temperature"`
	MaxTokens   int     `yaml:"max_tokens"`
}

type PlanningAgentInput struct {
	FixedQuery string `json:"fixed_query"`
	Intent     string `json:"intent"`
}

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

type QueryAgentConfig struct {
	Enabled     bool    `yaml:"enabled"`
	Model       string  `yaml:"model"`
	Temperature float64 `yaml:"temperature"`
	MaxTokens   int     `yaml:"max_tokens"`
}

type QueryAgentInput struct {
	Query string `json:"query"`
}

type QueryAgentOutput struct {
	IsWeaveQuery bool    `json:"is_weave_query"`
	FixedQuery   string  `json:"fixed_query"`
	Intent       string  `json:"intent"` // "list", "create", "query", "delete", etc.
	Confidence   float64 `json:"confidence"`
	Reason       string  `json:"reason"`
}

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

type ReportAgentConfig struct {
	Enabled     bool    `yaml:"enabled"`
	Temperature float64 `yaml:"temperature"`
	MaxTokens   int     `yaml:"max_tokens"`
}

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

type SchemaAgentConfig struct {
	Enabled                 bool    `yaml:"enabled"`
	Model                   string  `yaml:"model"`
	Temperature             float64 `yaml:"temperature"`
	MaxTokens               int     `yaml:"max_tokens"`
	MaxSamples              int     `yaml:"max_samples"`
	DefaultVectorDimensions int     `yaml:"default_vector_dimensions"`
	MinConfidence           float64 `yaml:"min_confidence"`
	WarnBelowConfidence     float64 `yaml:"warn_below_confidence"`
}

type SchemaAnalysisInput struct {
	SampleFiles    []string // Paths to sample documents
	CollectionName string   // Target collection name
	Requirements   string   // User requirements (optional)
	VDBType        string   // Target VDB type
	MaxSamples     int      // Max files to analyze (default: 50)
}

type SchemaAnalysisOutput struct {
	Schema         SchemaConfig            `json:"schema"`
	Reasoning      string                  `json:"reasoning"`
	FieldAnalysis  []FieldSuggestion       `json:"field_analysis"`
	Confidence     float64                 `json:"confidence"`
	Warnings       []string                `json:"warnings"`
	Alternatives   []SchemaConfig          `json:"alternatives,omitempty"`
	ChunkingAdvice *ChunkingRecommendation `json:"chunking_advice,omitempty"`
}

type SchemaConfig struct {
	CollectionName   string                 `yaml:"collection_name" json:"collection_name"`
	VectorDimensions int                    `yaml:"vector_dimensions" json:"vector_dimensions"`
	SimilarityMetric string                 `yaml:"similarity_metric" json:"similarity_metric"`
	Fields           []FieldConfig          `yaml:"fields" json:"fields"`
	Indexes          []IndexConfig          `yaml:"indexes" json:"indexes"`
	Metadata         map[string]interface{} `yaml:"metadata,omitempty" json:"metadata,omitempty"`
}

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

type WeaveAgentCommand struct {
	Tool      string                 `json:"tool"`
	Arguments map[string]interface{} `json:"arguments"`
	Timeout   time.Duration          `json:"timeout,omitempty"`
}

type WeaveAgentResult struct {
	Success  bool          `json:"success"`
	Output   interface{}   `json:"output"`
	Error    string        `json:"error,omitempty"`
	Duration time.Duration `json:"duration"`
	Retries  int           `json:"retries"`
}