AzureOpenAI

让我们使用设置了环境变量的OpenAI Embedding类，以指示使用Azure端点。

# 设置openai包所需的环境变量，以便知道如何访问Azure
import os

os.environ["OPENAI_API_TYPE"] = "azure"
os.environ["OPENAI_API_BASE"] = "https://<your-endpoint.openai.azure.com/"
os.environ["OPENAI_API_KEY"] = "your AzureOpenAI key"
os.environ["OPENAI_API_VERSION"] = "2023-05-15"

from langchain.embeddings import OpenAIEmbeddings

embeddings = OpenAIEmbeddings(deployment="your-embeddings-deployment-name")

text = "这是一个测试文档。"

query_result = embeddings.embed_query(text)

doc_result = embeddings.embed_documents([text])