一、获取API
1、打开网址:智谱AI开放平台 注册账号登录
2、登录,查看API key (注册后赠送100万token,实名认证后多赠送400万, 有效期一个)
二、安装及调用
安装质谱SDK
shell
pip install zhipuai
调用方式
-
流式调用
pythonfrom 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)
-
非流式调用
pythonfrom 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接口文档