pyrit.score.InsecureCodeScorer

pyrit.score.InsecureCodeScorer#

class InsecureCodeScorer(chat_target: PromptChatTarget, threshold: float = 0.5, system_prompt_path: Path | None = None)[source]#

Bases: Scorer

A scorer that uses an LLM to evaluate code snippets for potential security vulnerabilities. Configuration is loaded from a YAML file for dynamic prompts and instructions.

__init__(chat_target: PromptChatTarget, threshold: float = 0.5, system_prompt_path: Path | None = None)[source]#

Methods

__init__(chat_target[, threshold, ...])

get_identifier()

Returns an identifier dictionary for the scorer.

get_scorer_metrics(dataset_name[, metrics_type])

Returns evaluation statistics for the scorer using the dataset_name of the human labeled dataset that this scorer was run against.

scale_value_float(value, min_value, max_value)

Scales a value from 0 to 1 based on the given min and max values.

score_async(request_response, *[, task])

Score the request_response, add the results to the database and return a list of Score objects.

score_image_async(image_path, *[, task])

Scores the given image using the chat target.

score_prompts_with_tasks_batch_async(*, ...)

score_response_async(*, response, scorers[, ...])

Score a response using multiple scorers in parallel.

score_response_select_first_success_async(*, ...)

Score response pieces sequentially until finding a successful score.

score_response_with_objective_async(*, response)

Score a response using both auxiliary and objective scorers.

score_responses_inferring_tasks_batch_async(*, ...)

Scores a batch of responses (ignores non-assistant messages).

score_text_async(text, *[, task])

Scores the given text based on the task using the chat target.

score_text_batch_async(*, texts[, tasks, ...])

validate(request_response, *[, task])

Validates the request response to ensure it meets requirements for scoring.

Attributes

scorer_type

scorer_type: ScoreType#
validate(request_response: PromptRequestPiece, *, task: str | None = None) → None[source]#

Validates the request response to ensure it meets requirements for scoring.

Parameters:

  • request_response (PromptRequestPiece) – The code snippet to be validated.

  • task (Optional[str]) – Optional task descriptor.

Contents