donkey-kit 0.1.1__py3-none-any.whl

donkey_kit/__init__.py

@@ -0,0 +1,102 @@
+"""donkey-kit — an SDK for consuming Agent Fabric
+capabilities from your own agent framework.
+
+See the README for the maintainer + support statement and the trademark note.
+"Agent Fabric" is a MuleSoft product name; this package is descriptive.
+
+Public surface (BG §1.8):
+
+    from donkey_kit import Donkey
+    donkey = Donkey.from_env()
+    donkey.langgraph.chat_model("gpt-4o")   # native ChatOpenAI
+
+Working-instruction reminder (verification discipline, #2): many platform endpoints/headers/class
+names are UNVERIFIED. Those code paths raise
+``NotImplementedError("blocked on verification: …")`` rather than guessing. See
+docs/verified-apis.md.
+"""
+
+from __future__ import annotations
+
+from .core.budget import Budget
+from .core.cachecontrol import CacheControls
+from .core.config import DonkeyConfig, Region
+from .core.cost import CostTags
+from .core.errors import (
+    AgentKilled,
+    AuthError,
+    BudgetReserveReached,
+    ConfigError,
+    ContentSafetyBlocked,
+    DonkeyError,
+    GatewayUnavailable,
+    GovernanceDrift,
+    ModelNotRoutable,
+    ModelSubstituted,
+    PIIDetected,
+    PlatformTeamOnly,
+    PolicyViolation,
+    PromptInjectionBlocked,
+    ProvisioningError,
+    PublicationDrift,
+    RegistryError,
+    TokenBudgetExceeded,
+    ToolInvocationError,
+    UpstreamModelError,
+    UpstreamRequestError,
+    classify,
+)
+from .core.toolspec import ToolSpec, registered_tools
+from .donkey import Donkey
+from .registry import (
+    STRICT,
+    AssetRef,
+    AssetType,
+    Contact,
+    GovernanceCriteria,
+    Publication,
+    PublicationAssetType,
+)
+
+__version__ = "0.1.1"
+
+__all__ = [
+    "STRICT",
+    "AgentKilled",
+    "AssetRef",
+    "AssetType",
+    "AuthError",
+    "Budget",
+    "BudgetReserveReached",
+    "CacheControls",
+    "ConfigError",
+    "Contact",
+    "ContentSafetyBlocked",
+    "CostTags",
+    "Donkey",
+    "DonkeyConfig",
+    "DonkeyError",
+    "GatewayUnavailable",
+    "GovernanceCriteria",
+    "GovernanceDrift",
+    "ModelNotRoutable",
+    "ModelSubstituted",
+    "PIIDetected",
+    "PlatformTeamOnly",
+    "PolicyViolation",
+    "PromptInjectionBlocked",
+    "Publication",
+    "PublicationAssetType",
+    "ProvisioningError",
+    "PublicationDrift",
+    "Region",
+    "RegistryError",
+    "TokenBudgetExceeded",
+    "ToolInvocationError",
+    "ToolSpec",
+    "UpstreamModelError",
+    "UpstreamRequestError",
+    "__version__",
+    "classify",
+    "registered_tools",
+]

donkey_kit/conformance/__init__.py

@@ -0,0 +1,83 @@
+"""The customer-facing conformance kit (#191, `BG §1.5`).
+
+The public, run-it-against-*your*-agent sibling of the SDK's internal adapter
+matrix (``tests/conformance/``). Install ``donkey-kit[test]`` and run::
+
+    pytest --donkey-conformance --agent=my_app.agent:build
+
+- :mod:`~donkey_kit.conformance.suite` — the four scenarios and their types.
+- :mod:`~donkey_kit.conformance.harness` — the observable ``Donkey`` the
+  scenarios run against, plus :func:`run_conformance` for programmatic use.
+- :mod:`~donkey_kit.conformance.report` — the dependency-free status table.
+- :mod:`~donkey_kit.conformance.plugin` — the pytest11 plugin (auto-loaded).
+
+**Dev-only sibling of the five production layers** (the layered architecture): an import-linter
+contract forbids ``core``/``llm``/``registry``/``tools``/``integrations`` from
+importing it. Because the pytest11 entry point auto-loads
+:mod:`~donkey_kit.conformance.plugin` on *every* pytest run where this package
+is installed — including the base-only job — only the framework-free
+``suite``/``report`` submodules are imported at package top. The harness (which
+imports :class:`~donkey_kit.donkey.Donkey`) is exposed lazily via
+:func:`__getattr__`, so ``import donkey_kit.conformance`` never eagerly drags
+the full SDK — or a framework — onto the base import path.
+"""
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, Any
+
+from .report import StatusCounts, count_statuses, render_report
+from .suite import (
+    PROBE_CORRELATION_ID,
+    SCENARIO_NAMES,
+    SCENARIOS,
+    Observation,
+    Outcome,
+    Result,
+    Scenario,
+    ScenarioContext,
+    Status,
+    validate_known_limitations,
+)
+
+if TYPE_CHECKING:  # for type-checkers only; not imported at runtime module top
+    from .harness import (
+        ConformanceHarness,
+        ConformanceUsageError,
+        run_conformance,
+    )
+
+__all__ = [
+    "PROBE_CORRELATION_ID",
+    "SCENARIOS",
+    "SCENARIO_NAMES",
+    "ConformanceHarness",
+    "ConformanceUsageError",
+    "Observation",
+    "Outcome",
+    "Result",
+    "Scenario",
+    "ScenarioContext",
+    "Status",
+    "StatusCounts",
+    "count_statuses",
+    "render_report",
+    "run_conformance",
+    "validate_known_limitations",
+]
+
+# The harness pulls in Donkey; import it only when actually asked for, so the
+# auto-loaded plugin path stays light (mirrors the lazy-adapter discipline).
+_LAZY = {"ConformanceHarness", "ConformanceUsageError", "run_conformance"}
+
+
+def __getattr__(name: str) -> Any:
+    if name in _LAZY:
+        from . import harness
+
+        return getattr(harness, name)
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
+
+
+def __dir__() -> list[str]:
+    return sorted(__all__)

donkey_kit/conformance/gateway.py

@@ -0,0 +1,217 @@
+"""An out-of-process local gateway simulator a test can assert against (#278, `BG §1.5`).
+
+``simulate()`` (#190) and the in-process ``donkey`` fixture (#191) cover anything
+that calls the SDK in-process. They cannot cover the cases the vision deck names
+explicitly — a containerised agent, a Node service, an A2A client, a manual
+``curl`` — none of which import ``donkey_kit``. Those need a **real listener on a
+real port**, plus a way to ask the gateway what it *received* ("did my agent stop
+after the refusal, or retry four more times?").
+
+This module boots the same local gateway simulator
+(:func:`donkey_kit.simulator.build_app`, BG §1.4) that ``simulate()`` and
+``donkey mock`` replay — so a stock client sees byte-identical rejection shapes —
+on an **ephemeral port** (bind ``0``) in a daemon thread, and wraps it in a
+request-recording ASGI spy the test can assert on. The public entry point is the
+``gateway`` pytest fixture in :mod:`~donkey_kit.conformance.plugin`; this module
+is imported lazily from there so the auto-loaded plugin never drags ``uvicorn``
+onto every pytest run.
+
+Framework isolation (the layered architecture): ``uvicorn`` is imported lazily
+inside :func:`start_gateway`, and ``starlette`` only inside the simulator's own
+``build_app`` — so ``import donkey_kit.conformance.gateway`` stays green under the
+base-only CI job (``[dev]`` only, no ``[local]`` extra). Taking the fixture
+without that extra raises an :class:`ImportError` naming the exact
+``pip install`` — never a bare ``ModuleNotFoundError`` and never a silent skip.
+"""
+
+from __future__ import annotations
+
+import threading
+import time
+from collections.abc import Awaitable, Callable, Mapping, MutableMapping
+from dataclasses import dataclass
+from typing import Any
+
+from ..simulator.app import ASGIApp, SimulatorConfig, build_app
+from ..simulator.scenarios import Scenario, parse_scenario
+
+__all__ = ["Gateway", "RecordedRequest", "start_gateway"]
+
+# The one extra that makes the out-of-process path real: starlette (the ASGI app)
+# + uvicorn (the server). Both live in `[local]`, so the hint names that extra —
+# not `[test]`, which ships the fixture itself but not the server it drives.
+_LOCAL_EXTRA_HINT = (
+    "The `gateway` fixture needs the local simulator server extra "
+    "(starlette + uvicorn). Install it with:\n"
+    '    pip install "donkey-kit[local]"'
+)
+
+# Request headers whose values are redacted in the spy: the consumer-auth secret
+# (BG §1.1 — the `client_secret` request header the transport sends) and any
+# bearer token. Matched case-insensitively; ASGI already lowercases header names.
+_REDACTED_HEADERS = frozenset({"client_secret", "authorization"})
+_REDACTED = "***"
+
+# How long to wait for uvicorn to report `started` before giving up.
+_STARTUP_TIMEOUT_S = 10.0
+
+
+@dataclass(frozen=True)
+class RecordedRequest:
+    """One request the gateway received, as seen from the gateway's side.
+
+    ``headers`` has the consumer-auth secret redacted (see :data:`_REDACTED_HEADERS`),
+    so a captured request is safe to print in an assertion failure or a fixture
+    dump. It does not carry the body — the simulator replays captured shapes and
+    does not evaluate request content (#189/#250)."""
+
+    method: str
+    path: str
+    headers: Mapping[str, str]
+
+
+def _record(scope: MutableMapping[str, Any]) -> RecordedRequest:
+    """Build a :class:`RecordedRequest` from an ASGI ``http`` scope, redacting
+    the consumer-auth secret and any bearer header."""
+    raw: Any = scope.get("headers") or []
+    headers: dict[str, str] = {}
+    for name, value in raw:
+        key = name.decode("latin-1")
+        headers[key] = _REDACTED if key.lower() in _REDACTED_HEADERS else value.decode("latin-1")
+    return RecordedRequest(
+        method=str(scope.get("method", "")),
+        path=str(scope.get("path", "")),
+        headers=headers,
+    )
+
+
+class _RecordingApp:
+    """An ASGI app that records every HTTP request, then delegates to a swappable
+    inner app. The inner app is swapped (never the outer one uvicorn holds) so
+    :meth:`Gateway.set_scenarios` can reconfigure the simulator live, without
+    restarting the server or losing the request log."""
+
+    def __init__(self, app: ASGIApp) -> None:
+        self.app = app
+        self.requests: list[RecordedRequest] = []
+
+    async def __call__(
+        self,
+        scope: MutableMapping[str, Any],
+        receive: Callable[[], Awaitable[MutableMapping[str, Any]]],
+        send: Callable[[MutableMapping[str, Any]], Awaitable[None]],
+    ) -> None:
+        if scope.get("type") == "http":
+            self.requests.append(_record(scope))
+        await self.app(scope, receive, send)
+
+
+class Gateway:
+    """A running out-of-process simulator plus the spy over what it received.
+
+    Yielded by the ``gateway`` pytest fixture. ``.url`` is a real
+    ``http://127.0.0.1:<port>`` any process can point ``DONKEY_LLM_PROXY_URL`` at;
+    the simulator replays the same captured fixtures the SDK's typed refusals are
+    tested against, and every response carries ``x-donkey-simulator: true`` (#189).
+    """
+
+    def __init__(
+        self, *, url: str, recorder: _RecordingApp, shutdown: Callable[[], None]
+    ) -> None:
+        self._url = url
+        self._recorder = recorder
+        self._shutdown = shutdown
+
+    @property
+    def url(self) -> str:
+        """The base URL of the running simulator, e.g. ``http://127.0.0.1:54123``."""
+        return self._url
+
+    @property
+    def requests_received(self) -> int:
+        """How many requests the gateway has received since start (or the last
+        :meth:`reset`)."""
+        return len(self._recorder.requests)
+
+    @property
+    def requests(self) -> list[RecordedRequest]:
+        """A snapshot of the recorded requests, oldest first (secret redacted)."""
+        return list(self._recorder.requests)
+
+    def set_scenarios(self, *scenarios: str | Scenario) -> None:
+        """Reconfigure the simulator's #188 fault-injection scenarios for this
+        test, live (no restart). Each argument is either a parsed
+        :class:`~donkey_kit.simulator.scenarios.Scenario` or a ``--scenario`` spec
+        string (e.g. ``"pii_block:every=1"``). Scenarios are stateful and
+        single-use, so this builds a fresh simulator each call; the request log is
+        left untouched (use :meth:`reset` to clear it)."""
+        parsed = tuple(
+            parse_scenario(s) if isinstance(s, str) else s for s in scenarios
+        )
+        self._recorder.app = build_app(SimulatorConfig(scenarios=parsed))
+
+    def reset(self) -> None:
+        """Clear the recorded-request log (the running server is left alone)."""
+        self._recorder.requests.clear()
+
+    def close(self) -> None:
+        """Stop the server and join its thread. Idempotent; the fixture calls it
+        on teardown, including when the test failed."""
+        self._shutdown()
+
+
+def _bound_port(server: Any) -> int:
+    """Read the ephemeral port uvicorn actually bound. Valid only once
+    ``server.started`` is true (that is when ``server.servers`` is populated),
+    which is what makes ``port=0`` race-free — we never guess a port."""
+    for srv in getattr(server, "servers", None) or []:
+        for sock in getattr(srv, "sockets", None) or []:
+            return int(sock.getsockname()[1])
+    raise RuntimeError("could not determine the local gateway simulator's bound port")
+
+
+def start_gateway(
+    config: SimulatorConfig | None = None,
+    *,
+    host: str = "127.0.0.1",
+    startup_timeout: float = _STARTUP_TIMEOUT_S,
+) -> Gateway:
+    """Boot the local gateway simulator on an ephemeral port and return a
+    :class:`Gateway` handle. The caller owns teardown via :meth:`Gateway.close`.
+
+    Raises :class:`ImportError` (naming ``pip install "donkey-kit[local]"``) when
+    the ``[local]`` extra — ``starlette`` + ``uvicorn`` — is not installed.
+    """
+    try:
+        import uvicorn  # lazy: provided by the [local] extra
+    except ImportError as exc:
+        raise ImportError(_LOCAL_EXTRA_HINT) from exc
+
+    try:
+        # build_app imports starlette lazily; a missing [local] surfaces here too.
+        recorder = _RecordingApp(build_app(config))
+    except ImportError as exc:
+        raise ImportError(_LOCAL_EXTRA_HINT) from exc
+
+    uv_config = uvicorn.Config(recorder, host=host, port=0, log_level="warning")
+    server = uvicorn.Server(uv_config)
+    thread = threading.Thread(target=server.run, daemon=True)
+    thread.start()
+
+    deadline = time.time() + startup_timeout
+    while not server.started and time.time() < deadline:
+        time.sleep(0.02)
+    if not server.started:
+        server.should_exit = True
+        thread.join(timeout=5.0)
+        raise RuntimeError(
+            f"local gateway simulator did not start within {startup_timeout}s"
+        )
+
+    port = _bound_port(server)
+
+    def _shutdown() -> None:
+        server.should_exit = True
+        thread.join(timeout=10.0)
+
+    return Gateway(url=f"http://{host}:{port}", recorder=recorder, shutdown=_shutdown)