generator

Documentation

Index

• Constants
• type Generator
  • func New(o Opts) (*Generator, error)
  • func (g *Generator) GenerateStatements(ctx context.Context, vulns []Vulnerability, h func([]vex.Statement) error) error
• type Hints
• type Opts
• type Vulnerability

Constants

const (
	DefaultTemperature      = 0.0
	DefaultBatchSize        = 10
	DefaultSleepOnRateLimit = 10 * time.Second
	DefaultRetryOnRateLimit = 10
)

Types

type Generator

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

func New

func New(o Opts) (*Generator, error)

func (*Generator) GenerateStatements

func (g *Generator) GenerateStatements(ctx context.Context, vulns []Vulnerability, h func([]vex.Statement) error) error

type Hints

type Hints struct {
	Descriptions             []string
	Container                bool
	NotServer                bool
	UsedCommands             []string
	UnusedCommands           []string
	CompromiseOnAvailability bool // Focus on Confidentiality and Integrity
}

type Opts

type Opts struct {
	LLM              llms.Model
	Temperature      float64
	BatchSize        int // Decrease to avoid rate limit
	SleepOnRateLimit time.Duration
	RetryOnRateLimit int

	Hints Hints

	DebugDir string
}

type Vulnerability

type Vulnerability struct {
	VulnID      string `json:"vulnId"`
	PkgID       string `json:"pkgId"`
	Title       string `json:"title"`
	Description string `json:"description,omitempty"`
	Severity    string `json:"severity,omitempty"`
}