simpleaudit-studio 0.2.4__tar.gz → 0.2.6__tar.gz

{simpleaudit_studio-0.2.4 → simpleaudit_studio-0.2.6}/.env.example

@@ -82,10 +82,10 @@ SSE_MAX_CONNECTIONS_PER_USER=10
 
 # --- SimpleAudit engine ---------------------------------------------------------
 # No configuration needed here. The engine is a normal pip dependency pinned in
-# requirements.txt; its version + optional git commit are read from the installed
+# pyproject.toml; its version + optional git commit are read from the installed
 # package metadata automatically (infra/simpleaudit_package.py). To develop
 # against an unmerged revision, create a gitignored simpleaudit-dependency.yaml
-# at the project root with a single `git_ref:` key — do NOT edit requirements.txt.
+# at the project root with a single `git_ref:` key — do NOT edit pyproject.toml.
 
 # --- Model API credentials (resolved at execution time, never stored) ----------
 # Add one variable per model endpoint's `secret_reference`, e.g.:

{simpleaudit_studio-0.2.4 → simpleaudit_studio-0.2.6}/.env.local.example

@@ -9,7 +9,7 @@
 # (see `.env.example`).
 #
 # Prerequisites:
-#   1. A Python 3.12+ virtualenv with `pip install -r requirements.txt`
+#   1. A Python 3.12+ virtualenv with `uv sync --extra dev --extra postgres`
 #   2. Docker, for Postgres + Hatchet (the "partial Docker" pattern below)
 #   3. A SimpleAudit engine checkout (for the worker to execute audits)
 # =============================================================================
@@ -66,14 +66,14 @@ WORKER_POOL=cpu
 
 # --- SimpleAudit engine ---------------------------------------------------------
 # No configuration needed here. The engine is a normal pip dependency pinned in
-# requirements.txt; its version + optional git commit are read from the installed
+# pyproject.toml; its version + optional git commit are read from the installed
 # package metadata automatically (infra/simpleaudit_package.py).
 #
 # To develop against an unmerged engine revision, create a gitignored
 # simpleaudit-dependency.yaml at the project root containing ONLY:
 #     git_ref: <full-40-char-sha>
 # That value is recorded as the run's git_commit provenance. Do NOT edit
-# requirements.txt for a dev override.
+# pyproject.toml for a dev override.
 
 # --- Model API credentials (resolved at execution time, never stored) ----------
 # Add one variable per model endpoint's `secret_reference`, e.g.:

{simpleaudit_studio-0.2.4 → simpleaudit_studio-0.2.6}/.github/workflows/ci.yml

@@ -37,12 +37,14 @@ jobs:
       - uses: actions/setup-python@v5
         with:
           python-version: "3.12"
-          cache: pip
+      - uses: astral-sh/setup-uv@v7
       - name: Install dependencies
-        run: pip install -r requirements.txt
+        run: uv sync --frozen --extra dev --extra postgres
+      - name: Lint (ruff)
+        run: uv run ruff check .
       - name: Django checks
-        run: python manage.py check
+        run: uv run python manage.py check
       - name: Migrations from empty database
-        run: python manage.py migrate
+        run: uv run python manage.py migrate
       - name: Run tests
-        run: python manage.py test infra
+        run: uv run python manage.py test infra

{simpleaudit_studio-0.2.4 → simpleaudit_studio-0.2.6}/.gitignore

@@ -16,7 +16,7 @@ staticfiles/
 .env.local
 
 # Dev-only SimpleAudit engine override (git_ref). Never committed — production
-# relies on the requirements.txt pin. See infra/simpleaudit_package.py.
+# relies on the pyproject.toml pin. See infra/simpleaudit_package.py.
 simpleaudit-dependency.yaml
 
 # IDE

{simpleaudit_studio-0.2.4 → simpleaudit_studio-0.2.6}/Dockerfile

@@ -22,16 +22,20 @@ WORKDIR /app
 
 # --- System dependencies -----------------------------------------------------
 # curl: healthchecks
-# git: required to install SimpleAudit engine from its pinned git ref
 RUN apt-get update \
     && apt-get install -y --no-install-recommends \
         curl \
-        git \
     && rm -rf /var/lib/apt/lists/*
 
 # --- Python dependencies -----------------------------------------------------
-COPY requirements.txt .
-RUN pip install -r requirements.txt
+# pyproject.toml is the single source of truth; uv.lock pins exact versions.
+# The postgres extra is included because Django imports the psycopg driver at
+# startup even when SIMPLEAUDIT_MINIMAL=1 uses SQLite for the domain DB.
+# uv sync creates /app/.venv; the PATH update keeps the `python` entrypoint.
+COPY pyproject.toml uv.lock README.md ./
+RUN pip install uv \
+    && uv sync --frozen --no-install-project --no-dev --extra postgres
+ENV PATH="/app/.venv/bin:$PATH"
 
 # --- Application code --------------------------------------------------------
 COPY . .

{simpleaudit_studio-0.2.4 → simpleaudit_studio-0.2.6}/Makefile

@@ -14,8 +14,7 @@ help: ## Show available targets
 	@grep -E '^[a-zA-Z_-]+:.*?## ' $(MAKEFILE_LIST) | awk 'BEGIN {FS = ":.*?## "}; {printf "  \033[36m%-14s\033[0m %s\n", $$1, $$2}'
 
 venv: ## Create the local virtualenv and install dependencies
-	python3 -m venv $(VENV)
-	$(VENV)/bin/pip install -r requirements.txt
+	uv sync --extra dev --extra postgres
 
 local-setup: ## Local dev: venv + .env + migrations + bootstrap + seed (idempotent)
 	@test -d $(VENV) || $(MAKE) venv

{simpleaudit_studio-0.2.4 → simpleaudit_studio-0.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: simpleaudit-studio
-Version: 0.2.4
+Version: 0.2.6
 Summary: Production AI model audit platform — one-liner demo mode via uvx
 License: MIT
 License-File: LICENSE
@@ -15,14 +15,17 @@ Requires-Dist: pydantic<3,>=2.7
 Requires-Dist: python-dotenv==1.1.1
 Requires-Dist: requests==2.32.5
 Requires-Dist: sentry-sdk[django]>=2.0
-Requires-Dist: simpleaudit==0.2.0
+Requires-Dist: simpleaudit==0.2.1
 Requires-Dist: workos==10.4.0
 Provides-Extra: dev
 Requires-Dist: factory-boy==3.3.2; extra == 'dev'
 Requires-Dist: pytest; extra == 'dev'
 Requires-Dist: pytest-django; extra == 'dev'
+Requires-Dist: ruff; extra == 'dev'
 Provides-Extra: postgres
 Requires-Dist: psycopg[binary]==3.2.13; extra == 'postgres'
+Provides-Extra: server
+Requires-Dist: gunicorn==23.0.0; extra == 'server'
 Description-Content-Type: text/markdown
 
 ---
@@ -54,13 +57,14 @@ uvx simpleaudit-studio
 
 [`uvx`](https://docs.astral.sh/uv/#uvx) installs the [`simpleaudit-studio`](https://pypi.org/project/simpleaudit-studio/) package and runs it in an isolated environment. No Docker, no Postgres, no manual setup. Opens at http://localhost:8000 (login: `studio` / `admin123`). A mock model server is pre-seeded so you can start exploring audit results immediately.
 
+🌐 Or skip the setup entirely — try the live demo: <a href="https://sushantgautam-simpleaudit-studio.hf.space" target="_blank" rel="noopener noreferrer">sushantgautam-simpleaudit-studio.hf.space</a>
+
 Want the **latest unreleased code** from this repo instead of the PyPI release? Same one-liner, pointed at git:
 
 ```bash
 uvx --from "git+https://github.com/SushantGautam/SimpleAuditStudio" spin
 ```
 
-🌐 Or skip the setup entirely — try the live demo: <a href="https://sushantgautam-simpleaudit-studio.hf.space" target="_blank" rel="noopener noreferrer">sushantgautam-simpleaudit-studio.hf.space</a>
 
 ### 🤖 Using Real Models
 

{simpleaudit_studio-0.2.4 → simpleaudit_studio-0.2.6}/README.md

@@ -27,13 +27,14 @@ uvx simpleaudit-studio
 
 [`uvx`](https://docs.astral.sh/uv/#uvx) installs the [`simpleaudit-studio`](https://pypi.org/project/simpleaudit-studio/) package and runs it in an isolated environment. No Docker, no Postgres, no manual setup. Opens at http://localhost:8000 (login: `studio` / `admin123`). A mock model server is pre-seeded so you can start exploring audit results immediately.
 
+🌐 Or skip the setup entirely — try the live demo: <a href="https://sushantgautam-simpleaudit-studio.hf.space" target="_blank" rel="noopener noreferrer">sushantgautam-simpleaudit-studio.hf.space</a>
+
 Want the **latest unreleased code** from this repo instead of the PyPI release? Same one-liner, pointed at git:
 
 ```bash
 uvx --from "git+https://github.com/SushantGautam/SimpleAuditStudio" spin
 ```
 
-🌐 Or skip the setup entirely — try the live demo: <a href="https://sushantgautam-simpleaudit-studio.hf.space" target="_blank" rel="noopener noreferrer">sushantgautam-simpleaudit-studio.hf.space</a>
 
 ### 🤖 Using Real Models
 

{simpleaudit_studio-0.2.4 → simpleaudit_studio-0.2.6}/accounts/views.py

@@ -2,7 +2,6 @@
 import logging
 
 from django.contrib.auth import authenticate, login
-from django.db import connection
 from django.http import JsonResponse
 from rest_framework import status
 from rest_framework.authtoken.models import Token
@@ -10,28 +9,26 @@ from rest_framework.decorators import api_view, permission_classes
 from rest_framework.permissions import AllowAny, IsAuthenticated
 from rest_framework.response import Response
 
-from infra.exceptions import StableAPIError
 from accounts.models import Project, ProjectMembership, User
 from accounts.serializers import (
     MemberAddSerializer,
     MemberRoleSerializer,
     ProjectMembershipSerializer,
-    ProjectSerializer,
     RegisterSerializer,
     UserSerializer,
     WorkspaceCreateSerializer,
     WorkspaceItemSerializer,
     WorkspaceUpdateSerializer,
 )
-from infra.readiness import ready_payload
 from accounts.services import (
     DEFAULT_PROJECT_SLUG,
-    bootstrap_admin_and_default_project,
     create_workspace,
     delete_workspace,
     ensure_project_access,
     update_workspace,
 )
+from infra.exceptions import StableAPIError
+from infra.readiness import ready_payload
 
 logger = logging.getLogger(__name__)
 

{simpleaudit_studio-0.2.4 → simpleaudit_studio-0.2.6}/audits/comparison.py

@@ -160,7 +160,7 @@ def compare_runs(project, run_ids: list[int]) -> dict:
             elif label == "Scenario set version" and r.scenario_set_version:
                 url = f"/scenarios/?set={r.scenario_set_version.scenario_set_id}"
             cells.append(_cell(text, url))
-        differs = len(set(c["text"] for c in cells)) > 1
+        differs = len({c["text"] for c in cells}) > 1
         input_rows.append({"label": label, "cells": cells, "differs": differs})
 
     return {

{simpleaudit_studio-0.2.4 → simpleaudit_studio-0.2.6}/audits/events.py

@@ -15,8 +15,6 @@ events are inputs only; the worker bridges them into these rows.
 """
 from __future__ import annotations
 
-import json
-
 from django.db import models, transaction
 
 
@@ -124,8 +122,7 @@ def upsert_scenario_result(
             result=result or {},
         )
         return
-    if attempts > existing.attempts:
-        existing.attempts = attempts
+    existing.attempts = max(existing.attempts, attempts)
     existing.status = status
     if result is not None:
         existing.result = result

{simpleaudit_studio-0.2.4 → simpleaudit_studio-0.2.6}/audits/services.py

@@ -1,16 +1,15 @@
 """Audit run creation and freeze services."""
 import logging
 
-from django.conf import settings
 from django.db import transaction
 from django.utils import timezone
 
+from accounts.models import Project
 from audits.events import append_event
 from audits.models import AuditRun
 from infra.exceptions import StableAPIError
 from infra.simpleaudit_package import resolve_engine_provenance
 from model_registry.models import ModelEndpoint
-from accounts.models import Project
 from scenarios.models import ScenarioSetVersion
 from scenarios.services import require_project_role
 
@@ -131,7 +130,9 @@ def submit_audit_run(run: AuditRun) -> str | None:
     reason is recorded in ``runtime_metadata["submission"]`` so an operator (or a
     later retry command) can resubmit without re-freezing inputs.
     """
-    from infra.worker import submit_run_workflow  # lazy: needs no live server at import time
+    from infra.worker import (
+        submit_run_workflow,  # lazy: needs no live server at import time
+    )
 
     items = list(
         run.scenario_set_version.items.select_related("scenario", "revision").order_by("position")

{simpleaudit_studio-0.2.4 → simpleaudit_studio-0.2.6}/audits/views.py

@@ -9,6 +9,8 @@ from rest_framework.decorators import api_view, permission_classes
 from rest_framework.permissions import IsAuthenticated
 from rest_framework.response import Response
 
+from accounts.models import Project
+from accounts.services import ensure_project_access
 from audits.events import ScenarioResult, list_events
 from audits.models import AuditRun
 from audits.serializers import AuditRunCreateSerializer, AuditRunSerializer
@@ -16,9 +18,7 @@ from audits.services import create_audit_run, submit_audit_run
 from infra.exceptions import StableAPIError
 from infra.middleware import set_correlation_context
 from model_registry.models import ModelEndpoint
-from accounts.models import Project
 from scenarios.models import ScenarioSetVersion
-from accounts.services import ensure_project_access
 
 logger = logging.getLogger("simpleaudit.audit")
 
@@ -214,7 +214,6 @@ def stream_audit_run_events(request, project_id, run_id):
     with a higher id — standard SSE replay semantics. The stream ends when a
     terminal run event (completed/failed/cancelled) is observed.
     """
-    from django.contrib.auth.decorators import login_required
     from django.http import Http404
 
     # This is called via URL dispatch; enforce auth manually since we're not using DRF.

{simpleaudit_studio-0.2.4 → simpleaudit_studio-0.2.6}/deploy/compose/Dockerfile

@@ -23,16 +23,20 @@ ENV PYTHONDONTWRITEBYTECODE=1 \
 
 WORKDIR /app
 
-# curl: healthchecks; git: required to install the SimpleAudit engine from its
-# pinned git ref in requirements.txt.
+# curl: healthchecks
 RUN apt-get update \
     && apt-get install -y --no-install-recommends \
         curl \
-        git \
     && rm -rf /var/lib/apt/lists/*
 
-COPY requirements.txt .
-RUN pip install -r requirements.txt
+# pyproject.toml is the single source of truth; uv.lock pins exact versions.
+# The compose stack runs against external Postgres (postgres extra) and serves
+# via gunicorn (server extra). uv sync creates /app/.venv; the PATH update
+# keeps the `python`/`gunicorn` entrypoints working.
+COPY pyproject.toml uv.lock README.md ./
+RUN pip install uv \
+    && uv sync --frozen --no-install-project --no-dev --extra postgres --extra server
+ENV PATH="/app/.venv/bin:$PATH"
 
 COPY . .
 

{simpleaudit_studio-0.2.4 → simpleaudit_studio-0.2.6}/deploy/e2e_smoke.py

@@ -25,12 +25,13 @@ def _load_env(path=".env"):
     env = {}
     if not os.path.exists(path):
         return env
-    for line in open(path):
-        line = line.strip()
-        if not line or line.startswith("#") or "=" not in line:
-            continue
-        k, _, v = line.partition("=")
-        env[k.strip()] = v.strip()
+    with open(path) as f:
+        for line in f:
+            line = line.strip()
+            if not line or line.startswith("#") or "=" not in line:
+                continue
+            k, _, v = line.partition("=")
+            env[k.strip()] = v.strip()
     return env
 
 
@@ -56,7 +57,7 @@ class Client:
             raw = e.read().decode()
             try:
                 parsed = json.loads(raw)
-            except Exception:
+            except Exception:  # noqa: BLE001 - fall back to raw body if not JSON
                 parsed = raw
             return e.code, parsed
 
@@ -77,7 +78,8 @@ def main():
         sys.exit(2)
 
     c = Client(base_url)
-    step = lambda m: print(f"\n=== {m} ===")
+    def step(m):
+        print(f"\n=== {m} ===")
     # Unique suffix so the smoke test is repeatable without colliding with the
     # (project, display_name) / set-name uniqueness constraints on prior runs.
     import time as _t

{simpleaudit_studio-0.2.4 → simpleaudit_studio-0.2.6}/deploy/mock_openai_server.py

@@ -23,11 +23,7 @@ import re
 import threading
 import time
 import uuid
-
-try:
-    from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer
-except Exception:  # pragma: no cover - stdlib always present
-    raise
+from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer
 
 JUDGE_JSON_RE = re.compile(r"\{.*\}", re.DOTALL)
 
@@ -91,11 +87,11 @@ def _completion_body(messages, model) -> dict:
 
 
 class Handler(BaseHTTPRequestHandler):
-    def log_message(self, fmt, *args):  # noqa: N802 - stdlib signature
+    def log_message(self, fmt, *args):
         # Keep logs quiet but useful.
         print("[mock-openai] %s" % (fmt % args), flush=True)
 
-    def do_GET(self):  # noqa: N802
+    def do_GET(self):
         if self.path.rstrip("/") in ("", "/healthz"):
             self._send_json({"status": "ok"})
         elif self.path.rstrip("/") == "/v1/models":
@@ -108,7 +104,7 @@ class Handler(BaseHTTPRequestHandler):
         else:
             self._send_json({"error": "not found"}, status=404)
 
-    def do_POST(self):  # noqa: N802
+    def do_POST(self):
         length = int(self.headers.get("Content-Length", 0) or 0)
         raw = self.rfile.read(length) if length else b""
         try:

{simpleaudit_studio-0.2.4 → simpleaudit_studio-0.2.6}/deploy/seed_bulk.py

@@ -11,23 +11,29 @@ Run inside the web container:
 Or locally with SQLite:
     SIMPLEAUDIT_LOCAL_SQLITE=1 .venv/bin/python manage.py shell < deploy/seed_bulk.py
 """
+import hashlib
 import os
-import sys
 import random
-import hashlib
-from datetime import datetime, timedelta
+from datetime import UTC, datetime, timedelta
 
 os.environ.setdefault("DJANGO_SETTINGS_MODULE", "config.settings")
 # Do NOT set SIMPLEAUDIT_LOCAL_SQLITE — use whatever DB the environment provides
 
 import django
+
 django.setup()
 
-from accounts.models import User, Project
-from scenarios.models import Scenario, ScenarioRevision, ScenarioSet, ScenarioSetVersion, ScenarioSetVersionItem
-from model_registry.models import ModelEndpoint
-from audits.models import AuditRun
+from accounts.models import Project, User
 from audits.events import ScenarioResult, append_event
+from audits.models import AuditRun
+from model_registry.models import ModelEndpoint
+from scenarios.models import (
+    Scenario,
+    ScenarioRevision,
+    ScenarioSet,
+    ScenarioSetVersion,
+    ScenarioSetVersionItem,
+)
 
 random.seed(42)
 
@@ -256,7 +262,7 @@ def main():
                 version=ver,
                 scenario_count=len(items_for_ver),
                 content_hash=make_hash(f"{st.name}-v{ver}-{random.random()}"),
-                published_at=datetime.now() - timedelta(days=random.randint(0, 90)),
+                published_at=datetime.now(UTC) - timedelta(days=random.randint(0, 90)),
                 published_by=admin,
             )
             version_batch.append((vs, items_for_ver))
@@ -293,7 +299,7 @@ def main():
     print(f"Audit runs before: {AuditRun.objects.count()}")
     all_versions = list(ScenarioSetVersion.objects.all())
     run_batch = []
-    base_time = datetime.now() - timedelta(days=90)
+    base_time = datetime.now(UTC) - timedelta(days=90)
 
     for i in range(500):
         status = random.choice(RUN_STATUS_DISTRIBUTION)

{simpleaudit_studio-0.2.4 → simpleaudit_studio-0.2.6}/docs/deployment.md

@@ -188,7 +188,7 @@ MAX_CONCURRENT_AUDITS=2
 MAX_SCENARIOS_PER_RUN=500
 
 # SimpleAudit engine: no env vars needed. It is a pip dependency pinned in
-# requirements.txt; provenance is read from installed package metadata.
+# pyproject.toml; provenance is read from installed package metadata.
 
 # Optional observability
 OTEL_EXPORTER_OTLP_ENDPOINT=

{simpleaudit_studio-0.2.4 → simpleaudit_studio-0.2.6}/e2e/test_ui.py

@@ -35,7 +35,7 @@ def wait_for_health(timeout: int = 60) -> None:
             resp = urllib.request.urlopen(f"{BASE_URL}/healthz", timeout=3)
             if resp.status == 200:
                 return
-        except Exception:
+        except Exception:  # noqa: BLE001,S110 - retry loop; not healthy yet
             pass
         time.sleep(2)
     raise RuntimeError(f"Web service did not become healthy within {timeout}s")
@@ -198,7 +198,7 @@ def main() -> int:
                 fn(page)
                 print(f"  PASS  {name}")
                 passed += 1
-            except Exception as e:
+            except Exception as e:  # noqa: BLE001 - test harness reports any failure
                 print(f"  FAIL  {name}: {e}")
                 failed += 1
                 errors.append((name, str(e)))

{simpleaudit_studio-0.2.4 → simpleaudit_studio-0.2.6}/infra/engine.py

@@ -18,7 +18,6 @@ from __future__ import annotations
 
 import asyncio
 import os
-import sys
 from typing import Any
 
 
@@ -29,18 +28,19 @@ class EngineError(RuntimeError):
 def _ensure_engine_available() -> None:
     """Ensure the SimpleAudit engine package is importable.
 
-    The engine is a normal pip dependency (see ``requirements.txt``); there is no
-    path-based fallback. If it is not installed (e.g. a web-only process or a
-    test environment without the engine), raise a clean ``EngineError`` so the
-    caller can record a durable failure instead of crashing on import.
+    The engine is a normal pip dependency (declared in ``pyproject.toml``);
+    there is no path-based fallback. If it is not installed (e.g. a web-only
+    process or a test environment without the engine), raise a clean
+    ``EngineError`` so the caller can record a durable failure instead of
+    crashing on import.
     """
     try:
         import simpleaudit  # noqa: F401
         return
     except ModuleNotFoundError as exc:
         raise EngineError(
-            "SimpleAudit engine is not installed. Install it from requirements.txt "
-            "(pip install -r requirements.txt) before running audits."
+            "SimpleAudit engine is not installed. Install the project "
+            "dependencies (uv sync) before running audits."
         ) from exc
 
 
@@ -168,7 +168,7 @@ def build_model_auditor(*, target: dict, auditor: dict, judge: dict, generation:
     _ensure_engine_available()
     try:
         from simpleaudit.model_auditor import ModelAuditor
-    except Exception as exc:  # noqa: BLE001 - surface any import failure uniformly
+    except Exception as exc:
         raise EngineError(f"Failed to import SimpleAudit ModelAuditor: {exc}") from exc
 
     gen = dict(generation or {})
@@ -263,7 +263,7 @@ def build_model_auditor(*, target: dict, auditor: dict, judge: dict, generation:
         )
     except EngineError:
         raise
-    except Exception as exc:  # noqa: BLE001 - any construction failure (bad base_url, provider, etc.)
+    except Exception as exc:
         raise EngineError(f"Failed to construct ModelAuditor: {type(exc).__name__}: {exc}") from exc
     return instance, language
 
@@ -309,7 +309,7 @@ def run_scenario(
         )
     except EngineError:
         raise
-    except Exception as exc:  # noqa: BLE001 - unexpected engine crash
+    except Exception as exc:
         raise EngineError(f"Scenario execution crashed: {type(exc).__name__}: {exc}") from exc
 
     payload = result.to_dict()

{simpleaudit_studio-0.2.4 → simpleaudit_studio-0.2.6}/infra/health.py

@@ -70,7 +70,7 @@ def _postgres_probe() -> dict[str, Any]:
             row = cursor.fetchone()
             if row:
                 signals["db_size_bytes"] = int(row[0])
-    except Exception:  # noqa: BLE001 - non-Postgres or permission issue
+    except Exception:  # noqa: BLE001,S110 - non-Postgres or permission issue; size is optional
         pass
     return {"status": "up", **signals}
 
@@ -177,7 +177,7 @@ def _engine_probe() -> dict[str, Any]:
 
 def _model_endpoints_probe() -> dict[str, Any]:
     """Ping each registered model endpoint's base_url, grouped by connection."""
-    from model_registry.models import ModelConnection, RegisteredModel
+    from model_registry.models import ModelConnection
 
     groups: list[dict[str, Any]] = []
 

{simpleaudit_studio-0.2.4 → simpleaudit_studio-0.2.6}/infra/logging.py

@@ -1,13 +1,13 @@
 """Structured JSON logging for containers."""
 import json
 import logging
-from datetime import datetime, timezone
+from datetime import UTC, datetime
 
 
 class JsonFormatter(logging.Formatter):
     def format(self, record: logging.LogRecord) -> str:
         payload = {
-            "timestamp": datetime.now(timezone.utc).isoformat(),
+            "timestamp": datetime.now(UTC).isoformat(),
             "level": record.levelname,
             "logger": record.name,
             "message": record.getMessage(),

{simpleaudit_studio-0.2.4 → simpleaudit_studio-0.2.6}/infra/management/commands/purge_test_data.py

@@ -14,7 +14,12 @@ from django.db import transaction
 from audits.events import AuditEvent, ScenarioResult
 from audits.models import AuditRun
 from model_registry.models import ModelEndpoint
-from scenarios.models import Scenario, ScenarioSet, ScenarioSetVersion, ScenarioSetVersionItem
+from scenarios.models import (
+    Scenario,
+    ScenarioSet,
+    ScenarioSetVersion,
+    ScenarioSetVersionItem,
+)
 
 
 class Command(BaseCommand):

{simpleaudit_studio-0.2.4 → simpleaudit_studio-0.2.6}/infra/management/commands/run_worker.py

@@ -7,7 +7,6 @@ The worker connects to the external Hatchet server (HATCHET_SERVER_URL /
 HATCHET_GRPC_URL) and executes audit scenario/finalize tasks. It runs in its own
 process, separate from the web/API server, which never executes model calls.
 """
-import os
 
 from django.conf import settings
 from django.core.management.base import BaseCommand

{simpleaudit_studio-0.2.4 → simpleaudit_studio-0.2.6}/infra/management/commands/seed_demo_audits.py

@@ -54,9 +54,10 @@ class Command(BaseCommand):
         )
 
     def handle(self, *args, **options):
-        from accounts.models import Project
         from django.contrib.auth import get_user_model
 
+        from accounts.models import Project
+
         User = get_user_model()
         project_id = options["project"]
         try:

{simpleaudit_studio-0.2.4 → simpleaudit_studio-0.2.6}/infra/management/commands/seed_platform.py

@@ -16,7 +16,11 @@ import logging
 
 from django.core.management.base import BaseCommand, CommandError
 
-from infra.seed import DEFAULT_PACKS, import_scenario_pack, seed_default_model_connections
+from infra.seed import (
+    DEFAULT_PACKS,
+    import_scenario_pack,
+    seed_default_model_connections,
+)
 
 logger = logging.getLogger(__name__)
 
@@ -47,9 +51,10 @@ class Command(BaseCommand):
         )
 
     def handle(self, *args, **options):
-        from accounts.models import Project
         from django.contrib.auth import get_user_model
 
+        from accounts.models import Project
+
         User = get_user_model()
 
         project_id = options["project"]