如何用Python调用智谱AI的API

前言

智谱AI大模型以GLM（General Language Model）系列为核心，由清华大学、北京智源人工智能研究院等顶尖机构联合研发。这些模型通过自回归填空任务进行预训练，并采用Transformer架构，能够在各种自然语言理解和生成任务上进行微调，展现出强大的语言处理能力。

其中，GLM-130B是智谱AI在2022年开源的一个重要模型，它基于GLM的进一步开发，在归一化、激活函数、掩码机制等方面进行了优化，成为高精度千亿规模的中英双语语言模型。该模型在斯坦福大学的大模型中心评估中，展现出了与GPT-3 175B相当的表现，成为唯一被选中的亚洲模型，充分证明了其强大的实力。

除了GLM系列，智谱AI还推出了CodeGeeX等多编程语言代码生成预训练模型。CodeGeeX拥有130亿参数，支持Python、C++、Java等多种主流编程语言的代码生成，能够在不同编程语言之间进行准确的代码翻译转换。该模型在代码补全、生成、解释等方面表现出色，为开发者提供了强大的辅助工具。

进入2024年，智谱AI继续迭代升级其大模型系列，发布了GLM-4和CodeGeeX4-ALL-9B等新一代模型。GLM-4在整体性能上较上一代有了大幅提升，支持更强的上下文处理能力和更高的并发推理速度，同时增强了多模态能力，能够自主理解复杂指令并调用多种工具完成任务。CodeGeeX4-ALL-9B则在GLM-4语言能力基础上继续迭代，提高了代码生成能力，支持更多功能如代码解释器、联网搜索等。

这个系列用于记录学习大语言模型的过程，首先从学习智谱AI开始。今天介绍一下如何使用Python调用智谱API接口。

一、注册获取APIkey

首先注册智谱AI：可以领取免费的API额度，注册过程中需要输入手机号，还可以用过支付宝人脸识别得到更高的免费额度。

注册地址：智谱AI开放平台

获取API key：在左边菜单栏中进入API keys找到自己的API key

二、调用接口

接下来按照官方给的接口文档进行接口调用。

接口文档：接口文档

1.安装python包

pip install zhipuai

安装成功后就可以调用了

2.调用对话接口

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)

输出效果

由于官方给的代码一次只能问一个问题，利用while进行反复问答，后面answer的内容用content属性获得，修改后的效果如下图所示。

总结

以上就是今天要讲的内容，本文仅仅简单介绍了怎么用python调用智谱AI的API，当然智谱AI的API不止这么点，还有更多的可以去探究。