一、获取API
1、打开网址:智谱AI开放平台 注册账号登录
2、登录,查看API key (注册后赠送100万token,实名认证后多赠送400万, 有效期一个)
二、安装及调用
安装质谱SDK
pip install zhipuai
调用方式
-
流式调用
from zhipuai import ZhipuAI client = ZhipuAI(api_key="") # 请填写您自己的APIKey response = client.chat.completions.create(model="glm-4", # 填写需要调用的模型名称messages=[{"role": "user", "content": "你好!你叫什么名字"},],stream=True,) for chunk in response:print(chunk.choices[0].delta)
-
非流式调用
from zhipuai import ZhipuAI client = ZhipuAI(api_key="") # 填写您自己的APIKey response = client.chat.completions.create(model="glm-4", # 填写需要调用的模型名称messages=[{"role": "user", "content": "作为一名营销专家,请为我的产品创作一个吸引人的slogan"},{"role": "assistant", "content": "当然,为了创作一个吸引人的slogan,请告诉我一些关于您产品的信息"},{"role": "user", "content": "智谱AI开放平台"},{"role": "assistant", "content": "智启未来,谱绘无限一智谱AI,让创新触手可及!"},{"role": "user", "content": "创造一个更精准、吸引人的slogan"}], ) print(response.choices[0].message)
具体调用参数可参考官方文档: 智谱AI接口文档