我们使用智谱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)
这样一个简单的对话机器人就实现了。