pyrit.embedding.OpenAiTextEmbedding

class OpenAiTextEmbedding(*, model: str, api_key: str)

Bases: _TextEmbedding

Provides text embedding functionality using OpenAI services.

__init__(*, model: str, api_key: str) → None

Generate embedding using OpenAI API.

Parameters:

• model – The model to use

• api_key – The API key to use

Methods

__init__(*, model, api_key)	Generate embedding using OpenAI API.
generate_text_embedding(text, **kwargs)	Generate text embedding.