我們使用智譜AI來(lái)做演示。智譜AI,實(shí)際上使用起來(lái)較為簡(jiǎn)單,用戶需要去注冊(cè)賬號(hào):智譜AI。
實(shí)際上,開(kāi)發(fā)也是很簡(jiǎn)單的:入門開(kāi)發(fā)。
注意需要獲取到API Key,在開(kāi)發(fā)的時(shí)候,可以將API Key放到文件中,通過(guò)環(huán)境變量方式加載:
from dotenv import load_dotenv
import os
load_dotenv()
client = ZhipuAI(api_key=os.getenv("ZHIPU_API_KEY"))
在環(huán)境變量文件.env中如下:
ZHIPU_API_KEY={your-api-key}
具體的問(wèn)答代碼也比較簡(jiǎn)單,這里我們不考慮多輪多話的場(chǎng)景,只表達(dá)一個(gè)最簡(jiǎn)單的功能,復(fù)雜的功能用戶可以自己實(shí)現(xiàn):
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)
這樣一個(gè)簡(jiǎn)單的對(duì)話機(jī)器人就實(shí)現(xiàn)了。