execution

type AggregationStrategy string

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

type CacheStats struct {
	// Hits is the number of cache hits
	Hits int64
	// Misses is the number of cache misses
	Misses int64
	// Evictions is the number of cache evictions
	Evictions int64
	// TotalLookups is the total number of lookups
	TotalLookups int64
}

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

type CustomDetector func(ctx context.Context, criteria format.DetectionCriteria, response string) (bool, int, map[string]interface{}, error)

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

type DetectionEngine interface {
	// Detect detects vulnerabilities in an LLM response
	Detect(ctx context.Context, template *format.Template, response string) (bool, int, map[string]interface{}, error)
}

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

type ExecutionOptions struct {
	// Provider is the LLM provider to use
	Provider LLMProvider
	// DetectionEngine is the detection engine to use
	DetectionEngine DetectionEngine
	// RateLimiter is the rate limiter to use
	RateLimiter RateLimiter
	// InputValidator is the input validator to use
	InputValidator interfaces.InputValidator
	// Timeout is the timeout for template execution
	Timeout time.Duration
	// RetryCount is the number of retries for failed requests
	RetryCount int
	// RetryDelay is the delay between retries
	RetryDelay time.Duration
	// MaxConcurrent is the maximum number of concurrent executions
	MaxConcurrent int
	// Variables is a map of variables to substitute in templates
	Variables map[string]interface{}
	// ProviderOptions is a map of provider-specific options
	ProviderOptions map[string]interface{}
	// StrictValidation determines if validation errors should fail execution
	StrictValidation bool
	// SanitizePrompts determines if prompts should be sanitized before execution
	SanitizePrompts bool
	// UserID is the ID of the user making the request
	UserID string
	// EnableUserRateLimiting determines if user-specific rate limiting is enabled
	EnableUserRateLimiting bool
}

type ExecutionStats struct {
	// TotalExecutions is the total number of template executions
	TotalExecutions int64
	// SuccessfulExecutions is the number of successful executions
	SuccessfulExecutions int64
	// FailedExecutions is the number of failed executions
	FailedExecutions int64
	// CachedResponses is the number of cached responses used
	CachedResponses int64
	// TotalExecutionTime is the total time spent executing templates
	TotalExecutionTime time.Duration
	// RetryCount is the number of retries performed
	RetryCount int64
}

type LLMProvider interface {
	// SendPrompt sends a prompt to the LLM and returns the response
	SendPrompt(ctx context.Context, prompt string, options map[string]interface{}) (string, error)
	// GetSupportedModels returns the list of supported models
	GetSupportedModels() []string
	// GetName returns the name of the provider
	GetName() string
}

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

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

type PipelineStage int

type RateLimiter = interfaces.RateLimiter

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

type ResponseCacheEntry struct {
	// Response is the cached response
	Response string
	// CreatedAt is the time the entry was created
	CreatedAt time.Time
	// ExpiresAt is the time the entry expires
	ExpiresAt time.Time
	// Size is an estimate of the response size in bytes
	Size int
}

type ResultPostprocessor interface {
	// Postprocess postprocesses a result
	Postprocess(ctx context.Context, result *interfaces.TemplateResult) (*interfaces.TemplateResult, error)
}

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

type TemplatePreprocessor interface {
	// Preprocess preprocesses a template
	Preprocess(ctx context.Context, template *format.Template, options map[string]interface{}) (*format.Template, error)
}

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