使用 NVIDIA NeMo Retriever、託管 NIM、LanceDB、重新排序與基於事實生成建立多模態 RAG 管線

In this tutorial, we build an advanced multimodal retrieval-augmented generation pipeline with NVIDIA NeMo Retriever.We begin by configuring a Python 3.12 environment, installing the required packages, and performing offline PDF text extraction without relying on a GPU or external API key.

We then extend the workflow with hosted NVIDIA NIM endpoints to detect page elements, extract tables, charts, and infographics, generate dense vector embeddings, and store the processed content in LanceDB.

Finally, we implement dense retrieval, vision-language reranking, metadata-filtered search, grounded response generation with inline citations, and a lightweight recall-at-k evaluation to validate retrieval quality across multimodal document content.

Copy CodeCopiedUse a different Browserimport sys, os, subprocess, textwrap, json, time, warnings warnings.filterwarnings("ignore") assert sys.version_info[:2] == (3, 12), ( f"nemo-retriever requires Python 3.12.x (found {sys.version.split()[0]})." "Colab's default runtime is 3.

12; if you changed it, switch back." ) def sh(cmd): print(f"$ {cmd}") subprocess.

run(cmd, shell=True, check=False) try: import nemo_retriever print("nemo-retriever already installed") except ImportError: sh("pip install -q --ignore-installed PyJWT nemo-retriever openai") import nemo_retriever print("nemo-retriever version:", nemo_retriever.

__version__) from nemo_retriever import create_ingestor try: from nemo_retriever.io import to_markdown, to_markdown_by_page except ImportError: from nemo_retriever.common.io import to_markdown, to_markdown_by_page try: from nemo_retriever.

retriever import Retriever except ImportError: from nemo_retriever.graph.retriever import Retriever import pandas as pd pd.set_option("display.max_colwidth", 160) DOC = "multimodal_test.pdf" if not os.path.exists(DOC): sh(f"curl -sL -o {DOC} " "https://raw.githubusercontent.

com/NVIDIA/NeMo-Retriever/main/data/multimodal_test.pdf") print("document:", DOC, os.path.getsize(DOC), "bytes") DOCS = [DOC] print("\n=== STAGE 1: offline text extraction (no API key) ===") offline = ( create_ingestor(run_mode="inprocess", allow_no_gpu=True) .files(DOCS) .

extract( extract_text=True, extract_tables=False, extract_charts=False, extract_images=False, extract_infographics=False, use_page_elements=False, extract_page_as_image=False, method="pdfium", ) ) df_offline = offline.ingest() print("rows:", df_offline.shape, "\ncolumns:", list(df_offline.

columns)) print("\npage 1 text preview:\n", df_offline.iloc[0]["text"][:400]) We configure the Python 3.12 environment, install NVIDIA NeMo Retriever, and import the required ingestion and retrieval components.

We download the sample multimodal PDF and define it as the input document for the pipeline.We then perform CPU-based offline text extraction with PDFium and inspect the extracted rows, columns, and page content.Copy CodeCopiedUse a different Browserfrom getpass import getpass if not os.environ.

get("NVIDIA_API_KEY"): try: from google.colab import userdata os.environ["NVIDIA_API_KEY"] = userdata.get("NVIDIA_API_KEY") except Exception: os.environ["NVIDIA_API_KEY"] = getpass("NVIDIA_API_KEY (nvapi-...): ").strip() API_KEY = os.environ.get("NVIDIA_API_KEY", "").strip() HAVE_KEY = API_KEY.

startswith("nvapi-") print("API key present:", HAVE_KEY) PAGE_ELEMENTS_URL = "https://ai.api.nvidia.com/v1/cv/nvidia/nemotron-page-elements-v3" OCR_URL = "https://ai.api.nvidia.com/v1/cv/nvidia/nemotron-ocr-v1" TABLE_STRUCT_URL = "https://ai.api.nvidia.

com/v1/cv/nvidia/nemotron-table-structure-v1" GRAPHIC_ELEM_URL = "https://ai.api.nvidia.com/v1/cv/nvidia/nemotron-graphic-elements-v1" EMBED_URL = "https://integrate.api.nvidia.com/v1/embeddings" RERANK_URL = "https://ai.api.nvidia.

com/v1/retrieval/nvidia/llama-nemotron-rerank-vl-1b-v2/reranking" CHAT_URL = "https://integrate.api.nvidia.com/v1" EMBED_MODEL = "nvidia/llama-nemotron-embed-1b-v2" RERANK_MODEL = "nvidia/llama-nemotron-rerank-vl-1b-v2" LLM_MODEL = "nvidia/llama-3.3-nemotron-super-49b-v1.5" LANCEDB_URI, TABLE = ".

/lancedb", "colab_demo" df = df_offline if HAVE_KEY: print("\n=== STAGE 2: multimodal ingest via hosted NIMs ===") ing = ( create_ingestor( run_mode="inprocess", allow_no_gpu=True, error_policy="collect", ) .files(DOCS) .

extract( extract_text=True, extract_tables=True, extract_charts=True, extract_infographics=True, extract_images=False, method="pdfium", dpi=200, table_output_format="markdown", page_elements_invoke_url=PAGE_ELEMENTS_URL, ocr_invoke_url=OCR_URL, table_structure_invoke_url=TABLE_STRUCT_URL, graphic_elements_invoke_url=GRAPHIC_ELEM_URL, api_key=API_KEY, request_timeout_s=120.

0, split_config={"text": {"max_tokens": 512, "overlap_tokens": 64}}, ) .dedup(content_hash=True, bbox_iou=True, iou_threshold=0.45) .

embed( embedding_endpoint=EMBED_URL, model_name=EMBED_MODEL, embed_model_name=EMBED_MODEL, api_key=API_KEY, input_type="passage", inference_batch_size=16, nim_http_max_concurrent=8, ) .

vdb_upload( vdb_op="lancedb", vdb_kwargs={ "uri": LANCEDB_URI, "table_name": TABLE, "overwrite": True, "create_index": True, "index_type": "IVF_HNSW_SQ", "metric": "l2", }, ) ) t0 = time.time() df = ing.ingest(show_progress=True) print(f"ingested in {time.time()-t0:.1f}s -> {df.

shape}") We securely load the NVIDIA API key and define the hosted NIM endpoints for layout detection, OCR, table extraction, graphic analysis, embedding, reranking, and generation.

We create a multimodal ingestion pipeline that extracts text, tables, charts, and infographics while applying token-aware chunking and content deduplication.We generate embeddings for the extracted content and upload the resulting vectors and metadata to a LanceDB table.

Copy CodeCopiedUse a different Browserprint("\n=== Extraction inspection ===") for col in ["tables", "charts", "infographics", "images"]: if col in df.columns: n = int(df[col].apply(lambda v: len(v) if isinstance(v, (list, tuple)) else 0).

sum()) print(f" {col:<14} {n}") pages = to_markdown_by_page(df) print("\npages rendered to markdown:", list(pages.keys())) print("\n--- page 1 markdown (first 900 chars) ---\n", pages[min(pages)][:900]) full_md = to_markdown(df) if full_md: with open("extracted.md", "w") as f: f.

write(full_md) print("\nfull document markdown -> extracted.

md") if HAVE_KEY: print("\n=== STAGE 3: dense retrieval ===") retriever = Retriever( run_mode="service", top_k=5, rerank=False, vdb_kwargs={"uri": LANCEDB_URI, "table_name": TABLE}, embed_kwargs={ "embedding_endpoint": EMBED_URL, "model_name": EMBED_MODEL, "embed_model_name": EMBED_MODEL, "api_key": API_KEY, "input_type": "query", }, ) QUERIES = [ "Given their activities, which animal is responsible for the typos in my documents?

", "What is the most expensive gadget and how much does it cost?", "Which animal is at the beach?", ] def show(hits, label=""): print(f"\n--- {label} ---") for i, h in enumerate(hits, 1): meta = h.get("metadata") if isinstance(meta, str): try: meta = json.

loads(meta) except Exception: meta = {} page = (meta or {}).get("page_number", "?") score = h.get("_distance", h.get("rerank_score", "")) body = " ".join(str(h.get("text", "")).split())[:180] print(f" {i}.p{page} score={score} {body}") show(retriever.

query(QUERIES[0]), "single query") for q, hits in zip(QUERIES, retriever.queries(QUERIES, top_k=3)): show(hits, q[:60]) We inspect the extracted multimodal elements and convert the processed document into page-level and full-document Markdown.

We configure a dense retriever that embeds user queries and searches the LanceDB vector index for the most relevant document chunks.We test both individual and batched queries while displaying page numbers, similarity scores, and retrieved text previews.

Copy CodeCopiedUse a different Browserif HAVE_KEY: print("\n=== STAGE 4: retrieve + VL rerank ===") reranking = Retriever( run_mode="service", top_k=5, rerank=True, vdb_kwargs={"uri": LANCEDB_URI, "table_name": TABLE}, embed_kwargs={ "embedding_endpoint": EMBED_URL, "model_name": EMBED_MODEL, "emb