llm

type BaseProvider struct {
	Config interface{}
}

type ClaudeConfig struct {
	APIKey string `yaml:"api_key"`
	Model  string `yaml:"model"`
}

type ClaudeProvider struct {
	BaseProvider
	// contains filtered or unexported fields
}

type OllamaConfig struct {
	Model string `yaml:"model"`
	Host  string `yaml:"host"`
}

type OllamaProvider struct {
	BaseProvider
	// contains filtered or unexported fields
}

type OpenAIConfig struct {
	APIKey string `yaml:"api_key"`
	Model  string `yaml:"model"`
}

type OpenAIProvider struct {
	BaseProvider
	// contains filtered or unexported fields
}

type Provider interface {
	// GenerateScripts creates a main script and test script from a natural language description
	GenerateScripts(ctx context.Context, description string) (ScriptPair, error)
	// FixScripts attempts to fix both scripts based on test failures
	FixScripts(ctx context.Context, scripts ScriptPair, error string) (ScriptPair, error)
	// Name returns a human-readable name for the provider
	Name() string
}

type ProviderConfig struct {
	Provider string       `yaml:"provider"`
	Ollama   OllamaConfig `yaml:"ollama,omitempty"`
	Claude   ClaudeConfig `yaml:"claude,omitempty"`
	OpenAI   OpenAIConfig `yaml:"openai,omitempty"`
}

type ScriptPair struct {
	MainScript string // The feature script that implements the functionality
	TestScript string // The test script that verifies the feature script
}

type Test struct {
	Name        string
	Setup       []string
	Input       string
	Expected    string
	Timeout     time.Duration
	Environment map[string]string
}

type TestFailure struct {
	Test     Test
	Output   string
	Error    error
	ExitCode int
}