llama_index_azure_openai_text_embedding_quickstart.py

1import os
2from llama_index.embeddings.azure_openai import AzureOpenAIEmbedding
3
4# Set up the Azure OpenAI Embedding model
5# You can also set these as environment variables: 
6# AZURE_OPENAI_API_KEY, AZURE_OPENAI_ENDPOINT, AZURE_OPENAI_API_VERSION
7embed_model = AzureOpenAIEmbedding(
8    model="text-embedding-ada-002",
9    deployment_name="your-deployment-name",
10    api_key="your-api-key",
11    azure_endpoint="https://your-resource-name.openai.azure.com/",
12    api_version="2023-05-15",
13)
14
15# Get embedding for a single text
16embedding = embed_model.get_text_embedding("It is raining cats and dogs here!")
17print(f"Embedding length: {len(embedding)}")
18print(f"First 5 elements: {embedding[:5]}")
19
20# Get embedding for a query
21query_embedding = embed_model.get_query_embedding("What is the weather like?")
22print(f"Query embedding length: {len(query_embedding)}")