微信扫码
与创始人交个朋友
我要投稿
Document 和 Node 是 LlamaIndex 中的核心抽象。
Document 是PDF、API 输出或从数据库检索的数据等任何数据源的通用容器。它们可以手动构建,也可以通过 LlamaIndex 提供的各种 Reader 自动创建。默认情况下,Document 存储文本以及一些其他属性:
metadata:可以附加到文本的注释字典。
relationships:包含与其他 Document/Node 的关系的字典。
Node 表示 Document 的一个“块”。与 Document 类似,它们也包含 metadata 和 relationships 属性。
Node 是 LlamaIndex 中的一等公民。我们可以选择直接定义 Node 及其所有属性。我们还可以选择通过 LlamaIndex 的 NodeParser 类将 Document 解析为 Node。默认情况下,从 Document 派生的每个 Node 都会从该 Document 继承相同的 metadata(例如,Document 中归档的“文件名”会传播到每个 Node)。
Document
默认情况下所有的 reader 都会通过 load_data 方法返回 Document 对象。
from llama_index.core import SimpleDirectoryReader
documents = SimpleDirectoryReader("./data").load_data()
或者,我们可以直接创建 Document 对象:
from llama_index.core import Document
text_list = [text1, text2, ...]
documents = [Document(text=t) for t in text_list]
为了加快原型设计和开发速度,我们还可以使用一些默认文本快速创建 Documen:
document = Document.example()
Document 元数据
Document 有一个重要的属性是 metadata,它是以字典形式存储了像是文件名、文件类别等信息在内的元数据。从 Document 中创建的 Node 会继承 metadata 属性。后面在生成嵌入向量和调用 LLM 时也会用到这些元数据。
注意!有的向量数据库要求 metadata 字典中的键必须是字符串,值必须是 str、float 或者 int。
设置 metadata 字典的方法有很多:
1、在 Document 类的构造函数中:
document = Document(text="text",metadata={"filename": "<doc_file_name>", "category": "<category>"},)
2、创建完 Document 对象之后:
document.metadata = {"filename": "<doc_file_name>"}
3、使用 SimpleDirectoryReader 和 file_metadata 钩子自动设置文件名。这样将会在每个 Document 对象上自动运行钩子并设置 metadata 属性:
from llama_index.core import SimpleDirectoryReader
filename_fn = lambda filename: {"file_name": filename}
# automatically sets the metadata of each document according to filename_fn
documents = SimpleDirectoryReader(
"./data", file_metadata=filename_fn
).load_data()
Document ID
doc_id 属性存储了 Document 对象的唯一标识,当我们在 Document 的基础上构建完索引之后,如果后面 Document 内容有更新,那么可以根据 doc_id 去刷新对应的索引。在使用 SimpleDirectoryReader 时,我们可以将 doc_id 设为文件的全路径:
from llama_index.core import SimpleDirectoryReader
documents = SimpleDirectoryReader("./data", filename_as_id=True).load_data()
print([x.doc_id for x in documents])
当然,也可以设置成其他的值:
document.doc_id = "My new document id!"
高级元数据管理
所有的元数据默认情况下都会在生成嵌入向量和调用 LLM 生成答案时被使用。一般情况下元数据字典会包含很多的元数据键,我们可能不希望 LLM 在生成答案时读取所有的元数据,比如文件名。然而文件名对于生成嵌入向量时很有用,因为可能包含重要信息。我们可以直接指定 LLM 不读取的元数据键:
LLM 只能看到 category 元数据和文件内容了。
我们也可以定制在生成嵌入向量阶段要屏蔽的元数据键:
现在我们已经知道元数据和文本内容都会被发送到嵌入模型和 LLM,那么它们是以什么格式组织的呢?我们可以通过 Document 的以下三个属性得到:
这些属性都可以更改,以上只是默认值。
Node
Node 是 LlamaIndex 中的一等公民。Node 代表 Document 的“块”(文本块、图像等等)。它们还包含元数据以及与其他 Node 和索引结构的关系信息。
我们可以选择通过 NodeParser 类将 Document 解析为 Node:
from llama_index.core.node_parser import SentenceSplitter
parser = SentenceSplitter()
nodes = parser.get_nodes_from_documents(documents)
我们可以选择直接定义 Node 及其所有属性:
from llama_index.core.schema import TextNode, NodeRelationship, RelatedNodeInfo
node1 = TextNode(text="<text_chunk>", id_="<node_id>")
node2 = TextNode(text="<text_chunk>", id_="<node_id>")
# set relationships
node1.relationships[NodeRelationship.NEXT] = RelatedNodeInfo(
node_id=node2.node_id
)
node2.relationships[NodeRelationship.PREVIOUS] = RelatedNodeInfo(
node_id=node1.node_id
)
nodes = [node1, node2]
RelatedNodeInfo 类也可以包含必要的 metadata 信息:
node2.relationships[NodeRelationship.PARENT] = RelatedNodeInfo(node_id=node1.node_id, metadata={"key": "val"})
每个 Node 都有一个 node_id 属性,如果没有手动指定,该属性会自动生成。这个 ID 可用于多种目的:能够更新存储中的 Node、能够定义 Node 之间的关系(通过 IndexNode)等等。
print(node.node_id)node.node_id = "My new node_id!"
53AI,企业落地应用大模型首选服务商
产品:大模型应用平台+智能体定制开发+落地咨询服务
承诺:先做场景POC验证,看到效果再签署服务协议。零风险落地应用大模型,已交付160+中大型企业
2024-12-17
llamaindex实战-ChatEngine-Context(上下文)模式
2024-12-01
LlamaIndex,让AI唤醒你的数据
2024-11-29
llamaindex实战-Agent-自定义工具函数
2024-11-22
llamaindex实战-Agent-让Agent调用多个工具函数(本地部署)
2024-11-19
llamaindex实战-Workflow:工作流入门(本地部署断网运行)
2024-11-15
llamaindex实战-Agent-在Agent中使用RAG查询(本地部署)
2024-11-07
深度解析 REAcT Agent 的实现:利用 LlamaIndex 和 Gemini 提升智能代理工作流
2024-11-04
手把手教你用Coze零代码搭建一个智能搜索智能体,高时效性、保姆级!
2024-07-09
2024-04-20
2024-06-05
2024-04-25
2024-04-28
2024-05-09
2024-07-20
2024-04-26
2024-06-19
2024-04-08