本文介绍如何基于LlamaIndex框架,调用百炼平台提供的大模型。
前提条件
您已开通百炼服务并获得API-KEY, 请参考获取API Key。
已导入 API-KEY,请参考配置API Key到环境变量。
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。
文档内容是否对您有帮助?