pyrit.memory.MemoryEmbedding

class MemoryEmbedding(*, embedding_model: EmbeddingSupport | None = None)

Bases: object

The MemoryEmbedding class is responsible for encoding the memory embeddings.

Parameters:

embedding_model (EmbeddingSupport) – An instance of a class that supports embedding generation.

__init__(*, embedding_model: EmbeddingSupport | None = None)

Methods

__init__(*[, embedding_model])
generate_embedding_memory_data(*, message_piece)	Generates metadata for a message piece.

generate_embedding_memory_data(*, message_piece: MessagePiece) → EmbeddingDataEntry

Generates metadata for a message piece.

Parameters:

message_piece (MessagePiece) – the message piece for which to generate a text embedding

Returns:

The generated metadata.

Return type:

EmbeddingDataEntry