Dynamically Adding Document Number in create_stuff_documents_chain

[wo-yashsinghvi]

Checked other resources

• I added a very descriptive title to this question.
• I searched the LangChain documentation with the integrated search.
• I used the GitHub search to find a similar question and didn't find it.

Commit to Help

• I commit to help with one of those options 👆

Example Code

document_prompt = PromptTemplate(
    input_variables=["name", "page_number", "page_content"],
    template="Name: {name}\nPage Number: {page_number}\nContent: {page_content}\n"
)

chain = create_stuff_documents_chain(llm=llm, document_prompt=document_prompt, prompt=main_prompt)

Description

I'm using create_stuff_documents_chain and passing metadata like Name, Page Number, and Content through a document_prompt. Here's an example:

Document:  
- Name: file_4  
- Page Number: 20  
- Content: The plaintiffs are seeking unspecified damages...  

Document:  
- Name: file_1  
- Page Number: 10  
- Content: These forward-looking statements are not guarantees...  

But I also want to dynamically include the document number (e.g., Document 1, Document 2). How can I do this while keeping the code clean and reusable?

Any tips or examples would be greatly appreciated! 🙌

System Info

System Information

OS: Windows
OS Version: 10.0.22631
Python Version: 3.12.2 (tags/v3.12.2:6abddd9, Feb 6 2024, 21:26:36) [MSC v.1937 64 bit (AMD64)]

Package Information

langchain_core: 0.3.29
langchain: 0.3.14
langchain_community: 0.0.31
langsmith: 0.2.10
langchain_aws: 0.2.10
langchain_google_genai: 1.0.1
langchain_milvus: 0.1.7
langchain_openai: 0.1.23
langchain_text_splitters: 0.3.4
langchain_voyageai: 0.1.3

Optional packages not installed

langserve

Other Dependencies

aiohttp: 3.9.3
aiosqlite: Installed. No version info available.
aleph-alpha-client: Installed. No version info available.
anthropic: Installed. No version info available.
arxiv: Installed. No version info available.
assemblyai: Installed. No version info available.
async-timeout: Installed. No version info available.
atlassian-python-api: Installed. No version info available.
azure-ai-documentintelligence: Installed. No version info available.
beautifulsoup4: 4.12.3
bibtexparser: Installed. No version info available.
boto3: 1.35.92
cassio: Installed. No version info available.
chardet: 4.0.0
cloudpickle: Installed. No version info available.
cohere: Installed. No version info available.
databricks-vectorsearch: Installed. No version info available.
dataclasses-json: 0.6.4
datasets: Installed. No version info available.
dgml-utils: Installed. No version info available.
elasticsearch: Installed. No version info available.
esprima: Installed. No version info available.
faiss-cpu: Installed. No version info available.
feedparser: Installed. No version info available.
fireworks-ai: Installed. No version info available.
friendli-client: Installed. No version info available.
geopandas: Installed. No version info available.
gitpython: 3.1.43
google-cloud-documentai: Installed. No version info available.
google-generativeai: 0.7.1
gql: Installed. No version info available.
gradientai: Installed. No version info available.
hdbcli: Installed. No version info available.
hologres-vector: Installed. No version info available.
html2text: Installed. No version info available.
httpx: 0.26.0
javelin-sdk: Installed. No version info available.
jinja2: 3.1.3
jq: Installed. No version info available.
jsonpatch: 1.33
jsonschema: 4.21.1
langsmith-pyo3: Installed. No version info available.
lxml: 5.3.0
markdownify: Installed. No version info available.
motor: Installed. No version info available.
msal: Installed. No version info available.
mwparserfromhell: Installed. No version info available.
mwxml: Installed. No version info available.
newspaper3k: Installed. No version info available.
numexpr: Installed. No version info available.
numpy: 1.26.4
nvidia-riva-client: Installed. No version info available.
oci: Installed. No version info available.
openai: 1.57.3
openapi-pydantic: Installed. No version info available.
oracle-ads: Installed. No version info available.
orjson: 3.10.0
packaging: 23.2
pandas: 2.2.1
pdfminer-six: Installed. No version info available.
pgvector: Installed. No version info available.
pillow: 10.2.0
praw: Installed. No version info available.
premai: Installed. No version info available.
psychicapi: Installed. No version info available.
py-trello: Installed. No version info available.
pydantic: 2.10.4
pymilvus: 2.5.1
pymupdf: Installed. No version info available.
pypdf: 4.1.0
pypdfium2: Installed. No version info available.
pyspark: Installed. No version info available.
PyYAML: 6.0.1
rank-bm25: Installed. No version info available.
rapidfuzz: Installed. No version info available.
rapidocr-onnxruntime: Installed. No version info available.
rdflib: Installed. No version info available.
requests: 2.31.0
requests-toolbelt: 1.0.0
rspace_client: Installed. No version info available.
scikit-learn: 1.4.1.post1
SQLAlchemy: 2.0.29
sqlite-vss: Installed. No version info available.
streamlit: 1.32.2
sympy: 1.12
telethon: Installed. No version info available.
tenacity: 8.2.3
tidb-vector: Installed. No version info available.
tiktoken: 0.7.0
timescale-vector: Installed. No version info available.
tqdm: 4.66.5
tree-sitter: Installed. No version info available.
tree-sitter-languages: Installed. No version info available.
typer: 0.15.1
typing-extensions: 4.12.2
upstash-redis: Installed. No version info available.
vdms: Installed. No version info available.
voyageai: 0.3.2
xata: Installed. No version info available.
xmltodict: Installed. No version info available.
zhipuai: Installed. No version info available.
zstandard: Installed. No version info available.