python调用通义系列大模型 api 接口

不会飞的哪吒2024-10-13 21:42

示例代码

python调用通义系列大模型 api 接口示例代码

python 复制代码

from openai import OpenAI

api_key = "Your API Key"
base_url = "https://dashscope.aliyuncs.com/compatible-mode/v1"

client = OpenAI(api_key=api_key, base_url=base_url)

completion = client.chat.completions.create(
    model="qwen-turbo",
    messages=[
        {"role": "system", "content": "你是一个聪明且富有创造力的小说作家"},
        {"role": "user",
         "content": "请你作为童话故事大王，写一篇短篇童话故事，故事的主题是要永远保持一颗善良的心，要能够激发儿童的学习兴趣和想象力，同时也能够帮助儿童更好地理解和接受故事中所蕴含的道理和价值观。"}
    ],
    top_p=0.7,
    temperature=0.9
)

print(completion.choices[0].message)

模型概览

通义系列大模型概览

|------------------|-------------------------------------------------------------------------------------------------------|----------|------|
| 模型 | 描述 | 上下文 | 最大输出 |
| qwen-turbo | 通义千问系列速度最快、成本很低的模型，适合简单任务。 | 8000 | 2000 |
| qwen-plus | 能力均衡，推理效果、成本和速度介于通义千问-Max和通义千问-Turbo之间，适合中等复杂任务。 | 131072 | 8192 |
| qwen-max | 通义千问系列效果最好的模型，适合复杂、多步骤的任务。 | 8000 | 2000 |
| qwen-long | 支持总结和分析长达千万字的文档，且成本极低。 | 10000000 | 6000 |
| qwen-coder-turbo | 通义千问代码模型。 | 131072 | 8192 |
| qwen-math-plus | 通义千问数学模型是专门用于数学解题的语言模型。 | 4096 | 3072 |
| qwen-audio-turbo | 通义千问Audio是音频理解模型，支持输入多种音频（人类语音、自然音、音乐、歌声）和文本，并输出文本。该模型不仅能对输入的音频进行转录，还具备更深层次的语义理解、情感分析、音频事件检测、语音聊天等能力。 | 8000 | 1500 |
| qwen-vl-max | 通义千问VL是具有视觉（图像）理解能力的文本生成模型，不仅能进行OCR（图片文字识别），还能进一步总结和推理，例如从商品照片中提取属性，根据习题图进行解题等。 | 32000 | 2000 |