enhancer

type CompletionRequest struct {
	System   string
	User     string
	Messages []Message
}

type CompletionResponse struct {
	Text     string
	Provider string
	Model    string
}

type ContentWarningsConfig struct {
	Enabled bool
	// ExtraActions extends the irreversible-action word list (each entry is
	// its own synonym group; matched like the built-ins).
	ExtraActions []string
	// NumMaxLen is the maximum digit-run length checked by the number rule;
	// longer runs are ids/hashes/timestamps, not quantities. Default 5.
	NumMaxLen int
}

type Context struct {
	Files     []ContextFile `json:"files,omitempty"`
	Retrieval []string      `json:"retrieval,omitempty"`
}

type ContextFile struct {
	Path    string `json:"path,omitempty"`
	Content string `json:"content,omitempty"`
}

type ContextProvider interface {
	Retrieve(ctx context.Context, req Request) ([]string, error)
}

type LanguageGuardConfig struct {
	// Enabled turns the guard on.
	Enabled bool
	// Reanchor selects strategy 1 (re-request once with an explicit language
	// directive) on a detected mismatch. When false, the guard uses strategy 2
	// only (detect + warn, no extra model call), which adds zero latency.
	Reanchor bool
}

type LanguageGuardEvent struct {
	InputLang  string // script class of the user's ORIGINAL prompt (the source of truth)
	OutputLang string // script class of the first model output
	Mismatch   bool   // input and output classes differed
	Retried    bool   // a re-anchored retry was issued
	Corrected  bool   // the retry produced the input's language class
	FinalLang  string // script class of the returned text
}

type Message struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

type MessageStyle int

type Metadata struct {
	UsedContext []string      `json:"used_context,omitempty"`
	Sections    []SectionInfo `json:"sections,omitempty"`
	Provider    string        `json:"provider,omitempty"`
	Model       string        `json:"model,omitempty"`
}

type Options struct {
	MaxContextTokens int  `json:"max_context_tokens,omitempty"`
	ReturnMetadata   bool `json:"return_metadata,omitempty"`
}

type Provider interface {
	Complete(ctx context.Context, req CompletionRequest) (CompletionResponse, error)
}

type Request struct {
	Prompt     string    `json:"prompt"`
	Client     string    `json:"client,omitempty"`
	CWD        string    `json:"cwd,omitempty"`
	Mode       string    `json:"mode,omitempty"`
	History    []Message `json:"history,omitempty"`
	Rules      []string  `json:"rules,omitempty"`
	Guidelines []string  `json:"guidelines,omitempty"`
	Context    Context   `json:"context,omitempty"`
	Options    Options   `json:"options,omitempty"`
}

type Response struct {
	EnhancedPrompt string   `json:"enhanced_prompt"`
	Warnings       []string `json:"warnings,omitempty"`
	Metadata       Metadata `json:"metadata,omitempty"`
}

type SectionInfo struct {
	Name      string `json:"name"`
	Length    int    `json:"length"`
	Truncated bool   `json:"truncated"`
}

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

type ValidationError struct {
	Message string
}