Update graphgen/models/reader/csv_reader.py

ChenZiHong-Gavin · Copilot · web-flow · commit c8c697916ca6 · 2025-10-23T11:59:20.000+08:00
Co-authored-by: Copilot &lt;175728472+Copilot@users.noreply.github.com&gt;
diff --git a/graphgen/models/reader/csv_reader.py b/graphgen/models/reader/csv_reader.py
@@ -10,7 +10,7 @@ def read(self, file_path: str) -> List[Dict[str, Any]]:
 
         df = pd.read_csv(file_path)
         for _, row in df.iterrows():
-            if row.get("type") == "text" and self.text_column not in row:
+            if "type" in row and row["type"] == "text" and self.text_column not in row:
                 raise ValueError(
                     f"Missing '{self.text_column}' in document: {row.to_dict()}"
                 )

Original file line number	Diff line number	Diff line change
`@@ -10,7 +10,7 @@ def read(self, file_path: str) -> List[Dict[str, Any]]:`
`10`	`10`
`11`	`11`	`df = pd.read_csv(file_path)`
`12`	`12`	`for _, row in df.iterrows():`
`13`		`- if row.get("type") == "text" and self.text_column not in row:`
	`13`	`+ if "type" in row and row["type"] == "text" and self.text_column not in row:`
`14`	`14`	`raise ValueError(`
`15`	`15`	`f"Missing '{self.text_column}' in document: {row.to_dict()}"`
`16`	`16`	`)`