吴恩达团队开源aisuite,一个接口调用所有AI模型
- 免费干货
- 3小时前
- 6热度
- 0评论
项目里同时用ChatGPT、Claude、国内模型,每个接口都不一样,代码写得一团糟?吴恩达团队搞了个新工具,用一个"插座"搞定所有AI模型。
aisuite是什么
aisuite是吴恩达(Andrew Ng)团队开源的一个Python库,核心就一件事:用一个统一接口调用市面上所有主流AI模型。
吴恩达,斯坦福大学教授,前谷歌大脑项目负责人、前百度首席科学家,Coursera联合创始人。他的《机器学习》课程是全球播放量最高的AI入门课。
安装和使用
安装一行命令搞定:
pip install aisuite
调用不同模型的话,只需要改一个字符串:
import aisuite as ai
client = ai.Client()
# 调用GPT-4o
response = client.chat.completions.create(
model="openai:gpt-4o",
messages=[{"role": "user", "content": "你好"}]
)
# 换成Claude,其他代码完全不变
response = client.chat.completions.create(
model="anthropic:claude-3-5-sonnet-20240620",
messages=[{"role": "user", "content": "你好"}]
)
# 换成Gemini
response = client.chat.completions.create(
model="google:gemini-1.5-flash",
messages=[{"role": "user", "content": "你好"}]
)
只需要改 model="xxx" 这一个参数,其他代码完全一样。老实讲,这种设计思路我觉得挺聪明的……你不用为了换一家模型就重写一大堆适配代码,省事儿多了。话说回来,这种统一接口的思路在现在百花齐放的大模型时代,确实挺实用的对吧?
支持哪些模型提供商
- OpenAI(GPT系列)
- Anthropic(Claude系列)
- Google(Gemini系列)
- Mistral
- Hugging Face
- AWS Bedrock
- Cohere
- Ollama(本地模型)
- OpenRouter
具体每个提供商支持哪些模型版本,建议去官方文档确认。
实测:三个国内模型全部跑通
我用三个不同的国内模型接口做了测试,全部成功。
测试环境:
- 安装:pip install aisuite
- 测试模型:GLM 5.1、Kimi K2.6、MiMo v2.5
- 测试内容:让AI用一句话介绍自己
| 模型 | 接口格式 | 状态 | 回复 |
|---|---|---|---|
| GLM 5.1 | OpenAI兼容 | ✅ 成功 | "我是一个友好的助手,会用简短的中文为你解答问题!" |
| Kimi K2.6 | OpenAI兼容 | ✅ 成功 | "你好!我是一个由月之暗面开发的人工智能助手..." |
| MiMo v2.5 | Anthropic兼容 | ✅ 成功 | "我是MiMo-v2.5,由小米大模型Core团队开发..." |
关键发现:
- OpenAI兼容接口:大部分国内模型都支持,配置 OPENAI_BASE_URL 和 OPENAI_API_KEY 即可
- Anthropic兼容接口:MiMo v2.5 需要用 Anthropic 格式,配置 ANTHROPIC_BASE_URL 和 ANTHROPIC_API_KEY
- 流式输出:部分接口默认只支持流式模式,需要加 stream=True
- 切换模型:确实只需要改 model="xxx" 这一个参数
实测代码示例:
import aisuite as ai
import os
os.environ["OPENAI_BASE_URL"] = "你的接口地址"
os.environ["OPENAI_API_KEY"] = "你的密钥"
client = ai.Client()
response = client.chat.completions.create(
model="openai:glm-5.1",
messages=[{"role": "user", "content": "你好"}],
stream=True
)
MiMo v2.5 的特殊配置:
MiMo 支持两种接口格式:
- OpenAI 格式:/v1/chat/completions,返回 content
- Anthropic 格式:/anthropic/v1/messages,返回 content + reasoning_content(思考过程)
用 aisuite 调用 MiMo 时,Anthropic 格式兼容性更好:
os.environ["ANTHROPIC_BASE_URL"] = "https://token-plan-cn.xiaomimimo.com/anthropic"
os.environ["ANTHROPIC_API_KEY"] = "你的密钥"
response = client.chat.completions.create(
model="anthropic:mimo-v2.5",
messages=[{"role": "user", "content": "你好"}]
)
和同类工具有什么区别
有人可能会问了——市面上不是早就有类似的东西了吗?比如 CC Switch,对吧。
aisuite 是给开发者用的 Python 库,写代码的时候直接 import 进来就能调。它的核心就一句话:用统一接口去调不同的模型,你一行代码就能在模型之间切来切去,不用改一堆配置啥的。
CC Switch 呢,是个桌面端的 GUI 工具,专门管 Claude Code、Codex、Gemini CLI 这些 AI 编程工具的参数配置。核心卖点是帮你统一管理各种工具 API 设置,点两下按钮就能切配置。说白了就是个"工具箱管理器"。
打个比方你就懂了。
- aisuite 更像是万能充电线,什么手机都能插上就充,一根搞定
- CC Switch 更像是多口插座管理器,帮你把一堆充电器的插头整整齐齐归拢在一起,不用每次满桌子找线
两者其实完全不冲突,说实话它俩面向的人群都不一样。aisuite 是给"写代码的人"用的,CC Switch 是给"用 AI 编程工具的人"用的。你看,各管各的活嘛……
```html
还有这些功能
aisuite 除了基础的对话接口,还支持:
- 工具调用(Tool Calling):给 AI 模型提供自定义的 Python 函数作为"工具",AI 需要时会自动调用
- Agents API:构建更复杂的智能体,让 AI 自动完成多步骤任务。我觉得这个玩法挺多的,比如自动跑完一个流程,不用你一步步盯着
- MCP 协议支持:支持对接 MCP 服务器的工具
- OpenCoworker 桌面助手:附带了一个桌面 AI 助手,有 macOS 和 Windows 版本,数据完全本地。老实讲,本地数据这点挺让人安心的
注意:这些功能基于 README 描述,建议自己跑一下验证。
```
项目数据
- GitHub Stars:14,578
- Forks:1,520
- 开源协议:MIT
- 主要语言:Python
数据截止到2026年6月16日,GitHub上实时变动。
适合什么人用
- 同时用多个AI模型的开发者,不用写一堆适配代码了
- 需要对比模型效果的人,切换模型只需改一个字符串
- 做AI产品的人,方便做A/B测试
- 注重隐私的人,支持Ollama本地模型,数据不出本机
- 想快速上手AI开发的新手,一套接口学会,所有模型通用
```html
常见问题
aisuite支持哪些编程语言?
目前只支持 Python。如果你日常用其他语言搞 AI 开发的话,暂时用不了。说实话这点挺遗憾的,希望后面会慢慢加上吧。
国内模型怎么接入aisuite?
大部分国内模型都提供了 OpenAI 兼容接口,你只要配置一下 OPENAI_BASE_URL 和 OPENAI_API_KEY 就行。不过也有例外——比如 MiMo v2.5 这种,需要走 Anthropic 格式,稍微绕一下。你看,是不是也没想象中那么麻烦?
aisuite和LangChain有什么区别?
LangChain 是个很重的 AI 应用开发框架,功能比 aisuite 多了去了。aisuite 做的事其实就一件:把各种模型的调用接口统一一下。两者不冲突,你完全可以配合着用。
我觉得 aisuite 真正解决了一个让人头疼的问题——AI 模型越来越多,各家接口乱七八糟的,开发者光适配就够累的了。老实讲,如果你平时用 Python 做 AI 开发,又同时对接好几个模型,这个库确实能帮你省不少事。
```