Elsai Embeddings v0.1.0

Azure OpenAI embedding generation.

INFO

v0.2.0 adds Amazon Bedrock embeddings. See v2.0 docs.

Installation

pip install --extra-index-url https://core-packages.elsai.ai/root/elsai-embeddings/ elsai-embeddings==0.1.0

AzureOpenAIEmbeddingModel

from elsai_embeddings.azure_openai import AzureOpenAIEmbeddingModel

model = AzureOpenAIEmbeddingModel(
    azure_endpoint="https://your-resource.openai.azure.com/",
    openai_api_key="your-key",
    openai_api_version="2023-05-15",
    deployment_name="text-embedding-ada-002",
)

vector = model.embed_query("What is machine learning?")
vectors = model.embed_documents(["Document 1.", "Document 2."])