feat: feature flag io_publisher

Author: onur-askui

src/askui/chat/api/runs/events/io_publisher.py

@@ -5,18 +5,33 @@
 from typing import Any
 
 from askui.chat.api.runs.events.events import Event
+from askui.chat.api.settings import Settings
 
 
 class IOPublisher:
     """Publisher that serializes events to JSON and writes to stdout."""
 
+    def __init__(self, enabled: bool) -> None:
+        """
+        Initialize the IO publisher.
+
+        Args:
+            settings: The settings instance containing configuration for the IO publisher.
+        """
+        self._enabled = enabled
+
     def publish(self, event: Event) -> None:
         """
         Publish an event by serializing it to JSON and writing to stdout.
 
+        If the publisher is disabled, this method does nothing.
+
         Args:
             event: The event to publish
         """
+        if not self._enabled:
+            return
+
         try:
             event_dict: dict[str, Any] = event.model_dump(mode="json")
             event_json = json.dumps(event_dict)

src/askui/chat/api/runs/service.py

@@ -44,7 +44,7 @@ def __init__(
         self._chat_history_manager = chat_history_manager
         self._settings = settings
         self._event_service = EventService(settings.data_dir, self)
-        self._io_publisher = IOPublisher()
+        self._io_publisher = IOPublisher(settings.enable_io_events)
 
     def _find_by_id(
         self, workspace_id: WorkspaceId | None, thread_id: ThreadId, run_id: RunId

src/askui/chat/api/settings.py

@@ -118,3 +118,7 @@ class Settings(BaseSettings):
         default_factory=OtelSettings,
         description="OpenTelemetry configuration settings",
     )
+    enable_io_events: bool = Field(
+        default=False,
+        description="Whether to enable the publishing events to stdout",
+    )