定义与自定义文档#

定义文档#

文档可以通过数据加载器自动创建，也可以手动构建。

默认情况下，我们所有的数据加载器（包括LlamaHub提供的加载器）都会通过load_data函数返回Document对象。

from llama_index.core import SimpleDirectoryReader

documents = SimpleDirectoryReader("./data").load_data()

您也可以选择手动构建文档。LlamaIndex提供了Document结构体。

from llama_index.core import Document

text_list = [text1, text2, ...]
documents = [Document(text=t) for t in text_list]

为了加速原型设计和开发，您还可以使用默认文本快速创建文档：

document = Document.example()

自定义文档#

本节介绍自定义Document对象的各种方法。由于Document对象是我们TextNode对象的子类，所有这些设置和细节同样适用于TextNode对象类。

元数据#

文档还提供了包含有用元数据的机会。通过每个文档上的metadata字典，可以包含额外信息来帮助指导响应并追踪查询响应的来源。这些信息可以是任何内容，比如文件名或类别。如果您正在与向量数据库集成，请记住某些向量数据库要求键必须是字符串，值必须是平面类型（str、float或int）。

在文档的metadata字典中设置的任何信息都会显示在从该文档创建的每个源节点的metadata中。此外，这些信息包含在节点中，使索引能够在查询和响应中利用它。默认情况下，元数据会被注入到文本中，用于嵌入和LLM模型调用。

有几种设置这个字典的方法：

在文档构造函数中：

document = Document(
    text="text",
    metadata={"filename": "<doc_file_name>", "category": "<category>"},
)

文档创建后：

document.metadata = {"filename": "<doc_file_name>"}

使用SimpleDirectoryReader和file_metadata钩子自动设置文件名。这将自动在每个文档上运行钩子来设置metadata字段：

from llama_index.core import SimpleDirectoryReader

filename_fn = lambda filename: {"file_name": filename}

# 根据filename_fn自动设置每个文档的元数据
documents = SimpleDirectoryReader(
    "./data", file_metadata=filename_fn
).load_data()

自定义ID#

如文档管理部分所述，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！

document.doc_id = "My new document id!"

注意：ID也可以通过Document对象上的node_id或id_属性设置，类似于TextNode对象。

高级 - 元数据自定义#

上述的一个关键细节是，默认情况下您设置的任何元数据都包含在嵌入生成和LLM中。

自定义LLM元数据文本#

通常，文档可能有许多元数据键，但您可能不希望所有键在响应合成期间对LLM可见。在上面的例子中，我们可能不希望LLM读取文档的file_name。然而，file_name可能包含有助于生成更好嵌入的信息。这样做的一个关键优势是在不改变LLM最终读取内容的情况下，为检索偏置嵌入。

我们可以这样排除它：

document.excluded_llm_metadata_keys = ["file_name"]

然后，我们可以使用get_content()函数并指定MetadataMode.LLM来测试LLM实际会读取的内容：

from llama_index.core.schema import MetadataMode

print(document.get_content(metadata_mode=MetadataMode.LLM))

自定义嵌入元数据文本#

与自定义LLM可见的元数据类似，我们也可以自定义嵌入模型可见的元数据。在这种情况下，您可以专门排除对嵌入模型可见的元数据，以防您不希望特定文本偏置嵌入。

document.excluded_embed_metadata_keys = ["file_name"]

然后，我们可以使用get_content()函数并指定MetadataMode.EMBED来测试嵌入模型实际会读取的内容：

from llama_index.core.schema import MetadataMode

print(document.get_content(metadata_mode=MetadataMode.EMBED))

自定义元数据格式#

如您所知，当发送到LLM或嵌入模型时，元数据被注入到每个文档/节点的实际文本中。默认情况下，此元数据的格式由三个属性控制：

Document.metadata_seperator -> 默认 = "\n"

当连接元数据的所有键/值字段时，此字段控制每个键/值对之间的分隔符。

Document.metadata_template -> 默认 = "{key}: {value}"

此属性控制元数据中每个键/值对的格式。需要两个字符串键key和value。

Document.text_template -> 默认 = {metadata_str}\n\n{content}

一旦您的元数据使用metadata_seperator和metadata_template转换为字符串，此模板控制该元数据与文档/节点的文本内容连接时的外观。需要metadata和content字符串键。

总结#

了解所有这些后，让我们使用所有这些功能创建一个简短的示例：

from llama_index.core import Document
from llama_index.core.schema import MetadataMode

document = Document(
    text="This is a super-customized document",
    metadata={
        "file_name": "super_secret_document.txt",
        "category": "finance",
        "author": "LlamaIndex",
    },
    excluded_llm_metadata_keys=["file_name"],
    metadata_seperator="::",
    metadata_template="{key}=>{value}",
    text_template="Metadata: {metadata_str}\n-----\nContent: {content}",
)

print(
    "The LLM sees this: \n",
    document.get_content(metadata_mode=MetadataMode.LLM),
)
print(
    "The Embedding model sees this: \n",
    document.get_content(metadata_mode=MetadataMode.EMBED),
)

高级 - 自动元数据提取#

我们有初步示例展示如何使用LLM本身执行元数据提取。