Update graphgen/models/reader/jsonl_reader.py

ChenZiHong-Gavin · Copilot · web-flow · commit 30c43db1c241 · 2025-10-21T18:49:05.000+08:00
Co-authored-by: Copilot &lt;175728472+Copilot@users.noreply.github.com&gt;
diff --git a/graphgen/models/reader/jsonl_reader.py b/graphgen/models/reader/jsonl_reader.py
@@ -12,7 +12,7 @@ def read(self, file_path: str) -> List[Dict[str, Any]]:
             for line in f:
                 try:
                     doc = json.loads(line)
-                    if doc["type"] == "text" and self.text_column not in doc:
+                    if doc.get("type") == "text" and self.text_column not in doc:
                         raise ValueError(
                             f"Missing '{self.text_column}' in document: {doc}"
                         )

Original file line number	Diff line number	Diff line change
`@@ -12,7 +12,7 @@ def read(self, file_path: str) -> List[Dict[str, Any]]:`
`12`	`12`	`for line in f:`
`13`	`13`	`try:`
`14`	`14`	`doc = json.loads(line)`
`15`		`- if doc["type"] == "text" and self.text_column not in doc:`
	`15`	`+ if doc.get("type") == "text" and self.text_column not in doc:`
`16`	`16`	`raise ValueError(`
`17`	`17`	`f"Missing '{self.text_column}' in document: {doc}"`
`18`	`18`	`)`