fix(runners): satisfy pyink and mypy-diff for rewind receipt

Author: davidahmann

src/google/adk/events/event_actions.py

@@ -47,7 +47,7 @@ class EventCompaction(BaseModel):
   """The compacted content of the events."""
 
 
-class RewindAuditReceipt(BaseModel):
+class RewindAuditReceipt(BaseModel):  # type: ignore[misc]
   """Audit receipt metadata emitted for rewind operations."""
 
   model_config = ConfigDict(

src/google/adk/runners.py

@@ -628,20 +628,20 @@ def _build_rewind_audit_receipt(
     events_after = session.events[:rewind_event_index]
     boundary_after_invocation_id = None
     if rewind_event_index > 0:
-      boundary_after_invocation_id = (
-          session.events[rewind_event_index - 1].invocation_id
-      )
+      boundary_after_invocation_id = session.events[
+          rewind_event_index - 1
+      ].invocation_id
 
     history_before_hash = self._hash_rewind_events(events_before)
     history_after_hash = self._hash_rewind_events(events_after)
 
     receipt_payload = {
-      'rewind_before_invocation_id': rewind_before_invocation_id,
-      'boundary_after_invocation_id': boundary_after_invocation_id,
-      'events_before_rewind': len(events_before),
-      'events_after_rewind': len(events_after),
-      'history_before_hash': history_before_hash,
-      'history_after_hash': history_after_hash,
+        'rewind_before_invocation_id': rewind_before_invocation_id,
+        'boundary_after_invocation_id': boundary_after_invocation_id,
+        'events_before_rewind': len(events_before),
+        'events_after_rewind': len(events_after),
+        'history_before_hash': history_before_hash,
+        'history_after_hash': history_after_hash,
     }
     receipt_hash = self._hash_rewind_payload(receipt_payload)
 
@@ -659,18 +659,16 @@ def _hash_rewind_events(self, events: List[Event]) -> str:
     """Hashes event summaries for deterministic rewind audit receipts."""
     summarized_events = []
     for event in events:
-      summarized_events.append(
-          {
-              'event_id': event.id,
-              'invocation_id': event.invocation_id,
-              'author': event.author,
-              'state_delta': event.actions.state_delta,
-              'artifact_delta': event.actions.artifact_delta,
-              'rewind_before_invocation_id': (
-                  event.actions.rewind_before_invocation_id
-              ),
-          }
-      )
+      summarized_events.append({
+          'event_id': event.id,
+          'invocation_id': event.invocation_id,
+          'author': event.author,
+          'state_delta': event.actions.state_delta,
+          'artifact_delta': event.actions.artifact_delta,
+          'rewind_before_invocation_id': (
+              event.actions.rewind_before_invocation_id
+          ),
+      })
     return self._hash_rewind_payload({'events': summarized_events})
 
   def _hash_rewind_payload(self, payload: dict[str, Any]) -> str:

tests/unittests/runners/test_runner_rewind.py

@@ -287,6 +287,8 @@ async def test_rewind_receipt_hash_is_deterministic(self):
         rewind_before_invocation_id="invocation2",
     )
 
-    assert first_receipt.history_before_hash == second_receipt.history_before_hash
+    assert (
+        first_receipt.history_before_hash == second_receipt.history_before_hash
+    )
     assert first_receipt.history_after_hash == second_receipt.history_after_hash
     assert first_receipt.receipt_hash == second_receipt.receipt_hash