orchestra-cli 0.1.1__tar.gz → 0.1.3__tar.gz

{orchestra_cli-0.1.1 → orchestra_cli-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: orchestra-cli
-Version: 0.1.1
+Version: 0.1.3
 Summary: Orchestra CLI. Perform operations with Orchestra locally.
 Author-email: Orchestra Technologies <support@getorchestra.io>
 Requires-Python: >=3.10
@@ -87,7 +87,8 @@ export ORCHESTRA_API_KEY=...  # required
 
 orchestra import \
   --alias my-pipeline \
-  --path ./pipelines/pipeline.yaml
+  --path ./pipelines/pipeline.yaml \
+  --working-branch my-feature-branch   # optional; defaults to current local branch
 # or
 orchestra-cli import -a my-pipeline -p ./pipelines/pipeline.yaml
 ```
@@ -96,6 +97,7 @@ Options
 
 - `-a, --alias` (required): The alias you want to register the pipeline under.
 - `-p, --path` (required): Path to the YAML file. Must be inside a git repository.
+- `-w, --working-branch` (optional): Branch to associate as the working branch for this pipeline import. If omitted, the current local git branch is used.
 
 Notes
 
@@ -103,6 +105,7 @@ Notes
 - Git details are detected automatically:
   - Supported providers: GitHub, GitLab, Azure DevOps.
   - The default branch is detected from `origin`.
+  - The working branch defaults to your current local branch when not specified.
   - The YAML path is computed relative to the repository root.
 - On success, the command prints the created pipeline ID (or a success message).
 - Exit codes: `0` on success, `1` on failure.

{orchestra_cli-0.1.1 → orchestra_cli-0.1.3}/README.md

@@ -76,7 +76,8 @@ export ORCHESTRA_API_KEY=...  # required
 
 orchestra import \
   --alias my-pipeline \
-  --path ./pipelines/pipeline.yaml
+  --path ./pipelines/pipeline.yaml \
+  --working-branch my-feature-branch   # optional; defaults to current local branch
 # or
 orchestra-cli import -a my-pipeline -p ./pipelines/pipeline.yaml
 ```
@@ -85,6 +86,7 @@ Options
 
 - `-a, --alias` (required): The alias you want to register the pipeline under.
 - `-p, --path` (required): Path to the YAML file. Must be inside a git repository.
+- `-w, --working-branch` (optional): Branch to associate as the working branch for this pipeline import. If omitted, the current local git branch is used.
 
 Notes
 
@@ -92,6 +94,7 @@ Notes
 - Git details are detected automatically:
   - Supported providers: GitHub, GitLab, Azure DevOps.
   - The default branch is detected from `origin`.
+  - The working branch defaults to your current local branch when not specified.
   - The YAML path is computed relative to the repository root.
 - On success, the command prints the created pipeline ID (or a success message).
 - Exit codes: `0` on success, `1` on failure.

{orchestra_cli-0.1.1 → orchestra_cli-0.1.3}/orchestra_cli/src/import_pipeline.py

@@ -52,6 +52,13 @@ def _detect_default_branch(repo_root: Path) -> str | None:
     return None
 
 
+def _detect_current_branch(repo_root: Path) -> str | None:
+    ok, out = run_git_command(["rev-parse", "--abbrev-ref", "HEAD"], repo_root)
+    if ok and out:
+        return out
+    return None
+
+
 def _detect_storage_provider(repository_url: str | None) -> str:
     if not repository_url:
         typer.echo(red("Could not detect storage provider - no repository URL"))
@@ -104,6 +111,12 @@ def import_pipeline(
         resolve_path=True,
         help="Path to pipeline YAML inside a git repository",
     ),
+    working_branch: str | None = typer.Option(
+        None,
+        "--working-branch",
+        "-w",
+        help="Git branch to use for the imported pipeline (defaults to current local branch)",
+    ),
 ):
     """
     Create a pipeline in Orchestra by referencing a YAML file in your git repository.
@@ -146,6 +159,13 @@ def import_pipeline(
         typer.echo(red("Could not detect default branch from git"))
         raise typer.Exit(code=1)
 
+    # Determine working branch (explicit option or current branch)
+    if working_branch is None:
+        working_branch = _detect_current_branch(repo_root)
+        if not working_branch:
+            typer.echo(red("Could not detect current branch from git"))
+            raise typer.Exit(code=1)
+
     # Compute YAML path relative to repo root
     try:
         yaml_path = str(path.resolve().relative_to(repo_root.resolve()))
@@ -161,6 +181,7 @@ def import_pipeline(
         "storage_provider": _detect_storage_provider(_get_remote_url(repo_root)),
         "repository": repository_slug,
         "default_branch": default_branch,
+        "working_branch": working_branch,
         "yaml_path": yaml_path,
         "alias": alias,
     }

{orchestra_cli-0.1.1 → orchestra_cli-0.1.3}/orchestra_cli/src/run_pipeline.py

@@ -106,6 +106,7 @@ def run_pipeline(
         poll_interval_seconds = 5
         headers = {"Authorization": f"Bearer {api_key}"}
         status_url = f"{api_prefix}/pipeline_runs/{pipeline_run_id}/status"
+        in_progress_statuses = {"RUNNING", "QUEUED", "CREATED"}
 
         while True:
             time.sleep(poll_interval_seconds)
@@ -158,6 +159,9 @@ def run_pipeline(
                 typer.echo(yellow(lineage_url))
                 raise typer.Exit(code=1)
 
+            if status_value in in_progress_statuses:
+                continue
+
             typer.echo(
                 red(f"❌ Invalid status value: {status_value}\nResponse body: {status_body}"),
             )

{orchestra_cli-0.1.1 → orchestra_cli-0.1.3}/orchestra_cli/tests/test_import_pipeline.py

@@ -71,6 +71,7 @@ def test_import_success(
             "storage_provider": storage_provider,
             "repository": repository,
             "default_branch": "main",
+            "working_branch": "main",
             "yaml_path": "pipe.yaml",
             "alias": "demo",
         },
@@ -81,6 +82,7 @@ def test_import_success(
         ("rev-parse", "--show-toplevel"): (0, str(repo_root), ""),
         ("remote", "get-url", "origin"): (0, git_origin, ""),
         ("symbolic-ref", "refs/remotes/origin/HEAD"): (0, "refs/remotes/origin/main", ""),
+        ("rev-parse", "--abbrev-ref", "HEAD"): (0, "main", ""),
         ("status", "--porcelain"): (0, "", ""),
         # Do not provide upstream to skip that branch check path
         ("rev-parse", "--abbrev-ref", "--symbolic-full-name", "@{u}"): (1, "", ""),
@@ -148,6 +150,7 @@ def test_import_api_error(monkeypatch, tmp_path: Path, httpx_mock: HTTPXMock):
         ("rev-parse", "--show-toplevel"): (0, str(repo_root), ""),
         ("remote", "get-url", "origin"): (0, "git@github.com:org/repo.git", ""),
         ("symbolic-ref", "refs/remotes/origin/HEAD"): (0, "refs/remotes/origin/main", ""),
+        ("rev-parse", "--abbrev-ref", "HEAD"): (0, "main", ""),
         ("status", "--porcelain"): (0, "", ""),
         ("rev-parse", "--abbrev-ref", "--symbolic-full-name", "@{u}"): (1, "", ""),
     }
@@ -228,6 +231,14 @@ def test_warnings_printed(monkeypatch, tmp_path: Path, httpx_mock: HTTPXMock):
         url="https://app.getorchestra.io/api/engine/public/pipelines/import",
         json={"id": mock_pipeline_id},
         status_code=201,
+        match_json={
+            "storage_provider": "GITHUB",
+            "repository": "org/repo",
+            "default_branch": "main",
+            "working_branch": "some-branch",
+            "yaml_path": "p.yaml",
+            "alias": "demo",
+        },
     )
 
     import subprocess
@@ -236,6 +247,7 @@ def test_warnings_printed(monkeypatch, tmp_path: Path, httpx_mock: HTTPXMock):
         ("rev-parse", "--show-toplevel"): (0, str(repo_root), ""),
         ("remote", "get-url", "origin"): (0, "git@github.com:org/repo.git", ""),
         ("symbolic-ref", "refs/remotes/origin/HEAD"): (0, "refs/remotes/origin/main", ""),
+        ("rev-parse", "--abbrev-ref", "HEAD"): (0, "some-branch", ""),
         ("status", "--porcelain"): (0, " M p.yaml\n", ""),
         ("rev-parse", "--abbrev-ref", "--symbolic-full-name", "@{u}"): (0, "origin/main", ""),
         ("rev-parse", "HEAD"): (0, "aaaa", ""),

{orchestra_cli-0.1.1 → orchestra_cli-0.1.3}/orchestra_cli/tests/test_run_pipeline.py

@@ -187,6 +187,7 @@ def test_run_wait_success(httpx_mock: HTTPXMock, monkeypatch, tmp_path: Path):
         result.output.strip().splitlines()[1]
         == f"Started pipeline (alias: demo), run id: {mock_pipeline_run_id}"
     )  # noqa: E501
+    assert "Invalid status value: RUNNING" not in result.output
     assert result.output.strip().splitlines()[-2] == "✅ Pipeline succeeded"
     assert result.output.strip().splitlines()[-1] == mock_pipeline_run_id
 
@@ -225,6 +226,7 @@ def test_run_wait_failed(httpx_mock: HTTPXMock, monkeypatch, tmp_path: Path):
 
     result = runner.invoke(app, ["run", "--alias", "demo", "--wait"])
     assert result.exit_code == 1
+    assert "Invalid status value: RUNNING" not in result.output
     assert "status FAILED" in result.output
     assert "/pipeline-runs/run-xyz/lineage" in result.output
 

{orchestra_cli-0.1.1 → orchestra_cli-0.1.3}/orchestra_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: orchestra-cli
-Version: 0.1.1
+Version: 0.1.3
 Summary: Orchestra CLI. Perform operations with Orchestra locally.
 Author-email: Orchestra Technologies <support@getorchestra.io>
 Requires-Python: >=3.10
@@ -87,7 +87,8 @@ export ORCHESTRA_API_KEY=...  # required
 
 orchestra import \
   --alias my-pipeline \
-  --path ./pipelines/pipeline.yaml
+  --path ./pipelines/pipeline.yaml \
+  --working-branch my-feature-branch   # optional; defaults to current local branch
 # or
 orchestra-cli import -a my-pipeline -p ./pipelines/pipeline.yaml
 ```
@@ -96,6 +97,7 @@ Options
 
 - `-a, --alias` (required): The alias you want to register the pipeline under.
 - `-p, --path` (required): Path to the YAML file. Must be inside a git repository.
+- `-w, --working-branch` (optional): Branch to associate as the working branch for this pipeline import. If omitted, the current local git branch is used.
 
 Notes
 
@@ -103,6 +105,7 @@ Notes
 - Git details are detected automatically:
   - Supported providers: GitHub, GitLab, Azure DevOps.
   - The default branch is detected from `origin`.
+  - The working branch defaults to your current local branch when not specified.
   - The YAML path is computed relative to the repository root.
 - On success, the command prints the created pipeline ID (or a success message).
 - Exit codes: `0` on success, `1` on failure.

{orchestra_cli-0.1.1 → orchestra_cli-0.1.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "orchestra-cli"
-version = "0.1.1"
+version = "0.1.3"
 description = "Orchestra CLI. Perform operations with Orchestra locally."
 readme = "README.md"
 authors = [