手把手搭建一个公众号聊天系统-大模型服务(二)

我们使用智谱AI来做演示。智谱AI,实际上使用起来较为简单,用户需要去注册账号:智谱AI

实际上,开发也是很简单的:入门开发

注意需要获取到API Key,在开发的时候,可以将API Key放到文件中,通过环境变量方式加载:

from dotenv import load_dotenv
import os

load_dotenv()

client = ZhipuAI(api_key=os.getenv("ZHIPU_API_KEY"))

在环境变量文件.env中如下:

ZHIPU_API_KEY={your-api-key}

具体的问答代码也比较简单,这里我们不考虑多轮多话的场景,只表达一个最简单的功能,复杂的功能用户可以自己实现:

def get_completion(prompt, model="glm-4"):
    response = client.chat.completions.create(
        model=model,
        messages=[
            {"role": "user", "content": prompt},
        ],
    )

    return response

response = get_completion("你好")
print(response.choices[0].message)

这样一个简单的对话机器人就实现了。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容