专栏名称: 机器学习算法与Python实战
长期跟踪关注统计学、数据挖掘、机器学习算法、深度学习、人工智能技术与行业发展动态,分享Python、机器学习等技术文章。回复机器学习有惊喜资料。
51好读  ›  专栏  ›  机器学习算法与Python实战

刚刚,吴恩达宣布,推出新的开源 Python 包

机器学习算法与Python实战  · 公众号  ·  · 2024-11-26 11:01

正文

aisuite

大家好,我是玩机器学习的章北海

早上看到吴恩达老师的新推文,他开源了一个最新的 Python 包——aisuite

aisuite 是干什么的呢?一句话总结:

“面向多个生成式人工智能提供商的简单、统一的接口”

aisuite 使开发者能够通过标准化接口轻松使用多个大语言模型(LLM)。它采用类似于 OpenAI 的接口,让开发者能够方便地与最流行的大语言模型进行交互并比较结果。

它是围绕 Python 客户端库的一个轻量级包装,允许创作者在不更改代码的情况下无缝切换和测试来自不同 LLM 提供商的响应。目前,这个库主要专注于聊天补全功能,未来将扩展以涵盖更多用例。

安装与使用

安装

pip install aisuite

设置 api_key

export OPENAI_API_KEY="your-openai-api-key"
export ANTHROPIC_API_KEY="your-anthropic-api-key"

使用

import aisuite as ai
client = ai.Client()

models = ["openai:gpt-4o""anthropic:claude-3-5-sonnet-20240620"]

messages = [
    {"role""system""content""Respond in Pirate English."},
    {"role""user""content""Tell me a joke."},
]

for model in models:
    response = client.chat.completions.create(
        model=model,
        messages=messages,
        temperature=0.75
    )
    print(response.choices[0].message.content)

在调用 create() 时模型名称使用格式“:”。

要获取提供商值列表,可以查看目录 “aisuite/providers/”

当前支持的提供商包括 OpenAI、Anthropic、Azure、Google、AWS、Groq、Mistral、HuggingFace、Ollama 。为了最大限度地提高稳定性, aisuite 使用 HTTP 端点或 SDK 来调用提供商的服务。

说点别的

其实市面上早就大模型中间代理商,提供的也有统一的 api 调用方式,可以直接使用 N 多不同厂商的大模型。

比如我多次推荐的、适合国内使用的 siliconflow

地址: https://cloud.siliconflow.cn?referrer=cly7ai3ir000jqab7qaqp0qwf

它不但完美支持 OpenAI API 调用,还可以免费使用 Qwen、GLM、Yi 等模型。

调用方式:

from openai import OpenAI

client = OpenAI(api_key="YOUR_API_KEY", base_url="https://api.siliconflow.cn/v1")
response = client.chat.completions.create(






请到「今天看啥」查看全文