tuning

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

type PromptConfig struct {
	PromptsDir string
}

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

type QueryMetadata struct {
	QualityScore    *float64
	IsDeprecated    bool
	DeprecationNote string
	ContainsPII     bool
	RowCount        *int64
}

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

type Rules struct {
	// RequireDataHubCheck requires checking DataHub before writing queries.
	RequireDataHubCheck bool `yaml:"require_datahub_check"`

	// WarnOnDeprecated warns when accessing deprecated entities.
	WarnOnDeprecated bool `yaml:"warn_on_deprecated"`

	// QualityThreshold is the minimum quality score for data access.
	QualityThreshold float64 `yaml:"quality_threshold"`

	// MaxQueryLimit is the maximum number of rows a query can return.
	MaxQueryLimit int `yaml:"max_query_limit"`

	// RequirePIIAcknowledgment requires acknowledgment when accessing PII data.
	RequirePIIAcknowledgment bool `yaml:"require_pii_acknowledgment"`

	// Custom rules
	Custom map[string]any `yaml:"custom,omitempty"`
}

type Severity string

type Violation struct {
	Rule       string
	Severity   Severity
	Message    string
	Suggestion string
}