griptape-nodes 0.58.1__py3-none-any.whl → 0.59.1__py3-none-any.whl

griptape_nodes/retained_mode/managers/event_manager.py

@@ -15,9 +15,9 @@ from griptape_nodes.retained_mode.events.base_events import (
     EventResultFailure,
     EventResultSuccess,
     FlushParameterChangesRequest,
+    FlushParameterChangesResultSuccess,
     RequestPayload,
     ResultPayload,
-    WorkflowAlteredMixin,
 )
 from griptape_nodes.utils.async_utils import call_function
 
@@ -28,6 +28,18 @@ if TYPE_CHECKING:
 RP = TypeVar("RP", bound=RequestPayload, default=RequestPayload)
 AP = TypeVar("AP", bound=AppPayload, default=AppPayload)
 
+# Result types that should NOT trigger a flush request.
+#
+# After most requests complete, we queue a FlushParameterChangesRequest to flush any tracked
+# parameter changes to the UI. However, when the FlushParameterChangesRequest itself completes,
+# it returns a FlushParameterChangesResultSuccess. If we don't exclude this result type, it would
+# trigger another FlushParameterChangesRequest, which would return another FlushParameterChangesResultSuccess,
+# creating an infinite loop of flush requests.
+#
+# Add result types to this set if they should never trigger a flush (typically because they ARE
+# the flush operation itself, or other internal operations that don't modify workflow state).
+RESULT_TYPES_THAT_SKIP_FLUSH = {FlushParameterChangesResultSuccess}
+
 
 class ResultContext(TypedDict, total=False):
     response_topic: str | None
@@ -223,13 +235,9 @@ class EventManager:
         # Actually make the handler callback (support both sync and async):
         result_payload: ResultPayload = await call_function(callback, request)
 
-        # Handle workflow alteration events for async context
+        # Queue flush request for async context (unless result type should skip flush)
         with operation_depth_mgr:
-            if (
-                result_payload.succeeded()
-                and isinstance(result_payload, WorkflowAlteredMixin)
-                and not self._flush_in_queue
-            ):
+            if not self._flush_in_queue and type(result_payload) not in RESULT_TYPES_THAT_SKIP_FLUSH:
                 await self.aput_event(EventRequest(request=FlushParameterChangesRequest()))
                 self._flush_in_queue = True
 
@@ -276,13 +284,9 @@ class EventManager:
         else:
             result_payload: ResultPayload = callback(request)
 
-        # Handle workflow alteration events for sync context
+        # Queue flush request for sync context (unless result type should skip flush)
         with operation_depth_mgr:
-            if (
-                result_payload.succeeded()
-                and isinstance(result_payload, WorkflowAlteredMixin)
-                and not self._flush_in_queue
-            ):
+            if not self._flush_in_queue and type(result_payload) not in RESULT_TYPES_THAT_SKIP_FLUSH:
                 self.put_event(EventRequest(request=FlushParameterChangesRequest()))
                 self._flush_in_queue = True