Python人工智能系列之智谱AI（上）

今天我们来聊聊如何用Python实现人工智能。

我们采用的是智谱AI，所以我们要先在终端（即命令提示符cmd）使用以下指令安装第三方库：

pip install zhipuai

接下来我们要完成的是一个人工智能最最基础也是最最重要的功能——对话。

from zhipuai import ZhipuAI
client = ZhipuAI(api_key="567942ce881bea778ddfef319da37e71.KVjWmLZxxq0DlNlZ")

初始化对话上下文：

context = [
    {"role": "system", "content": "你是一个乐于回答各种问题的小助手，你的任务是提供专业、准确、有洞察力的建议。"},
]

 循环对话，并设置结束条件：

while True:
    question = input("请输入你的问题（或输入'退出'结束对话）:")
    if question.lower() == '退出':
        break

将用户的问题添加到上下文中：

    context.append({"role": "user", "content": question})

生成回答（注：model处可替换成你想要使用的模型名称）：

    response = client.chat.completions.create(
        model="glm-4-flash",
        messages=context,
        stream=True,
    )

打印回答的每一部分：

    for chunk in response:
        print(chunk.choices[0].delta.content, end='')

将AI的回答添加到上下文中，以便进行下一轮对话：

    context.append({"role": "assistant", "content": chunk.choices[0].delta.content})
    print()

完整代码：

from zhipuai import ZhipuAI
client = ZhipuAI(api_key="567942ce881bea778ddfef319da37e71.KVjWmLZxxq0DlNlZ")
context = [
    {"role": "system", "content": "你是一个乐于回答各种问题的小助手，你的任务是提供专业、准确、有洞察力的建议。"},
]
while True:
    question = input("请输入你的问题（或输入'退出'结束对话）:")
    if question.lower() == '退出':
        break
    context.append({"role": "user", "content": question})
    response = client.chat.completions.create(
        model="glm-4-flash",
        messages=context,
        stream=True,
    )
    for chunk in response:
        print(chunk.choices[0].delta.content,end='')
    context.append({"role": "assistant", "content": chunk.choices[0].delta.content})
    print()

不仅如此，有小伙伴还想要AI作图，没问题，这就为你准备！

from zhipuai import ZhipuAI
client = ZhipuAI(api_key="567942ce881bea778ddfef319da37e71.KVjWmLZxxq0DlNlZ")

AI作图就更简单啦！输入作图要求：

yourprompt=input("请输入你想生成图像的描述文字，然后等待图像生成:")

作图（注：model处可替换成你想要使用的模型名称）

response = client.images.generations(
    model="cogview-3-flash",
    prompt=yourprompt,
)

打印结果：

print('请点击此链接',response.data[0].url)

就完成了！真的超简单的！赶紧去试试吧！

完整代码：

from zhipuai import ZhipuAI
client = ZhipuAI(api_key="567942ce881bea778ddfef319da37e71.KVjWmLZxxq0DlNlZ")
yourprompt=input("请输入你想生成图像的描述文字，然后等待图像生成:")
response = client.images.generations(
    model="cogview-3-flash",
    prompt=yourprompt,
)
print('请点击此链接', response.data[0].url)

点赞破50光速更新下一期！

下期预告：AI不仅能作图，还能作视频！！