Update base.py Reading Fix

run-llama · Mar 1, 2024 · 29d0f53 · 29d0f53
1 parent 903e6f9
commit 29d0f53
Showing 1 changed file with 16 additions and 5 deletions.
diff --git a/...integrations/readers/llama-index-readers-google/llama_index/readers/google/sheets/base.py b/...integrations/readers/llama-index-readers-google/llama_index/readers/google/sheets/base.py
@@ -148,12 +148,23 @@ def _load_sheet_in_pandas(self, spreadsheet_id: str) -> List[pd.DataFrame]:
         sheets = spreadsheet_data.get("sheets")
         dataframes = []
         for sheet in sheets:
-            result = sheet.get("values", [])
-            name = sheet.get("properties", {}).get("title")
-            if not result:
-                print(f"No data found in {name}")
+            properties = sheet.get("properties")
+            title = properties.get("title")
+            grid_props = properties.get("gridProperties")
+            rows = grid_props.get("rowCount")
+            cols = grid_props.get("columnCount")
+            range_pattern = f"{title}!R1C1:R{rows}C{cols}"
+            response = (
+                sheets_service.spreadsheets()
+                .values()
+                .get(spreadsheetId=spreadsheet_id, range=range_pattern)
+                .execute()
+            )
+            values = response.get("values", [])
+            if not values:
+                print(f"No data found in {title}")
             else:
-                df = pd.DataFrame(result[1:], columns=result[0])
+                df = pd.DataFrame(values[1:], columns=values[0])
                 dataframes.append(df)
         return dataframes