Remove heartbeat from fetch

didx-xyz · Feb 16, 2025 · 7ad3f73 · 7ad3f73
1 parent 3da632f
commit 7ad3f73
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/waypoint/services/nats_service.py b/waypoint/services/nats_service.py
@@ -168,9 +168,7 @@ async def event_generator(*, subscription: JetStreamContext.PullSubscription):
                         break
 
                     try:
-                        messages = await subscription.fetch(
-                            batch=5, timeout=0.5, heartbeat=0.2
-                        )
+                        messages = await subscription.fetch(batch=5, timeout=0.5)
                         for message in messages:
                             event = orjson.loads(message.data)
                             bound_logger.trace("Received event: {}", event)
@@ -192,7 +190,8 @@ async def event_generator(*, subscription: JetStreamContext.PullSubscription):
                         except BadSubscriptionError as e:
                             # If we can't unsubscribe, log the error and continue
                             bound_logger.warning(
-                                "BadSubscriptionError unsubscribing from NATS: {}", e
+                                "BadSubscriptionError unsubscribing from NATS after subscription lost: {}",
+                                e,
                             )
 
                         subscription = await self._subscribe(