PR feedback

Author: treysp

sqlmesh/__init__.py

@@ -55,6 +55,7 @@ class RuntimeEnv(str, Enum):
     GOOGLE_COLAB = "google_colab"  # Not currently officially supported
     JUPYTER = "jupyter"
     DEBUGGER = "debugger"
+    CI = "ci"
     NON_INTERACTIVE = "non_interactive"  # CI or other envs that shouldn't use emojis
 
     @classmethod
@@ -66,13 +67,10 @@ def get(cls) -> RuntimeEnv:
         """
         runtime_env_var = os.getenv("SQLMESH_RUNTIME_ENVIRONMENT")
         if runtime_env_var:
-            runtime_env_var = runtime_env_var.lower().strip().replace(" ", "").replace("-", "_")
-            runtime_env_var = "non_interactive" if runtime_env_var == "ci" else runtime_env_var
-            runtime_env_var = "debugger" if "debug" in runtime_env_var else runtime_env_var
             try:
                 return RuntimeEnv(runtime_env_var)
             except ValueError:
-                valid_values = [f'"{member.value}"' for member in RuntimeEnv] + ['"ci"']
+                valid_values = [f'"{member.value}"' for member in RuntimeEnv]
                 raise ValueError(
                     f"Invalid SQLMESH_RUNTIME_ENVIRONMENT value: {runtime_env_var}. Must be one of {', '.join(valid_values)}."
                 )
@@ -112,6 +110,10 @@ def is_jupyter(self) -> bool:
     def is_google_colab(self) -> bool:
         return self == RuntimeEnv.GOOGLE_COLAB
 
+    @property
+    def is_ci(self) -> bool:
+        return self == RuntimeEnv.CI
+
     @property
     def is_non_interactive(self) -> bool:
         return self == RuntimeEnv.NON_INTERACTIVE

sqlmesh/core/console.py

@@ -3512,6 +3512,7 @@ def create_console(
         RuntimeEnv.TERMINAL: TerminalConsole,
         RuntimeEnv.GOOGLE_COLAB: NotebookMagicConsole,
         RuntimeEnv.DEBUGGER: DebuggerTerminalConsole,
+        RuntimeEnv.CI: MarkdownConsole,
         RuntimeEnv.NON_INTERACTIVE: MarkdownConsole,
     }
     rich_console_kwargs: t.Dict[str, t.Any] = {"theme": srich.theme}