From 7e550df6d4fb09f938545640f657c207e68307b4 Mon Sep 17 00:00:00 2001
From: Kayvane Shakerifar <42403093+kayvane1@users.noreply.github.com>
Date: Sun, 12 Mar 2023 16:29:27 +0000
Subject: [PATCH] =?UTF-8?q?feat:=20add=20lookup=20index=20to=20csv=20loade?=
 =?UTF-8?q?r=20to=20make=20retrieving=20the=20original=20=E2=80=A6=20(#161?=
 =?UTF-8?q?2)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

feat: add lookup index to csv loader to make retrieving the original csv
information easier using theDocument properties
---
 langchain/document_loaders/csv.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/langchain/document_loaders/csv.py b/langchain/document_loaders/csv.py
index af98db46..1108730f 100644
--- a/langchain/document_loaders/csv.py
+++ b/langchain/document_loaders/csv.py
@@ -34,13 +34,13 @@ class CSVLoader(BaseLoader):
 
         with open(self.file_path, newline="") as csvfile:
             csv = DictReader(csvfile, **self.csv_args)  # type: ignore
-            for row in csv:
+            for i, row in enumerate(csv):
                 docs.append(
                     Document(
                         page_content="\n".join(
                             f"{k.strip()}: {v.strip()}" for k, v in row.items()
                         ),
-                        metadata={"source": self.file_path},
+                        metadata={"source": self.file_path, "row": i},
                     )
                 )