Z.AI(Zhipu AI)
我們支援 Z.AI GLM 文字/聊天模型,傳送 completion 請求時只要將 zai/ 作為前綴即可
API 金鑰
# env variable
os.environ['ZAI_API_KEY']
使用範例
from litellm import completion
import os
os.environ['ZAI_API_KEY'] = ""
response = completion(
model="zai/glm-4.7",
messages=[
{"role": "user", "content": "hello from litellm"}
],
)
print(response)
使用範例 - 串流
from litellm import completion
import os
os.environ['ZAI_API_KEY'] = ""
response = completion(
model="zai/glm-4.7",
messages=[
{"role": "user", "content": "hello from litellm"}
],
stream=True
)
for chunk in response:
print(chunk)
支援的模型
我們支援所有 Z.AI GLM 模型,傳送 completion 請求時只要將 zai/ 作為前綴即可。
| 模型名稱 | 函式呼叫 | 備註 |
|---|---|---|
| glm-4.7 | completion(model="zai/glm-4.7", messages) | 最新旗艦,200K context,推理 |
| glm-4.6 | completion(model="zai/glm-4.6", messages) | 200K context |
| glm-4.5 | completion(model="zai/glm-4.5", messages) | 128K context |
| glm-4.5v | completion(model="zai/glm-4.5v", messages) | 視覺模型 |
| glm-4.5-x | completion(model="zai/glm-4.5-x", messages) | 高階等級 |
| glm-4.5-air | completion(model="zai/glm-4.5-air", messages) | 輕量級 |
| glm-4.5-airx | completion(model="zai/glm-4.5-airx", messages) | 快速輕量級 |
| glm-4-32b-0414-128k | completion(model="zai/glm-4-32b-0414-128k", messages) | 32B 參數模型 |
| glm-4.5-flash | completion(model="zai/glm-4.5-flash", messages) | 免費等級 |
模型價格
| 模型 | 輸入($/1M tokens) | 輸出($/1M tokens) | 快取輸入($/1M tokens) | 上下文視窗 |
|---|---|---|---|---|
| glm-4.7 | $0.60 | $2.20 | $0.11 | 200K |
| glm-4.6 | $0.60 | $2.20 | - | 200K |
| glm-4.5 | $0.60 | $2.20 | - | 128K |
| glm-4.5v | $0.60 | $1.80 | - | 128K |
| glm-4.5-x | $2.20 | $8.90 | - | 128K |
| glm-4.5-air | $0.20 | $1.10 | - | 128K |
| glm-4.5-airx | $1.10 | $4.50 | - | 128K |
| glm-4-32b-0414-128k | $0.10 | $0.10 | - | 128K |
| glm-4.5-flash | 免費 | 免費 | - | 128K |
與 LiteLLM Proxy 搭配使用
- SDK
- PROXY
from litellm import completion
import os
os.environ['ZAI_API_KEY'] = ""
response = completion(
model="zai/glm-4.7",
messages=[{"role": "user", "content": "Hello, how are you?"}],
)
print(response.choices[0].message.content)
- 設定 config.yaml
model_list:
- model_name: glm-4.7
litellm_params:
model: zai/glm-4.7
api_key: os.environ/ZAI_API_KEY
- model_name: glm-4.5-flash # Free tier
litellm_params:
model: zai/glm-4.5-flash
api_key: os.environ/ZAI_API_KEY
- 執行 proxy
litellm --config config.yaml
- 測試它!
curl -L -X POST 'http://0.0.0.0:4000/v1/chat/completions' \
-H 'Content-Type: application/json' \
-H 'Authorization: Bearer sk-1234' \
-d '{
"model": "glm-4.7",
"messages": [
{
"role": "user",
"content": "Hello, how are you?"
}
]
}'