Skip to content

Commit c20d0e7

Browse files
committed
fix: parse medium nested source responses
1 parent e14e794 commit c20d0e7

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/notebooklm/types.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -590,7 +590,7 @@ def from_api_response(cls, data: list[Any], notebook_id: str | None = None) -> "
590590
if not data or not isinstance(data, list):
591591
raise ValueError(f"Invalid source data: {data}")
592592

593-
# Try deeply nested format: [[[[id], title, metadata, ...]]]
593+
# Try nested formats: [[[id], title, metadata, ...]] or [[[[id], ...]]]
594594
if isinstance(data[0], list) and len(data[0]) > 0:
595595
if isinstance(data[0][0], list) and len(data[0][0]) > 0:
596596
# Check if deeply nested vs medium nested
@@ -600,8 +600,8 @@ def from_api_response(cls, data: list[Any], notebook_id: str | None = None) -> "
600600
source_id = entry[0][0] if isinstance(entry[0], list) else entry[0]
601601
title = entry[1] if len(entry) > 1 else None
602602
else:
603-
# Medium nested: [[['id'], 'title', ...]]
604-
entry = data[0][0]
603+
# Medium nested: [[[id], 'title', ...]]
604+
entry = data[0]
605605
source_id = entry[0][0] if isinstance(entry[0], list) else entry[0]
606606
title = entry[1] if len(entry) > 1 else None
607607

0 commit comments

Comments
 (0)