Add LocalStructureRunDriver.event_listeners to allow passing Event …

…Listeners to be active for a Structure's run

CHANGELOG.md

@@ -13,7 +13,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Added
 - Parameter `meta: dict` on `BaseEvent`.
 - `AzureOpenAiTextToSpeechDriver`.
-- Ability to use Event Listeners as Context Managers for temporarily setting the Event Bus listeners.
+- `LocalStructureRunDriver.event_listeners` for adding Event Listeners to a local Structure run.
 
 ### Changed
 - **BREAKING**: Drivers, Loaders, and Engines now raise exceptions rather than returning `ErrorArtifacts`.

griptape/drivers/structure_run/local_structure_run_driver.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import os
+from contextlib import ExitStack
 from typing import TYPE_CHECKING, Callable
 
 from attrs import define, field
@@ -9,18 +10,31 @@
 from griptape.drivers.structure_run.base_structure_run_driver import BaseStructureRunDriver
 
 if TYPE_CHECKING:
+    from griptape.events import EventListener
     from griptape.structures import Structure
 
 
 @define
 class LocalStructureRunDriver(BaseStructureRunDriver):
+    """Runs a structure locally.
+
+    Attributes:
+        structure_factory_fn: A function that returns a Structure.
+        event_listeners: A list of Event Listeners to add to the Event Bus for the Structure's run.
+    """
+
     structure_factory_fn: Callable[[], Structure] = field(kw_only=True)
+    event_listeners: list[EventListener] = field(factory=list, kw_only=True)
 
     def try_run(self, *args: BaseArtifact) -> BaseArtifact:
         old_env = os.environ.copy()
         try:
             os.environ.update(self.env)
-            structure_factory_fn = self.structure_factory_fn().run(*[arg.value for arg in args])
+
+            with ExitStack() as stack:
+                for event_listener in self.event_listeners:
+                    stack.enter_context(event_listener)
+                structure_factory_fn = self.structure_factory_fn().run(*[arg.value for arg in args])
         finally:
             os.environ.clear()
             os.environ.update(old_env)