从大量文本中解锁准确且富有洞察力的答案是大型语言模型 (LLM) 所实现的一项令人兴奋的功能。在构建 LLM 应用程序时,通常需要连接和查询外部数据源以为模型提供相关上下文。一种流行的方法是使用检索增强生成(RAG)来创建问答系统,该系统可以理解复杂的信息并对查询提供自然的响应。 RAG 允许模型利用庞大的知识库,并为聊天机器人和企业搜索助手等应用程序提供类似人类的对话。
在这篇文章中,我们将探讨如何利用 骆驼指数, 骆驼 2-70B-聊天及 浪链 构建强大的问答应用程序。借助这些最先进的技术,您可以提取文本语料库、索引关键知识并生成准确、清晰地回答用户问题的文本。
骆驼 2-70B-聊天
Llama 2-70B-Chat 是一个强大的法学硕士,可与领先模型竞争。它经过了 2022 万亿个文本标记的预训练,Meta 打算将其用于为用户提供聊天帮助。预训练数据来源于公开数据,截止日期为 2023 年 XNUMX 月,微调数据截止日期为 XNUMX 年 XNUMX 月。有关模型训练过程、安全注意事项、学习内容和预期用途的更多详细信息,请参阅论文 Llama 2:开放基础和微调的聊天模型。 Llama 2 型号可在 亚马逊SageMaker JumpStart 实现快速、简单的部署。
骆驼指数
骆驼指数 是一个可以构建 LLM 应用程序的数据框架。它提供的工具提供数据连接器,以各种来源和格式(PDF、文档、API、SQL 等)摄取现有数据。无论您的数据存储在数据库还是 PDF 中,LlamaIndex 都可以让您轻松地将这些数据用于法学硕士。正如我们在这篇文章中所演示的,LlamaIndex API 使数据访问变得毫不费力,并使您能够创建强大的自定义 LLM 应用程序和工作流程。
如果您正在尝试和构建 LLM,您可能会熟悉 LangChain,它提供了一个强大的框架,简化了 LLM 支持的应用程序的开发和部署。与LangChain类似,LlamaIndex提供了许多工具,包括数据连接器、数据索引、引擎和数据代理,以及工具和可观察性、追踪性和评估性等应用集成。 LlamaIndex 致力于弥合数据和强大的法学硕士之间的差距,通过用户友好的功能简化数据任务。 LlamaIndex 专为构建搜索和检索应用程序(例如 RAG)而设计和优化,因为它提供了一个用于查询 LLM 和检索相关文档的简单界面。
解决方案概述
在这篇文章中,我们演示如何使用 LlamaIndex 和 LLM 创建基于 RAG 的应用程序。下图显示了以下各节中概述的该解决方案的分步架构。
RAG 将信息检索与自然语言生成相结合,以产生更有洞察力的响应。出现提示时,RAG 首先搜索文本语料库以检索与输入最相关的示例。在响应生成过程中,模型会考虑这些示例来增强其功能。通过合并相关检索到的段落,与基本生成模型相比,RAG 响应往往更加真实、连贯且与上下文一致。这种检索-生成框架利用了检索和生成的优势,有助于解决纯自回归会话模型可能出现的重复和缺乏上下文等问题。 RAG 引入了一种有效的方法来构建对话代理和人工智能助手,并提供情境化的高质量响应。
构建解决方案包括以下步骤:
- 成立 亚马逊SageMaker Studio 作为开发环境并安装所需的依赖项。
- 从 Amazon SageMaker JumpStart 中心部署嵌入模型。
- 下载新闻稿作为我们的外部知识库。
- 根据新闻稿构建索引,以便能够查询并将其作为附加上下文添加到提示中。
- 查询知识库。
- 使用 LlamaIndex 和 LangChain 代理构建问答应用程序。
这篇文章中的所有代码都可以在 GitHub回购.
先决条件
对于此示例,您需要一个具有 SageMaker 域和适当的 AWS 账户 AWS身份和访问管理 (IAM) 权限。有关帐户设置说明,请参阅 创建一个 AWS 账户。如果您还没有 SageMaker 域,请参阅 Amazon SageMaker 域 概述来创建一个。在这篇文章中,我们使用 亚马逊SageMakerFullAccess 角色。不建议您在生产环境中使用此凭据。相反,您应该创建并使用具有最低权限的角色。您还可以探索如何使用 Amazon SageMaker 角色管理器 直接通过 SageMaker 控制台构建和管理基于角色的 IAM 角色,以满足常见机器学习需求。
此外,您至少需要访问以下实例大小:
- ml.g5.2xlarge 用于部署时的端点使用 抱脸GPT-J 文本嵌入模型
- ml.g5.48xlarge 用于部署 Llama 2-Chat 模型端点时的端点使用
要增加配额,请参阅 请求增加配额.
使用 SageMaker JumpStart 部署 GPT-J 嵌入模型
本部分为您提供部署 SageMaker JumpStart 模型时的两个选项。您可以使用提供的代码进行基于代码的部署,或使用 SageMaker JumpStart 用户界面 (UI)。
使用 SageMaker Python SDK 进行部署
您可以使用 SageMaker Python SDK 来部署 LLM,如以下所示 码 在存储库中可用。完成以下步骤:
- 使用以下命令设置用于部署嵌入模型的实例大小
instance_type = "ml.g5.2xlarge"
- 找到用于嵌入的模型的 ID。在 SageMaker JumpStart 中,它被标识为
model_id = "huggingface-textembedding-gpt-j-6b-fp16"
- 检索预先训练的模型容器并将其部署以进行推理。
成功部署嵌入模型后,SageMaker 将返回模型端点的名称和以下消息:
在 SageMaker Studio 中使用 SageMaker JumpStart 进行部署
要在 Studio 中使用 SageMaker JumpStart 部署模型,请完成以下步骤:
- 在 SageMaker Studio 控制台上,在导航窗格中选择 JumpStart。
- 搜索并选择 GPT-J 6B Embedding FP16 型号。
- 选择部署并自定义部署配置。
- 对于此示例,我们需要一个 ml.g5.2xlarge 实例,这是 SageMaker JumpStart 建议的默认实例。
- 再次选择部署以创建端点。
端点大约需要 5-10 分钟才能投入使用。
部署嵌入模型后,为了使用 LangChain 与 SageMaker API 的集成,您需要创建一个函数来处理输入(原始文本)并使用模型将其转换为嵌入。您可以通过创建一个名为的类来完成此操作 ContentHandler
,它接受输入数据的 JSON,并返回文本嵌入的 JSON: class ContentHandler(EmbeddingsContentHandler).
将模型端点名称传递给 ContentHandler
转换文本并返回嵌入的函数:
您可以在 SDK 的输出或 SageMaker JumpStart UI 的部署详细信息中找到端点名称。
您可以测试一下 ContentHandler
通过输入一些原始文本并运行,函数和端点按预期工作 embeddings.embed_query(text)
功能。您可以使用提供的示例 text = "Hi! It's time for the beach"
或者尝试您自己的文本。
使用 SageMaker JumpStart 部署和测试 Llama 2-Chat
现在您可以部署能够与用户进行交互式对话的模型。在本例中,我们选择 Llama 2-chat 模型之一,该模型通过以下方式识别
该模型需要使用以下命令部署到实时端点 predictor = my_model.deploy()
。 SageMaker 将返回模型的端点名称,您可以将其用于 endpoint_name
稍后引用的变量。
你定义一个 print_dialogue
函数将输入发送到聊天模型并接收其输出响应。有效负载包含模型的超参数,其中包括:
- 最大新令牌数 – 指模型可以在其输出中生成的最大令牌数。
- 顶部_p – 指模型在生成输出时可以保留的代币的累积概率
- 温度 – 指模型生成的输出的随机性。温度大于 0 或等于 1 会增加随机性级别,而温度为 0 将生成最有可能的标记。
您应该根据您的用例选择超参数并对其进行适当的测试。 Llama 系列等型号要求您包含一个附加参数,表明您已阅读并接受最终用户许可协议 (EULA):
要测试模型,请替换输入有效负载的内容部分: "content": "what is the recipe of mayonnaise?"
。您可以使用自己的文本值并更新超参数以更好地理解它们。
与嵌入模型的部署类似,您可以使用 SageMaker JumpStart UI 部署 Llama-70B-Chat:
- 在 SageMaker Studio 控制台上,选择 快速启动 在导航窗格中
- 搜索并选择
Llama-2-70b-Chat model
- 接受 EULA 并选择 部署,再次使用默认实例
与嵌入模型类似,您可以通过为聊天模型的输入和输出创建内容处理程序模板来使用 LangChain 集成。在这种情况下,您将输入定义为来自用户的输入,并指示它们受 system prompt
。 该 system prompt
告知模型其在帮助用户处理特定用例方面的作用。
除了上述超参数和自定义属性(EULA 接受)之外,调用模型时还会传递此内容处理程序。您可以使用以下代码解析所有这些属性:
当端点可用时,您可以测试它是否按预期工作。您可以更新 llm("what is amazon sagemaker?")
用你自己的文字。您还需要定义具体的 ContentHandler
使用LangChain调用LLM,如下所示 码 以及以下代码片段:
使用 LlamaIndex 构建 RAG
要继续,请安装 LlamaIndex 以创建 RAG 应用程序。您可以使用 pip 安装 LlamaIndex: pip install llama_index
您首先需要将数据(知识库)加载到 LlamaIndex 上以进行索引。这涉及几个步骤:
- 选择数据加载器:
LlamaIndex 提供了许多可用的数据连接器 骆驼中心 适用于 JSON、CSV 和文本文件等常见数据类型以及其他数据源,允许您摄取各种数据集。在这篇文章中,我们使用 SimpleDirectoryReader
提取一些 PDF 文件,如代码所示。我们的数据样本是两份亚马逊新闻稿的 PDF 版本,位于 新闻稿 我们的代码存储库中的文件夹。加载 PDF 后,您可以看到它们已转换为包含 11 个元素的列表。
除了直接加载文档之外,您还可以隐藏 Document
对象进入 Node
在将对象发送到索引之前。选择发送整个 Document
对象到索引或将文档转换为 Node
索引之前的对象取决于您的具体用例和数据结构。对于长文档,节点方法通常是一个不错的选择,在这种情况下,您想要分解和检索文档的特定部分而不是整个文档。欲了解更多信息,请参阅 文档/节点.
- 实例化加载器并加载文档:
此步骤初始化加载器类和任何所需的配置,例如是否忽略隐藏文件。欲了解更多详情,请参阅 简单目录阅读器.
- 呼叫装载机
load_data
方法来解析源文件和数据并将它们转换为 LlamaIndex Document 对象,准备索引和查询。您可以使用以下代码,利用 LlamaIndex 的索引和检索功能完成全文搜索的数据摄取和准备:
- 建立索引:
LlamaIndex 的关键特性是它能够在数据上构建有组织的索引,这些数据表示为文档或节点。索引有助于有效地查询数据。我们使用默认的内存向量存储和我们定义的设置配置来创建索引。骆驼指数 个人设置 是一个配置对象,为 LlamaIndex 应用程序中的索引和查询操作提供常用资源和设置。它充当单例对象,因此它允许您设置全局配置,同时还允许您通过将特定组件直接传递到使用它们的接口(例如 LLM、嵌入模型)来本地覆盖特定组件。当未明确提供特定组件时,LlamaIndex 框架会回退到中定义的设置 Settings
对象作为全局默认值。将我们的嵌入和 LLM 模型与 LangChain 结合使用并配置 Settings
我们需要安装 llama_index.embeddings.langchain
和 llama_index.llms.langchain
。我们可以配置 Settings
对象如以下代码所示:
默认情况下, VectorStoreIndex
使用内存中的 SimpleVectorStore
它被初始化为默认存储上下文的一部分。在现实生活中的用例中,您经常需要连接到外部向量存储,例如 亚马逊开放搜索服务. 有关详细信息,请参阅 适用于 Amazon OpenSearch Serverless 的矢量引擎.
现在,您可以使用以下命令对文档运行问答 查询引擎 来自 LlamaIndex。为此,请传递您之前为查询创建的索引并提出您的问题。查询引擎是查询数据的通用接口。它采用自然语言查询作为输入并返回丰富的响应。查询引擎通常构建在一个或多个之上 指标 运用 猎犬.
您可以看到 RAG 解决方案能够从提供的文档中检索正确的答案:
使用LangChain工具和代理
Loader
班级。该加载器旨在将数据加载到 LlamaIndex 中或随后作为 浪链代理。这为您提供了更多功能和灵活性,可以将其用作应用程序的一部分。您首先定义您的 工具 来自LangChain代理类。您传递给工具的函数会查询您使用 LlamaIndex 在文档上构建的索引。
然后,您选择要用于 RAG 实施的正确代理类型。在这种情况下,您使用 chat-zero-shot-react-description
代理人。通过此代理,LLM 将使用可用的工具(在本例中为知识库上的 RAG)来提供响应。然后,您可以通过传递工具、LLM 和代理类型来初始化代理:
您可以看到代理正在通过 thoughts
, actions
及 observation
,使用该工具(在这种情况下,查询索引文档);并返回结果:
您可以在随附的中找到端到端的实现代码 GitHub回购.
清理
为了避免不必要的成本,您可以通过以下代码片段或 Amazon JumpStart UI 清理资源。
要使用Boto3 SDK,请使用以下代码删除文本嵌入模型端点和文本生成模型端点以及端点配置:
要使用 SageMaker 控制台,请完成以下步骤:
- 在 SageMaker 控制台的导航窗格中的推理下,选择端点
- 搜索嵌入和文本生成端点。
- 在端点详细信息页面上,选择删除。
- 再次选择删除进行确认。
结论
对于专注于搜索和检索的用例,LlamaIndex 提供了灵活的功能。它擅长为法学硕士建立索引和检索,使其成为深度探索数据的强大工具。 LlamaIndex 使您能够创建有组织的数据索引、使用不同的 LLM、增强数据以获得更好的 LLM 性能以及使用自然语言查询数据。
这篇文章演示了一些关键的 LlamaIndex 概念和功能。我们使用 GPT-J 进行嵌入,并使用 Llama 2-Chat 作为 LLM 来构建 RAG 应用程序,但您可以使用任何合适的模型。您可以探索 SageMaker JumpStart 上提供的全面的模型。
我们还展示了 LlamaIndex 如何提供强大、灵活的工具来连接、索引、检索数据以及与 LangChain 等其他框架集成数据。通过 LlamaIndex 集成和 LangChain,您可以构建更强大、更通用、更有洞察力的 LLM 应用程序。
作者简介
罗米娜·沙里普尔博士 是 Amazon Web Services (AWS) 的高级机器学习和人工智能解决方案架构师。她花了 10 多年的时间,领导通过机器学习和人工智能的进步实现创新的端到端解决方案的设计和实施。 Romina 感兴趣的领域是自然语言处理、大型语言模型和 MLOps。
妮可平托 是澳大利亚悉尼的 AI/ML 专家解决方案架构师。她在医疗保健和金融服务方面的背景使她在解决客户问题方面具有独特的视角。她热衷于通过机器学习为客户提供支持,并为 STEM 领域的下一代女性提供支持。
- SEO 支持的内容和 PR 分发。 今天得到放大。
- PlatoData.Network 垂直生成人工智能。 赋予自己力量。 访问这里。
- 柏拉图爱流。 Web3 智能。 知识放大。 访问这里。
- 柏拉图ESG。 碳, 清洁科技, 能源, 环境, 太阳能, 废物管理。 访问这里。
- 柏拉图健康。 生物技术和临床试验情报。 访问这里。
- Sumber: https://aws.amazon.com/blogs/machine-learning/build-knowledge-powered-conversational-applications-using-llamaindex-and-llama-2-chat/