Skip to main content
Open on GitHub

01.AI

01.AI, founded by Dr. Kai-Fu Lee, is a global company at the forefront of AI 2.0. They offer cutting-edge large language models, including the Yi series, which range from 6B to hundreds of billions of parameters. 01.AI also provides multimodal models, an open API platform, and open-source options like Yi-34B/9B/6B and Yi-VL.

Installation and Setup​

Register and get an API key from either the China site here or the global site here.

LLMs​

See a usage example.

from langchain_community.llms import YiLLM
API Reference:YiLLM

Chat models​

See a usage example.

from langchain_community.chat_models import ChatYi
API Reference:ChatYi

Was this page helpful?