guard

type ChainMode int

type CheckResult struct {
	// Passed 是否通过
	Passed bool `json:"passed"`

	// Score 风险分数 (0-1，越高风险越大)
	Score float64 `json:"score"`

	// Category 风险类别
	Category string `json:"category,omitempty"`

	// Reason 原因
	Reason string `json:"reason,omitempty"`

	// Findings 发现的问题
	Findings []Finding `json:"findings,omitempty"`

	// Metadata 额外元数据
	Metadata map[string]any `json:"metadata,omitempty"`
}

type Finding struct {
	// Type 问题类型
	Type string `json:"type"`

	// Text 问题文本
	Text string `json:"text"`

	// Position 位置
	Position Position `json:"position,omitempty"`

	// Severity 严重程度: low, medium, high, critical
	Severity string `json:"severity"`

	// Suggestion 建议
	Suggestion string `json:"suggestion,omitempty"`
}

type Guard interface {
	// Name 返回守卫名称
	Name() string

	// Check 执行检查
	Check(ctx context.Context, input string) (*CheckResult, error)

	// Enabled 是否启用
	Enabled() bool
}

type GuardAction string

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

type GuardConfig struct {
	// Enabled 是否启用
	Enabled bool `json:"enabled" yaml:"enabled"`

	// Threshold 风险阈值（超过则拒绝）
	Threshold float64 `json:"threshold" yaml:"threshold"`

	// Action 触发后的动作
	Action GuardAction `json:"action" yaml:"action"`

	// Categories 要检查的类别
	Categories []string `json:"categories" yaml:"categories"`

	// Allowlist 允许列表
	Allowlist []string `json:"allowlist" yaml:"allowlist"`

	// Blocklist 阻止列表
	Blocklist []string `json:"blocklist" yaml:"blocklist"`
}

type InputGuard interface {
	Guard
	// IsInputGuard 标记为输入守卫
	IsInputGuard()
}

type Middleware func(ctx context.Context, input string, next func(context.Context, string) (string, error)) (string, error)

type OutputGuard interface {
	Guard
	// IsOutputGuard 标记为输出守卫
	IsOutputGuard()
}

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

type PIIOption func(*PIIGuard)

type Position struct {
	Start int `json:"start"`
	End   int `json:"end"`
}

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

type PromptInjectionOption func(*PromptInjectionGuard)