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