pyrit.models.DataTypeSerializer

pyrit.models.DataTypeSerializer#

class DataTypeSerializer[source]#

Bases: ABC

Abstract base class for data type normalizers.

Responsible for reading and saving multi-modal data types to local disk or Azure Storage Account.

__init__()#

Methods

__init__()

data_on_disk()

Returns True if the data is stored on disk.

get_data_filename([file_name])

Generates or retrieves a unique filename for the data file.

get_extension(file_path)

Get the file extension from the file path.

get_mime_type(file_path)

Get the MIME type of the file path.

get_sha256()

read_data()

Reads the data from the storage.

read_data_base64()

Reads the data from the storage.

save_b64_image(data[, output_filename])

Saves the base64 encoded image to storage.

save_data(data[, output_filename])

Saves the data to storage.

save_formatted_audio(data[, num_channels, ...])

Saves the PCM16 of other specially formatted audio data to storage.

Attributes

data_type

value

category

data_sub_directory

file_extension

category: str#
abstract data_on_disk() → bool[source]#

Returns True if the data is stored on disk.

data_sub_directory: str#
data_type: Literal['text', 'image_path', 'audio_path', 'video_path', 'url', 'reasoning', 'error', 'function_call', 'tool_call', 'function_call_output']#
file_extension: str#
async get_data_filename(file_name: str | None = None) → Path | str[source]#

Generates or retrieves a unique filename for the data file.

static get_extension(file_path: str) → str | None[source]#

Get the file extension from the file path.

static get_mime_type(file_path: str) → str | None[source]#

Get the MIME type of the file path.

async get_sha256() → str[source]#
async read_data() → bytes[source]#

Reads the data from the storage.

Returns:

The data read from storage.

Return type:

bytes

async read_data_base64() → str[source]#

Reads the data from the storage.

async save_b64_image(data: str, output_filename: str = None) → None[source]#

Saves the base64 encoded image to storage.

Parameters:

  • data – string with base64 data

  • output_filename (optional, str) – filename to store image as. Defaults to UUID if not provided

async save_data(data: bytes, output_filename: str | None = None) → None[source]#

Saves the data to storage.

Parameters:

  • data – bytes: The data to be saved.

  • output_filename (optional, str) – filename to store data as. Defaults to UUID if not provided

async save_formatted_audio(data: bytes, num_channels: int = 1, sample_width: int = 2, sample_rate: int = 16000, output_filename: str | None = None) → None[source]#

Saves the PCM16 of other specially formatted audio data to storage.

Parameters:

  • data – bytes with audio data

  • output_filename (optional, str) – filename to store audio as. Defaults to UUID if not provided

  • num_channels (optional, int) – number of channels in audio data. Defaults to 1

  • sample_width (optional, int) – sample width in bytes. Defaults to 2

  • sample_rate (optional, int) – sample rate in Hz. Defaults to 16000

value: str#
Contents