智谱AI: ChatGLM API的使用

一、获取API

1、打开网址：智谱AI开放平台 注册账号登录

2、登录，查看API key (注册后赠送100万token，实名认证后多赠送400万, 有效期一个)

二、安装及调用

安装质谱SDK

pip install zhipuai

调用方式

1. 流式调用

   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)

2. 非流式调用

   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接口文档