Read workflow runner from config

* Only count hybrid/cloud as valid options

GitOrigin-RevId: e1f3af88ce923c336233ffd7ad82012d16229fbc

src/gretel_client/cli/workflows.py

@@ -32,6 +32,13 @@ def get_workflows_api() -> WorkflowsApi:
     return get_session_config().get_v1_api(WorkflowsApi)
 
 
+def determine_runner_mode() -> str:
+    default_runner = get_session_config().default_runner
+    if not default_runner:
+        return RunnerMode.RUNNER_MODE_CLOUD
+    return RunnerMode.from_str(default_runner)
+
+
 @workflows.command(help="Create a new workflow.")
 @click.option("--name", metavar="NAME", help="Workflow name.")
 @project_option
@@ -45,7 +52,7 @@ def get_workflows_api() -> WorkflowsApi:
     "--runner_mode",
     metavar="NAME",
     help="The RunnerMode to use by default when running this workflow.",
-    default=RunnerMode.RUNNER_MODE_CLOUD.value,
+    default=determine_runner_mode,
 )
 @pass_session
 def create(
@@ -55,7 +62,6 @@ def create(
     project: str,
     runner_mode: str = RunnerMode.RUNNER_MODE_CLOUD.value,
 ):
-
     with open(config, encoding="utf-8") as file:
         workflow_config = yaml.safe_load(file)
 

src/gretel_client/workflows/runner_mode.py

@@ -21,4 +21,8 @@ def from_str(
             "hybrid": RUNNER_MODE_HYBRID,
         },
     ) -> RunnerMode:
+        if unstrict not in _string_map:
+            raise ValueError(
+                f"Workflows only supported for 'cloud' or 'hybrid', not '{unstrict}'"
+            )
         return _string_map[unstrict]

tests/gretel_client/test_cli.py

@@ -336,3 +336,29 @@ def test_get_agent_env_var_passing_fails(
     print(cmd.output)
     get_agent_mock.assert_not_called()
     assert cmd.exit_code == 1
+
+
+@patch("gretel_client.cli.workflows.get_session_config")
+def test_workflows_read_from_env_manual(
+    get_session_config: MagicMock,
+    get_project: MagicMock,
+    runner: CliRunner,
+):
+    client_config = ClientConfig()
+    client_config.default_runner = "manual"
+    get_session_config.return_value = client_config
+    cmd = runner.invoke(
+        cli,
+        [
+            "workflows",
+            "create",
+            "--config",
+            "not-there",
+        ],
+    )
+
+    assert (
+        cmd.output
+        == "ERROR: Workflows only supported for 'cloud' or 'hybrid', not 'manual'\n"
+    )
+    assert cmd.exit_code == 1