prompt

type AssembledPrompt struct {
	TaskType     string            `json:"task_type"`
	SystemPrompt string            `json:"system_prompt"`
	AllowedTools []string          `json:"allowed_tools,omitempty"` // Tools this prompt can use
	Validators   []ValidatorConfig `json:"validators,omitempty"`    // Validators to apply at runtime
}

type AudioConfig struct {
	// Maximum audio size in MB (0 = unlimited)
	MaxSizeMB int `yaml:"max_size_mb,omitempty" json:"max_size_mb,omitempty"`
	// Allowed formats: ["mp3", "wav", "ogg", "webm"]
	AllowedFormats []string `yaml:"allowed_formats,omitempty" json:"allowed_formats,omitempty"`
	// Max duration in seconds (0 = unlimited)
	MaxDurationSec int `yaml:"max_duration_sec,omitempty" json:"max_duration_sec,omitempty"`
	// Whether metadata (duration, bitrate) is required
	RequireMetadata bool `yaml:"require_metadata,omitempty" json:"require_metadata,omitempty"`
}

type ChangelogEntry struct {
	Version     string `yaml:"version"`          // Version number
	Date        string `yaml:"date"`             // Date of change (YYYY-MM-DD)
	Author      string `yaml:"author,omitempty"` // Author of change
	Description string `yaml:"description"`      // Description of change
}

type CompilationInfo struct {
	CompiledWith string `yaml:"compiled_with" json:"compiled_with"`       // Compiler version
	CreatedAt    string `yaml:"created_at" json:"created_at"`             // Timestamp (RFC3339)
	Schema       string `yaml:"schema,omitempty" json:"schema,omitempty"` // Pack schema version (e.g., "v1")
}

type Config struct {
	APIVersion string            `yaml:"apiVersion" json:"apiVersion"`
	Kind       string            `yaml:"kind" json:"kind"`
	Metadata   metav1.ObjectMeta `yaml:"metadata,omitempty" json:"metadata,omitempty"`
	Spec       Spec              `yaml:"spec" json:"spec"`
}

type CostEstimate struct {
	MinCostUSD float64 `yaml:"min_cost_usd"` // Minimum cost per execution
	MaxCostUSD float64 `yaml:"max_cost_usd"` // Maximum cost per execution
	AvgCostUSD float64 `yaml:"avg_cost_usd"` // Average cost per execution
}

type ExampleContentPart struct {
	// Content type: "text", "image", "audio", "video"
	Type string `yaml:"type" json:"type"`
	// Text content (for type=text)
	Text string `yaml:"text,omitempty" json:"text,omitempty"`
	// For media content
	Media *ExampleMedia `yaml:"media,omitempty" json:"media,omitempty"`
}

type ExampleMedia struct {
	// Relative path to media file
	FilePath string `yaml:"file_path,omitempty" json:"file_path,omitempty"`
	// External URL
	URL string `yaml:"url,omitempty" json:"url,omitempty"`
	// MIME type
	MIMEType string `yaml:"mime_type" json:"mime_type"`
	// Detail level for images
	Detail string `yaml:"detail,omitempty" json:"detail,omitempty"`
	// Optional caption
	Caption string `yaml:"caption,omitempty" json:"caption,omitempty"`
}

type FileWriter interface {
	WriteFile(path string, data []byte, perm os.FileMode) error
}

type Fragment struct {
	Type              string `yaml:"fragment_type"`
	Version           string `yaml:"version"`
	Description       string `yaml:"description"`
	Content           string `yaml:"content"`
	SourceFile        string `yaml:"source_file,omitempty"`         // Source file path (for pack compilation)
	ResolvedAtCompile bool   `yaml:"resolved_at_compile,omitempty"` // Whether resolved at compile time
}

type FragmentRef struct {
	Name     string `yaml:"name"`
	Path     string `yaml:"path,omitempty"` // Optional: relative path to fragment file
	Required bool   `yaml:"required"`
}

type FragmentRepository interface {
	LoadFragment(name string, relativePath string, baseDir string) (*Fragment, error)
}

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

type ImageConfig struct {
	// Maximum image size in MB (0 = unlimited)
	MaxSizeMB int `yaml:"max_size_mb,omitempty" json:"max_size_mb,omitempty"`
	// Allowed formats: ["jpeg", "png", "webp", "gif"]
	AllowedFormats []string `yaml:"allowed_formats,omitempty" json:"allowed_formats,omitempty"`
	// Default detail level: "low", "high", "auto"
	DefaultDetail string `yaml:"default_detail,omitempty" json:"default_detail,omitempty"`
	// Whether captions are required
	RequireCaption bool `yaml:"require_caption,omitempty" json:"require_caption,omitempty"`
	// Max images per message (0 = unlimited)
	MaxImagesPerMsg int `yaml:"max_images_per_msg,omitempty" json:"max_images_per_msg,omitempty"`
}

type Info struct {
	TaskType       string
	Version        string
	Description    string
	FragmentCount  int
	RequiredVars   []string
	OptionalVars   []string
	ToolAllowlist  []string
	ModelOverrides []string
}

type Loader interface {
	LoadConfig(taskType string) (*Config, error)
	ListTaskTypes() []string
}

type MediaConfig struct {
	// Enable multimodal support for this prompt
	Enabled bool `yaml:"enabled" json:"enabled"`
	// Supported content types: "image", "audio", "video"
	SupportedTypes []string `yaml:"supported_types,omitempty" json:"supported_types,omitempty"`
	// Image-specific configuration
	Image *ImageConfig `yaml:"image,omitempty" json:"image,omitempty"`
	// Audio-specific configuration
	Audio *AudioConfig `yaml:"audio,omitempty" json:"audio,omitempty"`
	// Video-specific configuration
	Video *VideoConfig `yaml:"video,omitempty" json:"video,omitempty"`
	// Example multimodal messages
	Examples []MultimodalExample `yaml:"examples,omitempty" json:"examples,omitempty"`
}

type Metadata struct {
	Domain       string              `yaml:"domain,omitempty"`        // Domain/category (e.g., "customer-support")
	Language     string              `yaml:"language,omitempty"`      // Primary language (e.g., "en")
	Tags         []string            `yaml:"tags,omitempty"`          // Tags for categorization
	CostEstimate *CostEstimate       `yaml:"cost_estimate,omitempty"` // Estimated cost per execution
	Performance  *PerformanceMetrics `yaml:"performance,omitempty"`   // Performance benchmarks
	Changelog    []ChangelogEntry    `yaml:"changelog,omitempty"`     // Version history
}

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

type ModelOverride struct {
	SystemTemplate       string `yaml:"system_template,omitempty"`
	SystemTemplateSuffix string `yaml:"system_template_suffix,omitempty"`
}

type ModelTestResultRef struct {
	Provider     string  `yaml:"provider"`
	Model        string  `yaml:"model"`
	Date         string  `yaml:"date"`
	SuccessRate  float64 `yaml:"success_rate"`
	AvgTokens    int     `yaml:"avg_tokens,omitempty"`
	AvgCost      float64 `yaml:"avg_cost,omitempty"`
	AvgLatencyMs int     `yaml:"avg_latency_ms,omitempty"`
}

type MultimodalExample struct {
	// Example name/identifier
	Name string `yaml:"name" json:"name"`
	// Human-readable description
	Description string `yaml:"description,omitempty" json:"description,omitempty"`
	// Message role: "user", "assistant"
	Role string `yaml:"role" json:"role"`
	// Content parts for this example
	Parts []ExampleContentPart `yaml:"parts" json:"parts"`
}

type Pack struct {
	// Schema reference for validation
	Schema string `json:"$schema,omitempty"` // JSON Schema URL for validation

	// Identity
	ID          string `json:"id"`          // Pack ID (e.g., "customer-support")
	Name        string `json:"name"`        // Human-readable name
	Version     string `json:"version"`     // Pack version
	Description string `json:"description"` // Pack description

	// Template Engine (shared across all prompts in pack)
	TemplateEngine *TemplateEngineInfo `json:"template_engine"`

	// Prompts - Map of task_type -> PackPrompt
	Prompts map[string]*PackPrompt `json:"prompts"`

	// Tools - Map of tool_name -> PackTool (per PromptPack spec Section 9)
	// Tools are defined at pack level and referenced by name in prompts
	Tools map[string]*PackTool `json:"tools,omitempty"`

	// Shared fragments (can be referenced by any prompt)
	Fragments map[string]string `json:"fragments,omitempty"` // Resolved fragments: name -> content

	// Metadata
	Metadata    *Metadata        `json:"metadata,omitempty"`
	Compilation *CompilationInfo `json:"compilation,omitempty"`
}

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

type PackPrompt struct {
	// Identity
	ID          string `json:"id"`          // Prompt ID (task_type)
	Name        string `json:"name"`        // Human-readable name
	Description string `json:"description"` // Prompt description
	Version     string `json:"version"`     // Prompt version

	// Prompt
	SystemTemplate string `json:"system_template"`

	// Variables
	Variables []VariableMetadata `json:"variables,omitempty"`

	// Tools
	Tools      []string        `json:"tools,omitempty"`       // Allowed tool names
	ToolPolicy *ToolPolicyPack `json:"tool_policy,omitempty"` // Tool usage policy

	// Multimodal media configuration
	MediaConfig *MediaConfig `json:"media,omitempty"`

	// Pipeline
	Pipeline map[string]interface{} `json:"pipeline,omitempty"` // Pipeline configuration

	// Parameters
	Parameters *ParametersPack `json:"parameters,omitempty"` // Model-specific parameters

	// Validators
	Validators []ValidatorConfig `json:"validators,omitempty"`

	// Model Testing
	TestedModels []ModelTestResultRef `json:"tested_models,omitempty"`

	// Model Overrides
	ModelOverrides map[string]ModelOverride `json:"model_overrides,omitempty"`
}

type PackTool struct {
	Name        string      `json:"name"`        // Tool function name (required)
	Description string      `json:"description"` // Tool description (required)
	Parameters  interface{} `json:"parameters"`  // JSON Schema for input parameters (required)
}

type ParametersPack struct {
	Temperature *float64 `json:"temperature,omitempty"`
	MaxTokens   *int     `json:"max_tokens,omitempty"`
	TopP        *float64 `json:"top_p,omitempty"`
	TopK        *int     `json:"top_k,omitempty"`
}

type ParsedTool struct {
	Name        string
	Description string
	InputSchema json.RawMessage
}

type PerformanceMetrics struct {
	AvgLatencyMs int     `yaml:"avg_latency_ms"` // Average latency in milliseconds
	P95LatencyMs int     `yaml:"p95_latency_ms"` // 95th percentile latency
	AvgTokens    int     `yaml:"avg_tokens"`     // Average tokens used
	SuccessRate  float64 `yaml:"success_rate"`   // Success rate (0.0-1.0)
}

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

type Repository interface {
	LoadPrompt(taskType string) (*Config, error)
	LoadFragment(name string, relativePath string, baseDir string) (*Fragment, error)
	ListPrompts() ([]string, error)
	SavePrompt(config *Config) error
}

type Spec struct {
	TaskType       string                   `yaml:"task_type" json:"task_type"`
	Version        string                   `yaml:"version" json:"version"`
	Description    string                   `yaml:"description" json:"description"`
	TemplateEngine *TemplateEngineInfo      `yaml:"template_engine,omitempty" json:"template_engine,omitempty"` // Template engine configuration
	Fragments      []FragmentRef            `yaml:"fragments,omitempty" json:"fragments,omitempty"`             // New: fragment assembly
	SystemTemplate string                   `yaml:"system_template" json:"system_template"`
	Variables      []VariableMetadata       `yaml:"variables,omitempty" json:"variables,omitempty"` // Variable definitions with rich metadata
	ModelOverrides map[string]ModelOverride `yaml:"model_overrides,omitempty" json:"model_overrides,omitempty"`
	AllowedTools   []string                 `yaml:"allowed_tools,omitempty" json:"allowed_tools,omitempty"` // Tools this prompt can use
	MediaConfig    *MediaConfig             `yaml:"media,omitempty" json:"media,omitempty"`                 // Multimodal media configuration
	Validators     []ValidatorConfig        `yaml:"validators,omitempty" json:"validators,omitempty"`       // Validators/Guardrails for production runtime
	TestedModels   []ModelTestResultRef     `yaml:"tested_models,omitempty" json:"tested_models,omitempty"` // Model testing metadata
	Metadata       *Metadata                `yaml:"metadata,omitempty" json:"metadata,omitempty"`           // Additional metadata for pack format
	Compilation    *CompilationInfo         `yaml:"compilation,omitempty" json:"compilation,omitempty"`     // Compilation information
}

type TemplateEngineInfo struct {
	Version  string   `yaml:"version" json:"version"`                       // Template engine version (e.g., "v1")
	Syntax   string   `yaml:"syntax" json:"syntax"`                         // Template syntax (e.g., "{{variable}}")
	Features []string `yaml:"features,omitempty" json:"features,omitempty"` // Supported features
}

type TestResultSummary struct {
	Success   bool
	Cost      float64
	LatencyMs int
	Tokens    int
}

type TimeProvider interface {
	Now() time.Time
}

type ToolData struct {
	FilePath string
	Data     []byte
}

type ToolPolicyPack struct {
	ToolChoice          string   `json:"tool_choice,omitempty"`
	MaxRounds           int      `json:"max_rounds,omitempty"`
	MaxToolCallsPerTurn int      `json:"max_tool_calls_per_turn,omitempty"`
	Blocklist           []string `json:"blocklist,omitempty"`
}

type ValidatorConfig struct {
	// Embed base config (Type, Params)
	validators.ValidatorConfig `yaml:",inline" json:",inline"`
	// Enable/disable validator (default: true)
	Enabled *bool `yaml:"enabled,omitempty" json:"enabled,omitempty"`
	// Fail execution on violation (default: true)
	FailOnViolation *bool `yaml:"fail_on_violation,omitempty" json:"fail_on_violation,omitempty"`
}

type VariableMetadata struct {
	Name        string                 `yaml:"name" json:"name"`
	Type        string                 `yaml:"type,omitempty" json:"type,omitempty"` // "string", "number", "boolean", "object", "array"
	Required    bool                   `yaml:"required" json:"required"`
	Default     interface{}            `yaml:"default,omitempty" json:"default,omitempty"`
	Description string                 `yaml:"description,omitempty" json:"description,omitempty"`
	Example     interface{}            `yaml:"example,omitempty" json:"example,omitempty"`
	Validation  map[string]interface{} `yaml:"validation,omitempty" json:"validation,omitempty"`
}

type VideoConfig struct {
	// Maximum video size in MB (0 = unlimited)
	MaxSizeMB int `yaml:"max_size_mb,omitempty" json:"max_size_mb,omitempty"`
	// Allowed formats: ["mp4", "webm", "ogg"]
	AllowedFormats []string `yaml:"allowed_formats,omitempty" json:"allowed_formats,omitempty"`
	// Max duration in seconds (0 = unlimited)
	MaxDurationSec int `yaml:"max_duration_sec,omitempty" json:"max_duration_sec,omitempty"`
	// Whether metadata (resolution, fps) is required
	RequireMetadata bool `yaml:"require_metadata,omitempty" json:"require_metadata,omitempty"`
}