Merge branch 'main' into claude/issue-154-20251028-1915

Author: strawgate

.github/workflows/test.yml

@@ -30,6 +30,52 @@ jobs:
           config: '.markdownlint.jsonc'
           globs: '**/*.md'
 
+  codegen_check:
+    runs-on: ubuntu-latest
+    permissions:
+      contents: read
+
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+
+      - name: "Install uv"
+        uses: astral-sh/setup-uv@v6
+
+      - name: "Set up Python 3.10"
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.10"
+
+      - name: "Install Node.js"
+        uses: actions/setup-node@v4
+        with:
+          node-version: "20"
+
+      - name: "Install dependencies"
+        run: make sync
+
+      - name: "Run codegen"
+        run: make codegen
+
+      - name: "Run lint"
+        run: make lint
+
+      - name: "Check for uncommitted changes"
+        run: |
+          if ! git diff --exit-code; then
+            echo "Error: Modified files detected after running 'make codegen lint'."
+            echo "Please run 'make codegen lint' locally and commit the changes."
+            exit 1
+          fi
+          if [ -n "$(git ls-files --others --exclude-standard)" ]; then
+            echo "Error: Untracked files detected after running 'make codegen lint'."
+            echo "Untracked files:"
+            git ls-files --others --exclude-standard
+            echo "Please run 'make codegen lint' locally and commit the changes."
+            exit 1
+          fi
+
   static_analysis:
     runs-on: ubuntu-latest
 
@@ -66,6 +112,7 @@ jobs:
   test_quick:
     needs:
       - markdown_lint
+      - codegen_check
       - static_analysis
     timeout-minutes: 10
     strategy:

AGENTS.md

@@ -216,6 +216,11 @@ make sync
 GitHub Actions workflows are in `.github/workflows/`:
 
 - `test.yml` - Run tests across packages
+  - `codegen_check` job - Ensures `make codegen lint` has been run before
+    commits
+  - `static_analysis` job - Runs linting and type checking per package
+  - `test_quick` and `test_all` jobs - Run tests across Python versions and
+    platforms
 - `publish.yml` - Publish packages to PyPI
 - `claude-on-mention.yml` - Claude Code assistant (can make PRs)
 - `claude-on-open-label.yml` - Claude triage assistant (read-only analysis)

DEVELOPING.md

@@ -100,6 +100,11 @@ make precommit
 
 This runs linting, type checking, and code generation.
 
+**Important**: CI will fail if `make codegen lint` has not been run before
+committing. The `codegen_check` job in the test workflow verifies that
+running these commands produces no file changes, ensuring all generated
+code and formatting is up to date.
+
 ## Using Makefile in CI
 
 The Makefile targets support per-project operations, making them

key-value/key-value-aio/src/key_value/aio/stores/mongodb/store.py

@@ -174,8 +174,12 @@ async def _setup_collection(self, *, collection: str) -> None:
 
         new_collection: AsyncCollection[dict[str, Any]] = await self._db.create_collection(name=collection)
 
+        # Index for efficient key lookups
         _ = await new_collection.create_index(keys="key")
 
+        # TTL index for automatic expiration of entries when expires_at is reached
+        _ = await new_collection.create_index(keys="expires_at", expireAfterSeconds=0)
+
         self._collections_by_name[collection] = new_collection
 
     @override

key-value/key-value-sync/src/key_value/sync/code_gen/stores/mongodb/store.py

@@ -181,8 +181,12 @@ def _setup_collection(self, *, collection: str) -> None:
 
         new_collection: Collection[dict[str, Any]] = self._db.create_collection(name=collection)
 
+        # Index for efficient key lookups
         _ = new_collection.create_index(keys="key")
 
+        # TTL index for automatic expiration of entries when expires_at is reached
+        _ = new_collection.create_index(keys="expires_at", expireAfterSeconds=0)
+
         self._collections_by_name[collection] = new_collection
 
     @override