llm

Documentation

Index

• type Client
  • func NewClient(spec string) (*Client, error)
  • func (c *Client) Generate(prompt string) (string, error)
  • func (c *Client) GeneratePayloads(endpoint, paramType, vulnClass string) []string
  • func (c *Client) TriageFinding(request, response, vulnTitle string) (isReal bool, confidence string, reason string)

Types

type Client

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

Client for LLM inference via Ollama local API

func NewClient

func NewClient(spec string) (*Client, error)

NewClient parses spec like "ollama:llama3" and returns a configured client

func (*Client) Generate

func (c *Client) Generate(prompt string) (string, error)

Generate sends a prompt to the LLM and returns the text response

func (*Client) GeneratePayloads

func (c *Client) GeneratePayloads(endpoint, paramType, vulnClass string) []string

GeneratePayloads asks the LLM to generate attack payloads for a given context

func (*Client) TriageFinding

func (c *Client) TriageFinding(request, response, vulnTitle string) (isReal bool, confidence string, reason string)

TriageFinding asks the LLM to classify whether a finding is a true or false positive