ai

type Agent interface {
	// GetType returns the agent type
	GetType() AgentType

	// GetConfig returns the agent configuration
	GetConfig() AgentConfig

	// ListModels returns available models for this agent
	ListModels(ctx context.Context) ([]Model, error)

	// ExecutePrompt processes a single prompt
	ExecutePrompt(ctx context.Context, request PromptRequest) (*PromptResponse, error)

	// ExecuteBatch processes multiple prompts
	ExecuteBatch(ctx context.Context, requests []PromptRequest) (map[string]*PromptResponse, error)

	GetCosts() Costs

	// Close cleans up resources
	Close() error
}

type AgentConfig struct {
	Type            AgentType     `json:"type"`
	Model           string        `json:"model"`
	CacheDBPath     string        `json:"cache_db_path,omitempty"`
	ProjectName     string        `json:"project_name,omitempty"`
	SessionID       string        `json:"session_id,omitempty"`
	CacheTTL        time.Duration `json:"cache_ttl,omitempty"`
	Temperature     float64       `json:"temperature,omitempty"`
	MaxTokens       int           `json:"max_tokens"`
	MaxConcurrent   int           `json:"max_concurrent"`
	Debug           bool          `json:"debug"`
	Verbose         bool          `json:"verbose"`
	StrictMCPConfig bool          `json:"strict_mcp_config"`
	NoCache         bool          `json:"no_cache,omitempty"`
}

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

type AgentType string

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

type AiderResponse struct {
	FilesChanged []string `json:"files_changed,omitempty"`
	Result       string   `json:"result"`
	Error        string   `json:"error,omitempty"`
}

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

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

type ClaudeOptions struct {
	Model           string
	MaxTokens       int
	Debug           bool
	StrictMCPConfig bool
	OutputFormat    string
	MaxConcurrent   int
}

type ClaudeResponse struct {
	Type              string   `json:"type"`
	Subtype           string   `json:"subtype,omitempty"`
	Result            string   `json:"result"`
	IsError           bool     `json:"is_error"`
	DurationMs        int      `json:"duration_ms"`
	DurationAPIMs     int      `json:"duration_api_ms,omitempty"`
	NumTurns          int      `json:"num_turns,omitempty"`
	SessionID         string   `json:"session_id,omitempty"`
	TotalCostUSD      float64  `json:"total_cost_usd"`
	UUID              string   `json:"uuid,omitempty"`
	PermissionDenials []string `json:"permission_denials,omitempty"`
	Usage             struct {
		InputTokens              int `json:"input_tokens"`
		CacheCreationInputTokens int `json:"cache_creation_input_tokens"`
		CacheReadInputTokens     int `json:"cache_read_input_tokens"`
		OutputTokens             int `json:"output_tokens"`
		ServerToolUse            struct {
			WebSearchRequests int `json:"web_search_requests"`
		} `json:"server_tool_use,omitempty"`
		ServiceTier   string `json:"service_tier,omitempty"`
		CacheCreation struct {
			Ephemeral1hInputTokens int `json:"ephemeral_1h_input_tokens"`
			Ephemeral5mInputTokens int `json:"ephemeral_5m_input_tokens"`
		} `json:"cache_creation,omitempty"`
	} `json:"usage"`
	ModelUsage map[string]ModelUsageDetail `json:"modelUsage,omitempty"`
}

type Cost struct {
	Model        string  `json:"model,omitempty"`
	InputTokens  int     `json:"input_tokens"`
	OutputTokens int     `json:"output_tokens"`
	TotalTokens  int     `json:"total_tokens"`
	InputCost    float64 `json:"input_cost"`
	OutputCost   float64 `json:"output_cost"`
}

type CostInterface interface {
	GetTotalCost() Cost
}

type Costs []Cost

type Model struct {
	ID          string            `json:"id"`
	Name        string            `json:"name"`
	Provider    string            `json:"provider"`
	Metadata    map[string]string `json:"metadata,omitempty"`
	InputPrice  float64           `json:"input_price_per_token,omitempty"`
	OutputPrice float64           `json:"output_price_per_token,omitempty"`
	MaxTokens   int               `json:"max_tokens,omitempty"`
}

type ModelUsageDetail struct {
	InputTokens              int     `json:"inputTokens"`
	OutputTokens             int     `json:"outputTokens"`
	CacheReadInputTokens     int     `json:"cacheReadInputTokens"`
	CacheCreationInputTokens int     `json:"cacheCreationInputTokens"`
	WebSearchRequests        int     `json:"webSearchRequests"`
	CostUSD                  float64 `json:"costUSD"`
	ContextWindow            int     `json:"contextWindow"`
}

type PromptRequest struct {
	Context          map[string]string `json:"context,omitempty"`
	Name             string            `json:"name"`
	Prompt           string            `json:"prompt"`
	StructuredOutput interface{}       `json:"structured_output,omitempty"` // Schema for structured JSON output
}

type PromptResponse struct {
	Request        PromptRequest `json:"request,omitempty"`
	Result         string        `json:"result"`
	StructuredData interface{}   `json:"structured_data,omitempty"` // Populated if structured output was requested
	Costs          Costs         `json:"costs,omitempty"`
	Model          string        `json:"model,omitempty"`
	Error          string        `json:"error,omitempty"`
	// Total wall-clock duration of the request
	Duration time.Duration `json:"duration,omitempty"`
	// Duration spent in the model processing as reported by the API
	DurationModel time.Duration `json:"duration_model,omitempty"`
	CacheHit      bool          `json:"cache_hit,omitempty"`
}

type Session struct {
	ID          string
	ProjectName string
	Costs       Costs
	// contains filtered or unexported fields
}

type Tokens struct {
	Input  int     `json:"input,omitempty"`
	Output int     `json:"output,omitempty"`
	Cost   float64 `json:"cost,omitempty"`
}

type TypedPromptRequest[T any] struct {
	Context map[string]string `json:"context,omitempty"`
	Name    string            `json:"name"`
	Prompt  string            `json:"prompt"`
}

type TypedPromptResponse[T any] struct {
	Request       TypedPromptRequest[T] `json:"request,omitempty"`
	Data          T                     `json:"data"`
	Costs         Costs                 `json:"costs,omitempty"`
	Model         string                `json:"model,omitempty"`
	Error         string                `json:"error,omitempty"`
	Duration      time.Duration         `json:"duration,omitempty"`
	DurationModel time.Duration         `json:"duration_model,omitempty"`
	CacheHit      bool                  `json:"cache_hit,omitempty"`
}