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'