fix(packaging): floor-only client pin, scope extra-files, prune heavy-wheel tests

Addresses review of the dual-wheel split (#370):

- adk: pin agentex-sdk-client floor-only (>=0.11.5). The two packages
  co-version and release-please can't rewrite a dep string, so a <0.12
  ceiling would make the first 0.12.0 cut unresolvable (slim is a new
  PyPI name with no 0.11.x to fall back to).
- adk: prune agentex/lib/** test files from the heavy wheel via a custom
  hatch build hook — force-include ignores `exclude` (hatchling #1395).
  Drops 14 test files; keeps the 138 .j2 templates and py.typed. The hook
  imports build-only hatchling, so it's excluded from pyright.
- release-please-config: scope extra-files (_version.py) to the slim
  package so a heavy-only release can't overwrite the slim's __version__.
- run_agent_test.sh: fail loud when a wheel is missing instead of
  silently testing the pre-installed SDK; fix the dead repo-root fallback
  glob (quoted inside ls).
- ci: add scripts/check-slim-deps guardrail asserting root pyproject keeps
  exactly the 6 slim deps — catches Stainless re-adding the ADK deps.
- requirements{,-dev}.lock: regenerate for the two-package workspace via
  rye sync — the locks still named the pre-split agentex-sdk and pinned
  openai-agents below the new >=0.14.3 floor.

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>

Author: max-parke-scale

.github/workflows/ci.yml

@@ -37,6 +37,9 @@ jobs:
       - name: Run lints
         run: ./scripts/lint
 
+      - name: Check slim dependency set
+        run: ./scripts/check-slim-deps
+
   build:
     if: (github.event_name == 'push' || github.event.pull_request.head.repo.fork) && (github.event_name != 'push' || github.event.head_commit.message != 'codegen metadata')
     timeout-minutes: 10

adk/hatch_build.py

@@ -0,0 +1,41 @@
+"""Builds the agentex/lib force-include map per-file so test files can be pruned
+— force-include ignores `exclude` (hatchling #1395)."""
+
+from __future__ import annotations
+
+import os
+
+from hatchling.builders.hooks.plugin.interface import BuildHookInterface
+
+_SKIP_DIRS = {"__pycache__", "tests"}
+_SKIP_NAMES = {"conftest.py", "pytest.ini", "run_tests.py"}
+# Floor below the ~333 shippable files: a collapse means the walk broke — fail
+# loud rather than ship a near-empty wheel.
+_MIN_FILES = 320
+
+
+def _is_test_file(name: str) -> bool:
+    return name in _SKIP_NAMES or (name.startswith("test_") and name.endswith(".py"))
+
+
+class CustomBuildHook(BuildHookInterface):
+    PLUGIN_NAME = "custom"
+
+    def initialize(self, version: str, build_data: dict) -> None:  # noqa: ARG002
+        lib_root = os.path.normpath(os.path.join(self.root, "..", "src", "agentex", "lib"))
+        force_include = build_data.setdefault("force_include", {})
+        collected = 0
+        for dirpath, dirnames, filenames in os.walk(lib_root):
+            dirnames[:] = [d for d in dirnames if d not in _SKIP_DIRS]
+            for name in filenames:
+                if _is_test_file(name):
+                    continue
+                src = os.path.join(dirpath, name)
+                rel = os.path.relpath(src, lib_root)
+                force_include[src] = os.path.join("agentex", "lib", rel)
+                collected += 1
+        if collected < _MIN_FILES:
+            raise RuntimeError(
+                f"agentex/lib force-include collected only {collected} files "
+                f"(expected >= {_MIN_FILES}); aborting build."
+            )

adk/pyproject.toml

@@ -16,12 +16,9 @@ authors = [
 readme = "README.md"
 
 dependencies = [
-    # Lockstep with this package's own version — both are co-released. The
-    # heavy package imports from `agentex.types.*` and `agentex.protocol.*`
-    # which ship from the slim, so a slim bump that drops/renames anything
-    # imported here would silently break heavy. Bump this range whenever the
-    # slim's major/minor changes.
-    "agentex-sdk-client>=0.11.4,<0.12",
+    # Co-released in lockstep; floor-only by design — a ceiling would
+    # eventually exclude the co-versioned slim (release-please can't bump it).
+    "agentex-sdk-client>=0.11.5",
     # CLI surface (agentex.lib.cli.*, agentex.lib.sdk.config.*)
     "typer>=0.16,<0.17",
     "questionary>=2.0.1,<3",
@@ -94,8 +91,10 @@ build-backend = "hatchling.build"
 [tool.hatch.build.targets.wheel]
 bypass-selection = true
 
-[tool.hatch.build.targets.wheel.force-include]
-"../src/agentex/lib" = "agentex/lib"
+# Builds the ../src/agentex/lib force-include map per-file (see hatch_build.py)
+# so test files can be pruned — force-include ignores `exclude` (hatchling #1395).
+[tool.hatch.build.targets.wheel.hooks.custom]
+path = "hatch_build.py"
 
 # Sdist deferred: hatchling can't represent the wheel's ../src/agentex/lib
 # force-include in an sdist include list. CI + bin/publish-pypi pass --wheel.

examples/tutorials/run_agent_test.sh

@@ -280,15 +280,17 @@ run_test() {
         local heavy_wheel slim_wheel
         heavy_wheel=$(ls /home/runner/work/*/*/adk/dist/agentex_sdk-*.whl 2>/dev/null | head -n1)
         if [[ -z "$heavy_wheel" ]]; then
-            heavy_wheel=$(ls "${SCRIPT_DIR}/../../adk/dist/agentex_sdk-*.whl" 2>/dev/null | head -n1)
+            heavy_wheel=$(ls "${SCRIPT_DIR}"/../../adk/dist/agentex_sdk-*.whl 2>/dev/null | head -n1)
         fi
         slim_wheel=$(ls /home/runner/work/*/*/dist/agentex_sdk_client-*.whl 2>/dev/null | head -n1)
         if [[ -z "$slim_wheel" ]]; then
-            slim_wheel=$(ls "${SCRIPT_DIR}/../../dist/agentex_sdk_client-*.whl" 2>/dev/null | head -n1)
+            slim_wheel=$(ls "${SCRIPT_DIR}"/../../dist/agentex_sdk_client-*.whl 2>/dev/null | head -n1)
         fi
-        if [[ -n "$heavy_wheel" && -n "$slim_wheel" ]]; then
-            pytest_cmd=("uv" "run" "--with" "$heavy_wheel" "--with" "$slim_wheel" "--with" "pytest" "--with" "pytest-asyncio" "pytest")
+        if [[ -z "$heavy_wheel" || -z "$slim_wheel" ]]; then
+            echo -e "${RED}❌ BUILD_CLI=true but a wheel is missing (heavy='${heavy_wheel}' slim='${slim_wheel}'); refusing to test against the pre-installed SDK${NC}"
+            return 1
         fi
+        pytest_cmd=("uv" "run" "--with" "$heavy_wheel" "--with" "$slim_wheel" "--with" "pytest" "--with" "pytest-asyncio" "pytest")
     fi
 
     local max_retries=5

pyproject.toml

@@ -186,6 +186,8 @@ exclude = [
     # Exclude autogenerated Stainless code from type checking
     "src/agentex/resources",
     "src/agentex/types",
+    # Build-time hook; imports hatchling, a build dep absent from the synced env.
+    "adk/hatch_build.py",
 ]
 
 reportImplicitOverride = true

release-please-config.json

@@ -1,7 +1,10 @@
 {
   "packages": {
     ".": {
-      "component": "agentex-sdk-client"
+      "component": "agentex-sdk-client",
+      "extra-files": [
+        "src/agentex/_version.py"
+      ]
     },
     "adk": {
       "component": "agentex-sdk"
@@ -64,8 +67,5 @@
       "hidden": true
     }
   ],
-  "release-type": "python",
-  "extra-files": [
-    "src/agentex/_version.py"
-  ]
+  "release-type": "python"
 }