checkpoint: massive refactor

events.py

@@ -5,27 +5,28 @@ import asyncio
 from typing import Callable, Any, Dict, List, Tuple
 
 class EventEmitter:
-    """
+ """
     Simple event emitter for decoupled communication between modules.
     """
-    def __init__(self):
-        """Initializes the EventEmitter with an empty listener map."""
-        self._listeners: Dict[str, List[Callable]] = {}
 
-    def on(self, event_name: str, callback: Callable):
-        """
+ def __init__(self):
+  """Initializes the EventEmitter with an empty listener map."""
+  self._listeners: Dict[str, List[Callable]] = {}
+
+ def on(self, event_name: str, callback: Callable):
+  """
         Registers a callback for a specific event.
         
         Args:
             event_name: The name of the event to listen for.
             callback: The function to call when the event is emitted.
         """
-        if event_name not in self._listeners:
-            self._listeners[event_name] = []
-        self._listeners[event_name].append(callback)
+  if event_name not in self._listeners:
+   self._listeners[event_name] = []
+  self._listeners[event_name].append(callback)
 
-    def emit(self, event_name: str, *args: Any, **kwargs: Any):
-        """
+ def emit(self, event_name: str, *args: Any, **kwargs: Any):
+  """
         Emits an event, calling all registered callbacks.
         
         Args:
@@ -33,53 +34,55 @@ class EventEmitter:
             *args: Positional arguments to pass to callbacks.
             **kwargs: Keyword arguments to pass to callbacks.
         """
-        if event_name in self._listeners:
-            for callback in self._listeners[event_name]:
-                callback(*args, **kwargs)
+  if event_name in self._listeners:
+   for callback in self._listeners[event_name]:
+    callback(*args, **kwargs)
 
 class AsyncEventQueue:
-    """
+ """
     Asynchronous event queue for decoupled communication using asyncio.Queue.
     """
-    def __init__(self):
-        """Initializes the AsyncEventQueue with an internal asyncio.Queue."""
-        self._queue: asyncio.Queue = asyncio.Queue()
 
-    async def put(self, event_name: str, payload: Any = None):
-        """
+ def __init__(self):
+  """Initializes the AsyncEventQueue with an internal asyncio.Queue."""
+  self._queue: asyncio.Queue = asyncio.Queue()
+
+ async def put(self, event_name: str, payload: Any = None):
+  """
         Puts an event into the queue.
         
         Args:
             event_name: The name of the event.
             payload: Optional data associated with the event.
         """
-        await self._queue.put((event_name, payload))
+  await self._queue.put((event_name, payload))
 
-    async def get(self) -> Tuple[str, Any]:
-        """
+ async def get(self) -> Tuple[str, Any]:
+  """
         Gets an event from the queue.
         
         Returns:
             A tuple containing (event_name, payload).
         """
-        return await self._queue.get()
+  return await self._queue.get()
 
 class UserRequestEvent:
-    """
+ """
     Payload for a user request event.
     """
-    def __init__(self, prompt: str, stable_md: str, file_items: List[Any], disc_text: str, base_dir: str):
-        self.prompt = prompt
-        self.stable_md = stable_md
-        self.file_items = file_items
-        self.disc_text = disc_text
-        self.base_dir = base_dir
 
-    def to_dict(self) -> Dict[str, Any]:
-        return {
-            "prompt": self.prompt,
-            "stable_md": self.stable_md,
-            "file_items": self.file_items,
-            "disc_text": self.disc_text,
-            "base_dir": self.base_dir
-        }
+ def __init__(self, prompt: str, stable_md: str, file_items: List[Any], disc_text: str, base_dir: str):
+  self.prompt = prompt
+  self.stable_md = stable_md
+  self.file_items = file_items
+  self.disc_text = disc_text
+  self.base_dir = base_dir
+
+ def to_dict(self) -> Dict[str, Any]:
+  return {
+   "prompt": self.prompt,
+   "stable_md": self.stable_md,
+   "file_items": self.file_items,
+   "disc_text": self.disc_text,
+   "base_dir": self.base_dir
+  }