stemtrace 0.2.2__tar.gz → 0.3.0__tar.gz

{stemtrace-0.2.2 → stemtrace-0.3.0}/.bumpversion.toml

@@ -1,5 +1,5 @@
 [tool.bumpversion]
-current_version = "0.2.2"
+current_version = "0.3.0"
 parse = "(?P<major>\\d+)\\.(?P<minor>\\d+)\\.(?P<patch>\\d+)"
 serialize = ["{major}.{minor}.{patch}"]
 commit = true

{stemtrace-0.2.2 → stemtrace-0.3.0}/.github/workflows/e2e.yml

@@ -13,6 +13,12 @@ jobs:
     name: API E2E Tests
     runs-on: ubuntu-latest
     timeout-minutes: 15
+    strategy:
+      fail-fast: false
+      matrix:
+        broker:
+          - redis
+          - rabbitmq
 
     steps:
       - uses: actions/checkout@v6
@@ -20,13 +26,25 @@ jobs:
       - name: Set up Docker Buildx
         uses: docker/setup-buildx-action@v3
 
+      - name: Select compose file and Celery env
+        run: |
+          if [ "${{ matrix.broker }}" = "rabbitmq" ]; then
+            echo "COMPOSE_FILE=docker-compose.e2e.rabbitmq.yml" >> "$GITHUB_ENV"
+            echo "CELERY_BROKER_URL=amqp://guest:guest@localhost:5672//" >> "$GITHUB_ENV"
+            echo "CELERY_RESULT_BACKEND=redis://localhost:16380/1" >> "$GITHUB_ENV"
+          else
+            echo "COMPOSE_FILE=docker-compose.e2e.yml" >> "$GITHUB_ENV"
+            echo "CELERY_BROKER_URL=redis://localhost:16379/0" >> "$GITHUB_ENV"
+            echo "CELERY_RESULT_BACKEND=redis://localhost:16379/1" >> "$GITHUB_ENV"
+          fi
+
       - name: Build images
         run: |
-          docker compose -f docker-compose.e2e.yml build
+          docker compose -f "$COMPOSE_FILE" build
 
       - name: Start services
         run: |
-          docker compose -f docker-compose.e2e.yml up -d
+          docker compose -f "$COMPOSE_FILE" up -d
           # Wait for services to be healthy
           echo "Waiting for services..."
           timeout 120 bash -c 'until curl -sf http://localhost:8000/stemtrace/api/health -o /dev/null; do sleep 2; done'
@@ -51,16 +69,23 @@ jobs:
       - name: Show logs on failure
         if: failure()
         run: |
-          docker compose -f docker-compose.e2e.yml logs
+          docker compose -f "$COMPOSE_FILE" logs
 
       - name: Stop services
         if: always()
-        run: docker compose -f docker-compose.e2e.yml down -v
+        run: |
+          docker compose -f "$COMPOSE_FILE" down -v
 
   playwright:
     name: Playwright E2E Tests
     runs-on: ubuntu-latest
     timeout-minutes: 20
+    strategy:
+      fail-fast: false
+      matrix:
+        broker:
+          - redis
+          - rabbitmq
 
     steps:
       - uses: actions/checkout@v6
@@ -68,13 +93,25 @@ jobs:
       - name: Set up Docker Buildx
         uses: docker/setup-buildx-action@v3
 
+      - name: Select compose file and Celery env
+        run: |
+          if [ "${{ matrix.broker }}" = "rabbitmq" ]; then
+            echo "COMPOSE_FILE=docker-compose.e2e.rabbitmq.yml" >> "$GITHUB_ENV"
+            echo "CELERY_BROKER_URL=amqp://guest:guest@localhost:5672//" >> "$GITHUB_ENV"
+            echo "CELERY_RESULT_BACKEND=redis://localhost:16380/1" >> "$GITHUB_ENV"
+          else
+            echo "COMPOSE_FILE=docker-compose.e2e.yml" >> "$GITHUB_ENV"
+            echo "CELERY_BROKER_URL=redis://localhost:16379/0" >> "$GITHUB_ENV"
+            echo "CELERY_RESULT_BACKEND=redis://localhost:16379/1" >> "$GITHUB_ENV"
+          fi
+
       - name: Build images
         run: |
-          docker compose -f docker-compose.e2e.yml build
+          docker compose -f "$COMPOSE_FILE" build
 
       - name: Start services
         run: |
-          docker compose -f docker-compose.e2e.yml up -d
+          docker compose -f "$COMPOSE_FILE" up -d
           echo "Waiting for services..."
           timeout 120 bash -c 'until curl -sf http://localhost:8000/stemtrace/api/health -o /dev/null; do sleep 2; done'
           echo "Services ready!"
@@ -133,7 +170,7 @@ jobs:
         uses: actions/upload-artifact@v6
         if: always()
         with:
-          name: playwright-report
+          name: playwright-report-${{ matrix.broker }}
           path: src/stemtrace/server/ui/frontend/playwright-report/
           retention-days: 7
 
@@ -141,15 +178,16 @@ jobs:
         uses: actions/upload-artifact@v6
         if: failure()
         with:
-          name: playwright-test-results
+          name: playwright-test-results-${{ matrix.broker }}
           path: src/stemtrace/server/ui/frontend/test-results/
           retention-days: 7
 
       - name: Show logs on failure
         if: failure()
         run: |
-          docker compose -f docker-compose.e2e.yml logs
+          docker compose -f "$COMPOSE_FILE" logs
 
       - name: Stop services
         if: always()
-        run: docker compose -f docker-compose.e2e.yml down -v
+        run: |
+          docker compose -f "$COMPOSE_FILE" down -v

{stemtrace-0.2.2 → stemtrace-0.3.0}/CHANGELOG.md

@@ -7,6 +7,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## [Unreleased]
 
+## [0.3.0] - 2026-01-04
+
 ## [0.2.2] - 2026-01-03
 
 ## [0.2.1] - 2026-01-03

{stemtrace-0.2.2 → stemtrace-0.3.0}/Makefile

@@ -168,9 +168,10 @@ e2e-mock:
 
 # Start E2E test environment (Docker)
 e2e-up:
+	docker compose -f docker-compose.e2e.yml build
 	docker compose -f docker-compose.e2e.yml up -d --wait
 	@echo "Waiting for services..."
-	@timeout 60 bash -c 'until curl -sf http://localhost:8000/stemtrace/api/health -o /dev/null; do sleep 2; done' || (echo "❌ Service failed to become healthy"; exit 1)
+	@uv run python scripts/wait_for_http.py http://localhost:8000/stemtrace/api/health --timeout 60
 	@echo "✅ E2E environment ready at http://localhost:8000"
 
 # Stop E2E test environment
@@ -193,5 +194,34 @@ e2e:
 	$(MAKE) e2e-down; \
 	exit $$status
 
+# =============================================================================
+# E2E Testing (RabbitMQ broker)
+# =============================================================================
+
+e2e-up-rabbitmq:
+	docker compose -f docker-compose.e2e.rabbitmq.yml build
+	docker compose -f docker-compose.e2e.rabbitmq.yml up -d --wait
+	@echo "Waiting for services..."
+	@uv run python scripts/wait_for_http.py http://localhost:8000/stemtrace/api/health --timeout 60
+	@echo "✅ RabbitMQ E2E environment ready at http://localhost:8000"
+
+e2e-down-rabbitmq:
+	docker compose -f docker-compose.e2e.rabbitmq.yml down -v
+
+e2e-api-rabbitmq:
+	CELERY_BROKER_URL=amqp://guest:guest@localhost:5672// \
+	CELERY_RESULT_BACKEND=redis://localhost:16380/1 \
+	uv run pytest tests/e2e/ -m e2e -v
+
+e2e-playwright-real-rabbitmq:
+	cd $(FRONTEND_DIR) && E2E_MODE=real PLAYWRIGHT_BASE_URL=http://localhost:8000 npm test
+
+e2e-rabbitmq:
+	$(MAKE) e2e-up-rabbitmq
+	$(MAKE) e2e-api-rabbitmq && $(MAKE) e2e-playwright-real-rabbitmq; \
+	status=$$?; \
+	$(MAKE) e2e-down-rabbitmq; \
+	exit $$status
+
 # Quick E2E alias (mock mode, no Docker)
 e2e-quick: e2e-mock

{stemtrace-0.2.2 → stemtrace-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: stemtrace
-Version: 0.2.2
+Version: 0.3.0
 Summary: A lightweight Celery task flow visualizer
 Project-URL: Homepage, https://github.com/iansokolskyi/stemtrace
 Project-URL: Documentation, https://github.com/iansokolskyi/stemtrace#readme
@@ -56,7 +56,7 @@ Description-Content-Type: text/markdown
 
 **Zero-infrastructure Celery task flow visualizer**
 
-[![PyPI version](https://img.shields.io/badge/pypi-v0.2.2-darklime)](https://pypi.org/project/stemtrace)
+[![PyPI version](https://img.shields.io/badge/pypi-v0.3.0-darklime)](https://pypi.org/project/stemtrace)
 [![Python](https://img.shields.io/pypi/pyversions/stemtrace.svg)](https://pypi.org/project/stemtrace/)
 [![CI](https://github.com/iansokolskyi/stemtrace/actions/workflows/ci.yml/badge.svg)](https://github.com/iansokolskyi/stemtrace/actions/workflows/ci.yml)
 [![codecov](https://codecov.io/gh/iansokolskyi/stemtrace/graph/badge.svg)](https://codecov.io/gh/iansokolskyi/stemtrace)
@@ -72,7 +72,7 @@ Ever stared at a failed Celery task wondering "what called this?" or "why did it
 
 Stemtrace captures your task executions as a graph — visualize parent→child flows, see retry chains, track groups and chords, all without adding any new infrastructure.
 
-**Today stemtrace supports Redis** for event transport (via Redis Streams). **RabbitMQ support is planned.**
+**stemtrace supports Redis and RabbitMQ** for event transport.
 
 ## ✨ Features
 
@@ -91,7 +91,7 @@ Stemtrace captures your task executions as a graph — visualize parent→child
 - **Registry status badges** — Quickly spot tasks that are active, never run, or not registered by any current worker
 
 **Production Ready**
-- **Zero Infrastructure** — Uses your existing Redis broker, no database needed
+- **Zero Infrastructure** — Uses your existing broker (Redis or RabbitMQ), no database needed
 - **Sensitive Data Scrubbing** — Passwords and API keys filtered automatically
 - **Read-Only** — Safe for production; never modifies your task queue
 - **FastAPI Integration** — Mount into your existing app with one line
@@ -162,6 +162,14 @@ stemtrace server
 
 Open [http://localhost:8000](http://localhost:8000).
 
+Tip: make sure the server is pointed at the same broker as your workers:
+
+```bash
+stemtrace server --broker-url redis://localhost:6379/0
+# or:
+stemtrace server --broker-url amqp://guest:guest@localhost:5672//
+```
+
 **Option B: Embed in your FastAPI app** (no extra container)
 
 ```python
@@ -175,7 +183,7 @@ stemtrace.init_app(app, broker_url="redis://localhost:6379/0")
 
 Access at `/stemtrace/` in your existing app — no new services to deploy.
 
-See [Deployment Options](#️-deployment-options) for auth, scaling, and more.
+See [Deployment Options](#deployment-options) for auth, scaling, and more.
 
 ## 📦 Architecture
 
@@ -305,7 +313,7 @@ batch_processor
 | Broker | URL Scheme | Status |
 |--------|------------|--------|
 | Redis | `redis://`, `rediss://` | ✅ Supported |
-| RabbitMQ | `amqp://`, `amqps://` | 🚧 Planned |
+| RabbitMQ | `amqp://`, `amqps://`, `pyamqp://` | ✅ Supported |
 
 ## 🐳 Docker
 
@@ -315,6 +323,14 @@ docker run -p 8000:8000 \
     ghcr.io/iansokolskyi/stemtrace
 ```
 
+RabbitMQ example:
+
+```bash
+docker run -p 8000:8000 \
+    -e STEMTRACE_BROKER_URL=amqp://guest:guest@host.docker.internal:5672// \
+    ghcr.io/iansokolskyi/stemtrace
+```
+
 Or with Docker Compose:
 
 ```yaml
@@ -327,6 +343,8 @@ services:
       - STEMTRACE_BROKER_URL=redis://redis:6379/0
 ```
 
+For a local RabbitMQ setup, see [`docker-compose.rabbitmq.yml`](docker-compose.rabbitmq.yml).
+
 ## 🖥️ Deployment Options
 
 `stemtrace` offers two deployment modes depending on your needs:
@@ -437,12 +455,12 @@ stemtrace.init_app(
 - ✅ **Sensitive data scrubbing** — Passwords and API keys filtered automatically
 - ✅ **Real-time updates** — WebSocket-powered live dashboard
 - ✅ **FastAPI integration** — Mount into your existing app
+- ✅ **RabbitMQ support** — Use your existing RabbitMQ broker (`amqp://`, `amqps://`, `pyamqp://`)
 - ✅ **Workers page** — Monitor online/offline workers and their registered tasks
 - ✅ **Task registry** — Browse discovered + registered tasks with clear status badges
 
 ### Coming Soon
 
-- 🔜 **RabbitMQ support** — Use your existing RabbitMQ broker
 - 🔜 **Anomaly detection** — Spot stuck, orphaned, or failed tasks
 - 🔜 **Dashboard with stats** — Success rates, durations, failure trends
 - 🔜 **OpenTelemetry export** — Send traces to Jaeger, Tempo, Datadog

{stemtrace-0.2.2 → stemtrace-0.3.0}/README.md

@@ -2,7 +2,7 @@
 
 **Zero-infrastructure Celery task flow visualizer**
 
-[![PyPI version](https://img.shields.io/badge/pypi-v0.2.2-darklime)](https://pypi.org/project/stemtrace)
+[![PyPI version](https://img.shields.io/badge/pypi-v0.3.0-darklime)](https://pypi.org/project/stemtrace)
 [![Python](https://img.shields.io/pypi/pyversions/stemtrace.svg)](https://pypi.org/project/stemtrace/)
 [![CI](https://github.com/iansokolskyi/stemtrace/actions/workflows/ci.yml/badge.svg)](https://github.com/iansokolskyi/stemtrace/actions/workflows/ci.yml)
 [![codecov](https://codecov.io/gh/iansokolskyi/stemtrace/graph/badge.svg)](https://codecov.io/gh/iansokolskyi/stemtrace)
@@ -18,7 +18,7 @@ Ever stared at a failed Celery task wondering "what called this?" or "why did it
 
 Stemtrace captures your task executions as a graph — visualize parent→child flows, see retry chains, track groups and chords, all without adding any new infrastructure.
 
-**Today stemtrace supports Redis** for event transport (via Redis Streams). **RabbitMQ support is planned.**
+**stemtrace supports Redis and RabbitMQ** for event transport.
 
 ## ✨ Features
 
@@ -37,7 +37,7 @@ Stemtrace captures your task executions as a graph — visualize parent→child
 - **Registry status badges** — Quickly spot tasks that are active, never run, or not registered by any current worker
 
 **Production Ready**
-- **Zero Infrastructure** — Uses your existing Redis broker, no database needed
+- **Zero Infrastructure** — Uses your existing broker (Redis or RabbitMQ), no database needed
 - **Sensitive Data Scrubbing** — Passwords and API keys filtered automatically
 - **Read-Only** — Safe for production; never modifies your task queue
 - **FastAPI Integration** — Mount into your existing app with one line
@@ -108,6 +108,14 @@ stemtrace server
 
 Open [http://localhost:8000](http://localhost:8000).
 
+Tip: make sure the server is pointed at the same broker as your workers:
+
+```bash
+stemtrace server --broker-url redis://localhost:6379/0
+# or:
+stemtrace server --broker-url amqp://guest:guest@localhost:5672//
+```
+
 **Option B: Embed in your FastAPI app** (no extra container)
 
 ```python
@@ -121,7 +129,7 @@ stemtrace.init_app(app, broker_url="redis://localhost:6379/0")
 
 Access at `/stemtrace/` in your existing app — no new services to deploy.
 
-See [Deployment Options](#️-deployment-options) for auth, scaling, and more.
+See [Deployment Options](#deployment-options) for auth, scaling, and more.
 
 ## 📦 Architecture
 
@@ -251,7 +259,7 @@ batch_processor
 | Broker | URL Scheme | Status |
 |--------|------------|--------|
 | Redis | `redis://`, `rediss://` | ✅ Supported |
-| RabbitMQ | `amqp://`, `amqps://` | 🚧 Planned |
+| RabbitMQ | `amqp://`, `amqps://`, `pyamqp://` | ✅ Supported |
 
 ## 🐳 Docker
 
@@ -261,6 +269,14 @@ docker run -p 8000:8000 \
     ghcr.io/iansokolskyi/stemtrace
 ```
 
+RabbitMQ example:
+
+```bash
+docker run -p 8000:8000 \
+    -e STEMTRACE_BROKER_URL=amqp://guest:guest@host.docker.internal:5672// \
+    ghcr.io/iansokolskyi/stemtrace
+```
+
 Or with Docker Compose:
 
 ```yaml
@@ -273,6 +289,8 @@ services:
       - STEMTRACE_BROKER_URL=redis://redis:6379/0
 ```
 
+For a local RabbitMQ setup, see [`docker-compose.rabbitmq.yml`](docker-compose.rabbitmq.yml).
+
 ## 🖥️ Deployment Options
 
 `stemtrace` offers two deployment modes depending on your needs:
@@ -383,12 +401,12 @@ stemtrace.init_app(
 - ✅ **Sensitive data scrubbing** — Passwords and API keys filtered automatically
 - ✅ **Real-time updates** — WebSocket-powered live dashboard
 - ✅ **FastAPI integration** — Mount into your existing app
+- ✅ **RabbitMQ support** — Use your existing RabbitMQ broker (`amqp://`, `amqps://`, `pyamqp://`)
 - ✅ **Workers page** — Monitor online/offline workers and their registered tasks
 - ✅ **Task registry** — Browse discovered + registered tasks with clear status badges
 
 ### Coming Soon
 
-- 🔜 **RabbitMQ support** — Use your existing RabbitMQ broker
 - 🔜 **Anomaly detection** — Spot stuck, orphaned, or failed tasks
 - 🔜 **Dashboard with stats** — Success rates, durations, failure trends
 - 🔜 **OpenTelemetry export** — Send traces to Jaeger, Tempo, Datadog

stemtrace-0.3.0/docker-compose.e2e.rabbitmq.yml

@@ -0,0 +1,88 @@
+# Docker Compose for stemtrace E2E testing (RabbitMQ broker)
+#
+# Usage:
+#   docker compose -f docker-compose.e2e.rabbitmq.yml up -d --wait
+#   docker compose -f docker-compose.e2e.rabbitmq.yml down -v
+#
+# Services:
+#   - rabbitmq: RabbitMQ broker (with management UI)
+#   - worker: Celery worker running E2E test tasks
+#   - server: stemtrace server with UI and API
+
+services:
+  rabbitmq:
+    image: rabbitmq:3-management
+    ports:
+      - "5672:5672"
+      - "15672:15672"
+    healthcheck:
+      test: ["CMD", "rabbitmq-diagnostics", "-q", "ping"]
+      interval: 2s
+      timeout: 5s
+      retries: 20
+
+  redis:
+    image: redis:7-alpine
+    ports:
+      - "16380:6379" # Dedicated port for RabbitMQ E2E result backend
+    healthcheck:
+      test: ["CMD", "redis-cli", "ping"]
+      interval: 2s
+      timeout: 2s
+      retries: 10
+
+  worker:
+    build:
+      context: .
+      dockerfile: Dockerfile.e2e
+    environment:
+      - CELERY_BROKER_URL=amqp://guest:guest@rabbitmq:5672//
+      # Redis result backend required for group->callback workflows (Celery upgrades
+      # chain(group(...), callback) to a chord, which rpc:// does not support).
+      - CELERY_RESULT_BACKEND=redis://redis:6379/1
+    depends_on:
+      rabbitmq:
+        condition: service_healthy
+      redis:
+        condition: service_healthy
+    command: ["celery", "-A", "tests.e2e.tasks", "worker", "--loglevel=info", "--concurrency=2", "-E"]
+    healthcheck:
+      test: ["CMD", "celery", "-A", "tests.e2e.tasks", "inspect", "ping"]
+      interval: 5s
+      timeout: 10s
+      retries: 10
+      start_period: 10s
+
+  server:
+    build:
+      context: .
+      dockerfile: Dockerfile
+    ports:
+      - "8000:8000"
+    environment:
+      - STEMTRACE_BROKER_URL=amqp://guest:guest@rabbitmq:5672//
+    depends_on:
+      rabbitmq:
+        condition: service_healthy
+    command:
+      [
+        "server",
+        "--broker-url",
+        "amqp://guest:guest@rabbitmq:5672//",
+        "--host",
+        "0.0.0.0",
+        "--port",
+        "8000",
+      ]
+    healthcheck:
+      test:
+        [
+          "CMD",
+          "python",
+          "-c",
+          "import httpx; httpx.get('http://localhost:8000/stemtrace/api/health').raise_for_status()",
+        ]
+      interval: 5s
+      timeout: 10s
+      retries: 10
+      start_period: 5s

stemtrace-0.3.0/docker-compose.rabbitmq.yml

@@ -0,0 +1,44 @@
+# Docker Compose for stemtrace with RabbitMQ
+#
+# Usage:
+#   docker compose -f docker-compose.rabbitmq.yml up -d
+#   docker compose -f docker-compose.rabbitmq.yml logs -f server
+#   docker compose -f docker-compose.rabbitmq.yml down
+#
+# Access:
+#   - stemtrace UI: http://localhost:8000
+#   - RabbitMQ management UI: http://localhost:15672 (guest/guest)
+
+services:
+  rabbitmq:
+    image: rabbitmq:3-management
+    ports:
+      - "5672:5672"
+      - "15672:15672"
+    healthcheck:
+      test: ["CMD", "rabbitmq-diagnostics", "-q", "ping"]
+      interval: 5s
+      timeout: 5s
+      retries: 10
+
+  server:
+    build:
+      context: .
+      dockerfile: Dockerfile
+    ports:
+      - "8000:8000"
+    environment:
+      - STEMTRACE_BROKER_URL=amqp://guest:guest@rabbitmq:5672//
+    depends_on:
+      rabbitmq:
+        condition: service_healthy
+    command:
+      [
+        "server",
+        "--broker-url",
+        "amqp://guest:guest@rabbitmq:5672//",
+        "--host",
+        "0.0.0.0",
+        "--port",
+        "8000",
+      ]

{stemtrace-0.2.2 → stemtrace-0.3.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "stemtrace"
-version = "0.2.2"
+version = "0.3.0"
 description = "A lightweight Celery task flow visualizer"
 readme = "README.md"
 license = { text = "MIT" }
@@ -109,6 +109,14 @@ ignore_missing_imports = true
 module = "typer.*"
 ignore_missing_imports = true
 
+[[tool.mypy.overrides]]
+module = "kombu.*"
+ignore_missing_imports = true
+
+[[tool.mypy.overrides]]
+module = "amqp.*"
+ignore_missing_imports = true
+
 # =============================================================================
 # ruff
 # =============================================================================

stemtrace-0.3.0/scripts/wait_for_http.py

@@ -0,0 +1,41 @@
+"""Wait for an HTTP endpoint to respond with a 2xx status."""
+
+from __future__ import annotations
+
+import argparse
+import time
+import urllib.error
+import urllib.request
+
+
+def _parse_args() -> argparse.Namespace:
+    parser = argparse.ArgumentParser(description=__doc__)
+    parser.add_argument("url")
+    parser.add_argument("--timeout", type=float, default=60.0)
+    parser.add_argument("--interval", type=float, default=2.0)
+    return parser.parse_args()
+
+
+def main() -> None:
+    """Wait for an HTTP endpoint to become healthy (2xx) or exit non-zero."""
+    args = _parse_args()
+    deadline = time.monotonic() + args.timeout
+    last_error: str | None = None
+
+    while time.monotonic() < deadline:
+        try:
+            request = urllib.request.Request(args.url, method="GET")
+            with urllib.request.urlopen(request, timeout=2.0) as response:
+                if 200 <= response.status < 300:
+                    return
+                last_error = f"HTTP {response.status}"
+        except (urllib.error.URLError, TimeoutError) as exc:
+            last_error = str(exc)
+
+        time.sleep(args.interval)
+
+    raise SystemExit(f"Service not healthy within {args.timeout}s: {last_error}")
+
+
+if __name__ == "__main__":
+    main()

{stemtrace-0.2.2 → stemtrace-0.3.0}/src/stemtrace/__init__.py

@@ -38,7 +38,7 @@ if TYPE_CHECKING:
     from celery import Celery
     from fastapi import FastAPI
 
-__version__ = "0.2.2"
+__version__ = "0.3.0"
 __all__ = [
     "ConfigurationError",
     "StemtraceConfig",

{stemtrace-0.2.2 → stemtrace-0.3.0}/src/stemtrace/core/exceptions.py

@@ -21,5 +21,5 @@ class UnsupportedBrokerError(ConfigurationError):
         self.scheme = scheme
         super().__init__(
             f"Unsupported broker scheme: '{scheme}'. "
-            f"Supported: redis, rediss, amqp, amqps, memory"
+            f"Supported: redis, rediss, amqp, amqps, pyamqp, memory"
         )

{stemtrace-0.2.2 → stemtrace-0.3.0}/src/stemtrace/library/transports/__init__.py

@@ -11,6 +11,7 @@ if TYPE_CHECKING:
 _SCHEME_ALIASES: dict[str, str] = {
     "rediss": "redis",
     "amqps": "amqp",
+    "pyamqp": "amqp",
 }
 
 
@@ -25,6 +26,10 @@ def get_transport(
         from stemtrace.library.transports.redis import RedisTransport
 
         return RedisTransport.from_url(url, prefix=prefix, ttl=ttl)
+    elif scheme == "amqp":
+        from stemtrace.library.transports.rabbitmq import RabbitMQTransport
+
+        return RabbitMQTransport.from_url(url, prefix=prefix, ttl=ttl)
     elif scheme == "memory":
         from stemtrace.library.transports.memory import MemoryTransport