吳恩達(dá)開(kāi)源AISuite，簡(jiǎn)單高效調(diào)用多個(gè)大模型
AISuite是個(gè)Python庫(kù)，能讓你通過(guò)單一接口輕松接入多家生成式AI服務(wù)。今天帶大家一步步了解如何安裝和配置AISuite，并使用它與各種AI模型進(jìn)行高效互動(dòng)。
1 安裝 AISuite
AISuite 托管在 GitHub 上，可以通過(guò) pip 安裝。運(yùn)行以下命令：
pip install aisuite此命令將獲取并安裝該庫(kù)及其依賴項(xiàng)。
2 設(shè)置 API 密鑰
要與生成式 AI 提供商交互，需要從計(jì)劃使用的提供商那里獲取 API 密鑰。AISuite 支持多個(gè)提供商，例如：OpenAI，Anthropic，Azure，Google，AWS，HuggingFace 等等。
以下是設(shè)置密鑰的步驟：
- 在你的項(xiàng)目目錄中創(chuàng)建?
?.env?? 文件。 - 按照以下格式添加你的密鑰：
OPENAI_API_KEY=你的_openai密鑰
ANTHROPIC_API_KEY=你的_anthropic密鑰
AZURE_API_KEY=你的_azure密鑰- AISuite 在執(zhí)行 API 調(diào)用時(shí)會(huì)自動(dòng)讀取這些密鑰。
3 在 Python 腳本中使用 AISuite
以下是一個(gè)基本腳本，幫助你開(kāi)始：
from aisuite import AIClient
# 初始化 AISuite 客戶端
client = AIClient()
# 指定你想要使用的提供商（例如 'openai', 'anthropic'）
response = client.query(
provider='openai',
prompt='法國(guó)的首都是哪里？',
max_tokens=50
)
# 打印響應(yīng)
print(response['output'])此腳本演示了如何：
- 初始化 AISuite 的客戶端。
- 選擇一個(gè)提供商（例如 OpenAI）。
- 向提供商發(fā)送提示。
- 檢索并顯示響應(yīng)。
4 輕松切換不同AI服務(wù)商
使用AISuite，你可以便捷地在不同的AI服務(wù)商之間切換。比如，如果想用Anthropic的Claude模型，可以這樣操作：
response = client.query(
provider='anthropic',
prompt='法國(guó)的首都是哪里？',
max_tokens=50
)
print(response['output'])代碼結(jié)構(gòu)簡(jiǎn)單一致，讓你輕松嘗試多種模型。
5 高級(jí)功能
AISuite 包括高級(jí)功能，例如：
- 模型選擇：你可以在提供商內(nèi)選擇特定的模型，例如：
response = client.query(
provider='openai',
model='gpt-4',
prompt='用簡(jiǎn)單的術(shù)語(yǔ)解釋量子計(jì)算。',
max_tokens=100
)
print(response['output'])- 自定義參數(shù)：調(diào)整溫度、top-p 以及其他生成設(shè)置：
response = client.query(
provider='openai',
model='gpt-3.5-turbo',
prompt='講個(gè)笑話。',
max_tokens=50,
temperature=0.7
)
print(response['output'])- 批量查詢：一次性發(fā)送多個(gè)提示以提高效率。
6 處理響應(yīng)
AISuite 將所有提供商的響應(yīng)標(biāo)準(zhǔn)化為字典格式，通常包括：
- output：生成的文本。
- model：使用的模型。
- provider：提供商名稱。
示例：
response = client.query(provider='openai', prompt='什么是 AI？')
print(f"提供商：{response['provider']}")
print(f"模型：{response['model']}")
print(f"輸出：{response['output']}")7 調(diào)試和日志
AISuite 提供了日志記錄功能，方便調(diào)試。你可以啟用詳細(xì)輸出以跟蹤 API 請(qǐng)求和響應(yīng)：
from aisuite import AIClient
client = AIClient(verbose=True) # 啟用詳細(xì)模式8 結(jié)語(yǔ)
AISuite是與多家生成式AI服務(wù)商合作的得力助手，憑借統(tǒng)一的接口、靈活的特性和用戶友好的設(shè)計(jì)，是你AI工具箱中不可或缺的一部分。想要深入了解更多功能和獲取最新更新，不妨訪問(wèn)其GitHub倉(cāng)庫(kù)。（https://github.com/andrewyng/aisuite）
本文轉(zhuǎn)載自 ??AI科技論談??，作者： AI科技論談
