浅谈人工智能之python调用通义千问API

奔波儿灞爱霸波尔奔2024-09-09 0:04

浅谈人工智能之python调用通义千问API

API-KEY建立

第一步：我们登录阿里云百炼

第二步：点击界面上查看我的API-KEY

第三步：在跳出来的界面中，点击创建API-KEY

第四步：在跳出来的界面中，在描述中输入"test"，然后点击确定

第五步：我们可以看到我们创建的API-KEY

至此，我们创建完成API-KEY。

API调用

第一步：在上述的查看API-KEY界面中，点击查看

第二步：点击操作栏的复制按钮，复制API-KEY

第三步：我们进入阿里云API调用文档界面

第四步：我们在API详情中的场景示例中找到python版本的OpenAI兼容，复制里面的代码

第五步：我们打开pyCharm，然后将代码复制到里面，代码如下

python 复制代码

import openai
from openai import OpenAI
import os

def get_response(messages):
    client = OpenAI(
        # 如果您没有配置环境变量，请在此处用您的API Key进行替换
        api_key="XXXXX",
        # 填写DashScope服务的base_url
        base_url="https://dashscope.aliyuncs.com/compatible-mode/v1",
    )
    completion = client.chat.completions.create(
        model="qwen-max",
        messages=messages
        )
    return completion

messages = [{'role': 'system', 'content': 'You are a helpful assistant.'}]
# 您可以自定义设置对话轮数，当前为3
for i in range(3):
    user_input = input("请输入：")
    # 将用户问题信息添加到messages列表中
    messages.append({'role': 'user', 'content': user_input})
    assistant_output = get_response(messages).choices[0].message.content
    # 将大模型的回复信息添加到messages列表中
    messages.append({'role': 'assistant', 'content': assistant_output})
    print(f'用户输入：{user_input}')
    print(f'模型输出：{assistant_output}')
    print('\n')

上述代码中的api_key为刚上述API-KEY复制出来的内容，model为需求调用的模型类型。

第六步：输入以后，我们运行代码，我们可以在运行输入框看到如下内容

第七步：我们输入想要问答的内容，如上述内容输入为：请你自我介绍下？我们获取的答案如下:

至此，我们就已经通过python调用通义千问的API接口。