19.Python实现简易 AI 助手

Java朝阳 • 2024年4月5日上午9:24 • 技术漫谈 • 阅读 300

今天学长给大家带来了 Python 对接阿里大模型-通义千问，实现一个简易自制的 AI 助手。

编写代码

要求：实现一个简易的 AI 助手，可以多次提问，它返回对应的结果。

下面是一个简单的 Python 脚本，可以快速实现这个功能，代码如下：

from http import HTTPStatus

import dashscope
from dashscope import Generation
from dashscope.api_entities.dashscope_response import Role

# 此处填上面自己申请的Key
dashscope.api_key = "***"


def conversation_with_messages():
    messages = []
    while True:
        user_input = input("请输入消息（输入'exit'退出）：")
        if user_input.lower() == 'exit':
            break
        messages.append({'role': Role.USER, 'content': user_input})

        response = Generation.call(
            Generation.Models.qwen_turbo,
            messages=messages,
            result_format='message',  # set the result to be "message" format.
        )

        if response.status_code == HTTPStatus.OK:
            # append result to messages.
            messages.append({'role': response.output.choices[0]['message']['role'],
                             'content': response.output.choices[0]['message']['content']})
            print("AI回复：", response.output.choices[0]['message']['content'])
        else:
            print('Request id: %s, Status code: %s, error code: %s, error message: %s' % (
                response.request_id, response.status_code,
                response.code, response.message
            ))


if __name__ == '__main__':
    conversation_with_messages()