Only test for output redirect for SQL cells

Author: kesmit13

singlestoredb/magics/run_personal.py

@@ -108,12 +108,12 @@ def get_cells() -> Any:
                     if cell.cell_type == 'code':
                         if not cell.source.strip():
                             continue
-                        output_redirect = getattr(
-                            cell, 'metadata', {},
-                        ).get('output_variable', '') or ''
-                        if output_redirect:
-                            output_redirect = f' {output_redirect} <<'
                         if getattr(cell, 'metadata', {}).get('language', '') == 'sql':
+                            output_redirect = getattr(
+                                cell, 'metadata', {},
+                            ).get('output_variable', '') or ''
+                            if output_redirect:
+                                output_redirect = f' {output_redirect} <<'
                             yield f'%%sql{output_redirect}\n{cell.source}'
                         else:
                             yield cell.source

singlestoredb/magics/run_shared.py

@@ -105,12 +105,12 @@ def get_cells() -> Any:
                     if cell.cell_type == 'code':
                         if not cell.source.strip():
                             continue
-                        output_redirect = getattr(
-                            cell, 'metadata', {},
-                        ).get('output_variable', '') or ''
-                        if output_redirect:
-                            output_redirect = f' {output_redirect} <<'
                         if getattr(cell, 'metadata', {}).get('language', '') == 'sql':
+                            output_redirect = getattr(
+                                cell, 'metadata', {},
+                            ).get('output_variable', '') or ''
+                            if output_redirect:
+                                output_redirect = f' {output_redirect} <<'
                             yield f'%%sql{output_redirect}\n{cell.source}'
                         else:
                             yield cell.source