使用百炼大模型

本文介绍如何基于LlamaIndex框架,调用百炼平台提供的大模型。

前提条件

OPENAI LIKE 方式调用

LlamIndex 的 OpenAI-like 封装可以用于调用百炼支持 OpenAI 兼容的大模型。完整列表请参考:OpenAI 兼容模式支持的模型列表。调用费用、输入输出上限等请参考:模型总览

使用前需要安装 LlamaIndex 核心组件和 OpenAI Like:

pip install llama-index-core
pip install llama-index-llms-openai-like

修改全局设置(可选)

import os
from llama_index.core import Settings
from llama_index.llms.openai_like import OpenAILike

# LlamaIndex默认使用的大模型被替换为百炼
Settings.llm = OpenAILike(
    model="qwen-plus",
    api_base="https://dashscope.aliyuncs.com/compatible-mode/v1",
    api_key=os.getenv("DASHSCOPE_API_KEY"),
    is_chat_model=True
)

大模型调用示例

import os
from llama_index.llms.openai_like import OpenAILike

# 所调用的模型
llm = OpenAILike(
    model="qwen-plus",
    api_base="https://dashscope.aliyuncs.com/compatible-mode/v1",
    api_key=os.getenv("DASHSCOPE_API_KEY"),
    is_chat_model=True
)

response = llm.complete("帮我推荐一下江浙沪5天的旅游攻略。")
print(response)

详细介绍以及完整的 API Reference 请前往 LlamaIndex 官方的 OpenAILike API Reference

DashScope 方式调用

支持百炼所有的文本生成模型,完整列表与调用费用请参考:模型总览。(也支持部署后的模型)

使用前需要安装 LlamaIndex 核心组件和 DashScopeLLM:

pip install llama-index-core
pip install llama-index-llms-dashscope

修改全局设置(可选)

import os
from llama_index.core import Settings
from llama_index.llms.dashscope import DashScope

# LlamaIndex默认使用的大模型被替换为百炼
Settings.llm = DashScope(model_name="qwen-plus", api_key=os.getenv("DASHSCOPE_API_KEY"))

大模型调用示例

import os
from llama_index.llms.dashscope import DashScope

llm = DashScope(model_name="qwen-plus", api_key=os.getenv("DASHSCOPE_API_KEY"))
response = llm.complete("帮我推荐一下江浙沪5天的旅游攻略。")
print(response)

详细介绍与更多示例请前往 LlamaIndex 官方的 DashScope LLM Examples,完整的 API Reference 请前往 LlamaIndex 官方的 DashScope API Reference