# 引言
01.AI,由李开复博士创立,是AI 2.0领域的全球领导者,提供先进的大规模语言模型,包括Yi系列。这些模型参数范围从6B到数千亿。同时,01.AI提供多模态模型、开放API平台以及开源选项,如Yi-34B/9B/6B和Yi-VL。本篇文章旨在指导读者如何使用Langchain与Yi LLM进行集成,并提供实用的代码示例。
# 主要内容
## 1. 安装Langchain包
要使用Yi LLM的集成功能,首先需要安装`langchain-community`包:
```bash
%pip install -qU langchain-community
2. 前置条件
要访问Yi LLM API,需要一个API密钥。请访问01.AI官网申请API密钥。在申请时,需要指定是否用于国内(中国)或国际使用。
3. 使用Yi LLM
import os
# 设置API密钥
os.environ["YI_API_KEY"] = "YOUR_API_KEY"
from langchain_community.llms import YiLLM
# 加载模型
llm = YiLLM(model