Production deployment

.github/workflows/test-python.yml

@@ -0,0 +1,43 @@
+# This workflow will build and test the Rust code in this repository. There are
+# separate workflows for testing the python portion of the code base.
+#
+# This file is *not* generated from automation and is manually maintained.
+#
+name: "[tower] Test python"
+
+on:
+  pull_request:
+  push:
+    branches:
+      - '*'
+    tags-ignore:
+      - '**'
+
+concurrency:
+  group: ${{ github.workflow }}-${{ github.ref }}
+  cancel-in-progress: true
+
+jobs:
+  test:
+    runs-on: ${{ matrix.os }}
+    strategy:
+      fail-fast: false
+      matrix:
+        os: [ubuntu-latest, windows-latest]
+
+    steps:
+      - uses: actions/checkout@v4
+
+      - name: Set up Python
+        uses: actions/setup-python@v5
+
+      - name: Install the latest version of uv
+        uses: astral-sh/setup-uv@v6
+
+      - name: Install dependencies
+        if: github.ref_name != 'main'
+        run: uv sync --all-extras
+
+      - name: Run tests
+        if: github.ref_name != 'main'
+        run: uv run -m pytest --tb=short --disable-warnings

Cargo.toml

@@ -4,7 +4,7 @@ resolver = "2"
 
 [workspace.package]
 edition = "2021"
-version = "0.3.9"
+version = "0.3.10"
 description = "Tower is the best way to host Python data apps in production"
 rust-version = "1.77"
 authors = ["Brad Heller <brad@tower.dev>"]

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "maturin"
 
 [project]
 name = "tower"
-version = "0.3.9"
+version = "0.3.10"
 description = "Tower CLI and runtime environment for Tower."
 authors = [{ name = "Tower Computing Inc.", email = "brad@tower.dev" }]
 readme = "README.md"
@@ -45,7 +45,11 @@ dependencies = [
 ai = ["huggingface-hub>=0.30.2", "ollama>=0.4.7"]
 iceberg = ["polars>=1.27.1", "pyarrow>=19.0.1", "pyiceberg>=0.9.0"]
 all = ["tower[ai,iceberg]"]
-dev = ["openapi-python-client>=0.12.1"]
+dev = [
+	"openapi-python-client>=0.12.1",
+  "pytest>=8.3.5",
+  "pytest-httpx>=0.35.0",
+]
 
 [tool.maturin]
 bindings = "bin"

scripts/generate-python-api-client.sh

@@ -6,5 +6,9 @@
 TOWER_OPENAPI_URL="https://api.tower.dev/v1/openapi-3.0.yaml"
 OUTPUT_DIR="src/tower"
 
-cd ${OUTPUT_DIR} && uv run openapi-python-client update --meta none \
+# We have to remove the previously-generated source in case some things were
+# removed, etc.
+rm -rf ${OUTPUT_DIR}/tower_api_client
+
+cd ${OUTPUT_DIR} && uv run openapi-python-client generate --meta none \
 	--url ${TOWER_OPENAPI_URL}

src/tower/_client.py

@@ -15,6 +15,10 @@
 )
 from .tower_api_client.models.error_model import ErrorModel
 
+# WAIT_TIMEOUT is the amount of time to wait between requests when polling the
+# Tower API.
+WAIT_TIMEOUT = 2
+
 # DEFAULT_TOWER_URL is the default tower URL to use when connecting back to
 # Tower.
 DEFAULT_TOWER_URL = "https://api.tower.dev"
@@ -43,7 +47,7 @@ def _env_client(ctx: TowerContext) -> AuthenticatedClient:
 
 
 def run_app(
-    name: str,
+    slug: str,
     environment: Optional[str] = None,
     parameters: Optional[Dict[str, str]] = None,
 ) -> Run:
@@ -68,7 +72,7 @@ def run_app(
     )
 
     output: Optional[Union[ErrorModel, RunAppResponse]] = run_app_api.sync(
-        name=name, client=client, json_body=input_body
+        slug=slug, client=client, body=input_body
     )
 
     if output is None:
@@ -92,7 +96,7 @@ def wait_for_run(run: Run) -> None:
 
     while True:
         output: Optional[Union[DescribeRunResponse, ErrorModel]] = describe_run_api.sync(
-            name=run.app_name,
+            slug=run.app_slug,
             seq=run.number,
             client=client
         )
@@ -114,4 +118,4 @@ def wait_for_run(run: Run) -> None:
                 elif desc.status == "errored":
                     return
                 else:
-                    time.sleep(2)
+                    time.sleep(WAIT_TIMEOUT)

src/tower/tower_api_client/__init__.py

@@ -1,4 +1,5 @@
-""" A client library for accessing Tower API """
+"""A client library for accessing Tower API"""
+
 from .client import AuthenticatedClient, Client
 
 __all__ = (

src/tower/tower_api_client/api/__init__.py

@@ -1 +1 @@
-""" Contains methods for accessing the API """
+"""Contains methods for accessing the API"""

src/tower/tower_api_client/api/default/__init__.py

@@ -0,0 +1 @@
+"""Contains endpoint functions for accessing the API"""