fix(capabilities): preserve experiment freshness argument (#1284)

Author: stateofkate

backend/api/services/agent_capabilities.py

@@ -468,6 +468,7 @@ def is_stale(
     current_hash: str,
     schema_version: int,
     task_version_id: str,
+    experiment_id: str | None = None,
 ) -> bool:
     """Whether a stored block must be regenerated. See ``stale_reason``."""
     return (
@@ -476,6 +477,7 @@ def is_stale(
             current_hash=current_hash,
             schema_version=schema_version,
             task_version_id=task_version_id,
+            experiment_id=experiment_id,
         )
         is not None
     )

backend/tests/test_agent_capabilities_endpoint.py

@@ -46,6 +46,38 @@ def test_not_stale_when_all_match():
     )
 
 
+def test_is_stale_when_experiment_changed():
+    block_meta = {
+        "cohort_hash": "aaa",
+        "schema_version": 1,
+        "task_version_id": "v1",
+        "experiment_id": "exp-1",
+    }
+    assert is_stale(
+        block_meta,
+        current_hash="aaa",
+        schema_version=1,
+        task_version_id="v1",
+        experiment_id="exp-2",
+    )
+
+
+def test_not_stale_when_experiment_matches():
+    block_meta = {
+        "cohort_hash": "aaa",
+        "schema_version": 1,
+        "task_version_id": "v1",
+        "experiment_id": "exp-1",
+    }
+    assert not is_stale(
+        block_meta,
+        current_hash="aaa",
+        schema_version=1,
+        task_version_id="v1",
+        experiment_id="exp-1",
+    )
+
+
 def test_missing_metadata_is_stale():
     assert (
         is_stale(None, current_hash="aaa", schema_version=1, task_version_id="v1")