From 701bbfef587efec86f576c6071da3614421f5e05 Mon Sep 17 00:00:00 2001
From: Shivang Patel <30023891+shivangp76@users.noreply.github.com>
Date: Wed, 17 Apr 2024 13:46:29 -0400
Subject: [PATCH] Added support for more SageTeX environments

---
 autoload/vimtex/syntax/p/sagetex.vim | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/autoload/vimtex/syntax/p/sagetex.vim b/autoload/vimtex/syntax/p/sagetex.vim
index a413f3d659..8cf9e1b3e8 100644
--- a/autoload/vimtex/syntax/p/sagetex.vim
+++ b/autoload/vimtex/syntax/p/sagetex.vim
@@ -18,11 +18,14 @@ function! vimtex#syntax#p#sagetex#load(cfg) abort " {{{1
         \})
   call vimtex#syntax#core#new_opt('texSagetexOpt', {'next': 'texSagetexArg'})
 
-  call vimtex#syntax#core#new_env({
-        \ 'name': 'sagesilent',
-        \ 'region': 'texSagetexZone',
-        \ 'contains': '@vimtex_nested_python'
-        \})
+  for sage_env in ['sageblock', 'sagesilent', 'sageverbatim', 'sageexample', 'sagecommandline']
+      let config = {
+      \ 'name': sage_env,
+      \ 'region': 'texSagetexZone',
+      \ 'contains': '@vimtex_nested_python'
+      \}
+      call vimtex#syntax#core#new_env(config)
+  endfor
 
   highlight def link texCmdSagetex texCmd
 endfunction