ChatGLM是由清华大学唐杰教授和智谱AI共同开发的大语言模型(https://chatglm.cn/chat)。ChatGLM还可以为开发者提供API接口,方便开发者开发各种聊天机器人和智能客服系统。这里我们介绍如何使用ChatGLM开发一个聊天机器人。
首先安装zhipuai,使用命令:pip install zhipuai。
要使用ChatGLM的API,首先需要在智谱AI注册(https://open.bigmodel.cn/)。注册后,在开发工作台的账号管理中,查看API keys,可以自己创建新的API keys(见图2-2)。每位新注册的用户会得到免费的API试用金额。
图2-2 ChatGLM的API keys相关界面
如果我们想更美观地在Jupyter Notebook中输出机器人反馈,这里先自定义一个函数:
接着我们使用chatglm_130b模型,该模型的参数达到1000亿个,相比chatglm_6b的推理能力更强。
接着我们询问这个聊天机器人:how to exit while True loop in Python?下面的截图(见图2-3)给出了ChatGLM的部分回复。正如上面的代码所示,要退出while True,我们的聊天机器人程序已经给出了exit出口。
图2-3 ChatGLM的部分回复