LangChain是一个简化使用大型语言模型的应用框架,很多朋友都是基于LangChain做的AI大语言模型开发,本文就分享下LangChain中如何使用便携AI聚合API。
Python使用LangChain需要安装langchain和openai包,请先安装对应的库。
在LangChain中使用便携AI聚合API有两种方法,都很方便。
方法1:初始化模型时设置openai_api_base
参数
其中openai_api_key
就是你在便携AI聚合API后台生成的令牌(获取方法:《便携AI聚合API新建令牌(API key)教程》),openai_api_base
是中转接口地址https://api.bianxie.ai/v1:
from langchain.chat_models import ChatOpenAI llm = ChatOpenAI( model='gpt-3.5-turbo', openai_api_base="https://api.bianxie.ai/v1", openai_api_key="sk-xxxx", ) res = llm.predict("Who are you") print(res)
方法2:设置环境变量
直接设置两个环境变量:OPENAI_BASE_UR
是中转接口地址https://api.bianxie.ai/v1,LOPENAI_API_KEY
是你在便携AI聚合API后台生成的令牌(获取方法:《便携AI聚合API新建令牌(API key)教程》):
from langchain.chat_models import ChatOpenAI import os os.environ["OPENAI_BASE_URL"] = 'https://api.bianxie.ai/v1' os.environ["OPENAI_API_KEY"] = 'sk-xxxx' llm = ChatOpenAI( model='gpt-3.5-turbo', ) res = llm.predict("Who are you") print(res)
这两种方法都能得到GPT的正确回复: