Azure OpenAI(AOAI) 서비스를 이용한 Text Embedding

Azure OpenAI 서비스에는 다양한 AI 서비스와 모델을 지원합니다. 그 중 텍스트를 벡터로 변환 해주는 Text Embedding 모델을 지원하고 있습니다. 해당 모델을 이용해 파이썬으로 텍스트를 임베딩 모델을 통해 벡터 결과를 얻는 코드를 소개합니다.

기본적으로 준비 사항은 OAAI의 Text Embedding 모델을 배포해주셔야 하고 해당 모델은 한국에서는 지원하지 않아 해외에 배포해야 합니다. 저는 text-embedding-ada-002 모델을 배포하고 실습했습니다.

아래는 해당 모델 사용 코드입니다. api키와 endpoint는 Azure AI Foundry에서 확인 가능합니다.

API Key 및 Endpoint URI 확인

 

파이썬 예제 코드

# text embedding model 예제 코드
import openai

# Azure OpenAI 엔드포인트 및 API 키 설정
client = openai.AzureOpenAI(
        api_key="1Jqh72gRIKJmKNrZqpsbTV56IW0SE6LF6sul4QxHgm0tLF4qlyYrJQQJ99BCACHYHv6XJ3w3AAAAACOGTJqH",
        api_version="2023-05-15",  # Azure OpenAI 버전
        azure_endpoint="https://jkjk1-m7tzfh67-eastus2.cognitiveservices.azure.com/openai/deployments/text-embedding-ada-002/embeddings?api-version=2023-05-15"
)

# 테스트할 텍스트
text = "Azure OpenAI embedding 테스트 중입니다."

# Azure OpenAI에 요청
response = client.embeddings.create(input=text, model="text-embedding-ada-002")

# 결과 출력
embedding = response.data[0].embedding
print(embedding)
print(f"Embedding 길이: {len(embedding)}")