From 1b480cf3ceb1f76029cec399bcbf55a658d8700d Mon Sep 17 00:00:00 2001
From: rootflo-hardik <hardik@rootflo.ai>
Date: Fri, 12 Sep 2025 11:18:10 +0530
Subject: [PATCH] exporting arium event types from flo_ai

---
 flo_ai/flo_ai/__init__.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/flo_ai/flo_ai/__init__.py b/flo_ai/flo_ai/__init__.py
index 643c424d..3965e83f 100644
--- a/flo_ai/flo_ai/__init__.py
+++ b/flo_ai/flo_ai/__init__.py
@@ -24,6 +24,9 @@
     EndNode,
     Edge,
     AriumBuilder,
+    AriumEvent,
+    AriumEventType,
+    default_event_callback,
 )
 
 # Utils package - Utility functions
@@ -64,6 +67,10 @@
     # Builder
     'AgentBuilder',
     'AriumBuilder',
+    # Arium Event system
+    'AriumEventType',
+    'AriumEvent',
+    'default_event_callback',
 ]
 
 __version__ = '1.0.0'