stream

Author: glasnt

alloydb/notebooks/embeddings_batch_processing.ipynb

@@ -594,9 +594,9 @@
     "async def debug_sql(pool):\n",
     "    # Check table schema\n",
     "    async with pool.connect() as db_conn:\n",
-    "        result = await db_conn.execute(sqlalchemy.text(\"\"\"\n",
+    "        result = await db_conn.stream(sqlalchemy.text(\"\"\"\n",
     "            select table_name, column_name, data_type, character_maximum_length, column_default, is_nullable\n",
-    "            from INFORMATION_SCHEMA.COLUMNS where table_name = '{table_name}' order by column_name;\"\"\")).fetchall()\n",
+    "            from INFORMATION_SCHEMA.COLUMNS where table_name = '{table_name}' order by column_name;\"\"\"))\n",
     "        await db_conn.commit()\n",
     "\n",
     "        for row in result:\n",