zbplan

type ClaudeConfig struct {
	// APIKey is the Anthropic API key. Required.
	APIKey string
	// Model defaults to "claude-sonnet-4-6" when empty.
	Model string
	// BaseURL overrides the Anthropic API endpoint when non-empty.
	BaseURL string
	// MaxTokens defaults to 16000 when zero. Note: with adaptive thinking
	// enabled, max_tokens covers both thinking tokens and response text.
	MaxTokens int
	// DisableThinking disables adaptive thinking when true.
	// By default, adaptive thinking is enabled.
	DisableThinking bool
}

type Config struct {
	// Model is the eino tool-calling model that drives the planning agent.
	// Use NewClaudeModel or NewClaudeModelFromEnv for a Claude-backed model.
	Model model.ToolCallingChatModel

	// BuildKitAddr is the optional BuildKit daemon endpoint
	// (e.g. "tcp://host:1234" or "docker-container://buildkitd").
	// When empty, BuildKit's client falls back to the system default address.
	BuildKitAddr string

	// ContextDir is the source-code directory to plan for.
	ContextDir string

	// Variables are injected as ZEABUR_ENV_* build args into every FROM stage.
	Variables map[string]string
	// UserDockerfile is an existing Dockerfile to try before invoking the agent.
	// If it builds successfully the agent is skipped entirely.
	// If it fails, the agent receives it alongside the build error as its
	// starting context. The attempt does not count toward MaxBuildAttempts.
	UserDockerfile string
	// OCIOutput receives the OCI image tarball on a successful build.
	// When nil no OCI tarball is produced (cheaper — only a verify build runs).
	//
	// io.WriteCloser is required (rather than io.Writer) because BuildKit's
	// session layer calls Close() as part of stream finalization: it flushes
	// any trailing bytes and propagates close errors back into the solve result.
	// This means Close() carries real semantics — wrapping formats such as
	// gzip or zstd rely on it to write their closing blocks, and an
	// io.PipeWriter relies on it to signal EOF to the reader.
	//
	// BuildKit calls Close() before Run returns, so callers do not need to
	// close OCIOutput themselves. Closing it again after Run is harmless for
	// most implementations (e.g. *os.File silently returns an error that can
	// be ignored), but is unnecessary.
	OCIOutput io.WriteCloser
	// ExtraTools are appended to the default eight plantools tools.
	ExtraTools []tool.BaseTool
	// SystemPrompt overrides DefaultSystemPrompt.
	SystemPrompt string
	// MaxBuildAttempts is the number of agent generate→build cycles before
	// Run returns an error. Defaults to 3.
	MaxBuildAttempts int
	// MaxAgentSteps is the maximum number of ReAct steps per Generate call.
	// Defaults to 100.
	MaxAgentSteps int
	// Logger defaults to slog.Default() when nil.
	Logger *slog.Logger
}

type OpenAIConfig struct {
	// APIKey is the OpenAI API key. Required.
	APIKey string
	// Model defaults to "gpt-5.5" when empty.
	Model string
	// BaseURL overrides the OpenAI API endpoint when non-empty.
	BaseURL string
}

type Result struct {
	// Dockerfile is the working Dockerfile text.
	Dockerfile string
	// Attempts is the number of agent generate→build cycles used.
	// Zero means UserDockerfile built successfully without invoking the agent.
	Attempts int
	// FromUser is true when Dockerfile is the unchanged UserDockerfile.
	FromUser bool
}