吴恩达出手,开源最新Python包,一个接口调用OpenAI等模型

图灵汇官网

机器学习相关的最新动态显示,一款名为 aisuite 的工具正在帮助开发者简化与多个大模型提供商的集成工作。这款工具由知名人工智能专家、斯坦福大学教授吴恩达推出,旨在通过一致的代码框架调用 OpenAI、Anthropic 和 Google 等公司的大模型,从而实现快速的模型切换和性能对比。

最近,吴恩达在社交媒体上分享了这个新项目的详情,并开源了一款名为 aisuite 的 Python 工具包。借助这个工具,开发者可以更便捷地利用来自不同提供商的语言模型。

aisuite 的核心功能在于简化集成流程。吴恩达指出,在开发应用的过程中,他遇到了跨多个提供商集成的困难,于是开发了这款工具。用户只需调整一个字符串参数(比如 openai:gpt-4o 或 anthropic:claude-3-5-sonnet-20241022 等),即可切换不同的模型。

aisuite 是一个轻量级的封装工具,基于 Python 构建,允许用户在不改动原有代码的情况下轻松切换和测试不同提供商的模型响应。目前,它专注于聊天补全功能,未来还会拓展到更多场景。目前支持的提供商包括 OpenAI、Anthropic、Azure、Google、AWS、Groq、Mistral、HuggingFace 和 Ollama。为了保证稳定运行,aisuite 通过 HTTP 接口或 SDK 与各提供商通信。

安装 aisuite 有几种方式。若只需安装基础包,可以运行以下命令:

pip install aisuite

如果需要支持 Anthropic,可以使用以下命令:

pip install 'aisuite [anthropic]'

要安装全部提供商相关的库,则输入:

pip install 'aisuite [all]'

在使用时,需要从各提供商处获取 API 密钥,并将其设置为环境变量。具体操作可以参考 aisuite 的示例文件夹。

以下是一个简单的代码示例,展示了如何使用 aisuite 获取 GPT-4o 和 Claude-3-5-Sonnet 的聊天补全结果:

```

设置 API 密钥

export OPENAIAPIKEY="your-openai-api-key" export ANTHROPICAPIKEY="your-anthropic-api-key"

使用 Python 客户端

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) ```

如今,随着大模型技术的普及,拥有一个统一接口来调用各类模型对开发者来说意义重大,不仅降低了开发难度,还减少了成本投入。我们希望未来能见到更多类似的实用开源项目,如果你有推荐,欢迎在评论区分享!

本文来源: 互联网 文章作者: 博览群书