fixed minor bugs

Author: GioeleB00

src/asyncflow/components/__init__.py

@@ -1,9 +1,9 @@
 """Public components: re-exports Pydantic schemas (topology)."""
 from __future__ import annotations
 
+from asyncflow.schemas.event.injection import EventInjection
 from asyncflow.schemas.topology.edges import Edge
 from asyncflow.schemas.topology.endpoint import Endpoint
-from asyncflow.schemas.event.injection import EventInjection
 from asyncflow.schemas.topology.nodes import (
     Client,
     LoadBalancer,
@@ -12,13 +12,13 @@
 )
 
 __all__ = [
-    "Client", 
-    "Edge", 
+    "Client",
+    "Edge",
+    "Endpoint",
     "EventInjection",
-    "Endpoint", 
-    "LoadBalancer", 
-    "Server", 
-    "ServerResources"
+    "LoadBalancer",
+    "Server",
+    "ServerResources",
     ]
 
 

tests/unit/public_api/test_import.py

@@ -14,6 +14,7 @@
     Client,
     Edge,
     Endpoint,
+    EventInjection,
     LoadBalancer,
     Server,
     ServerResources,
@@ -43,6 +44,7 @@ def test_components_public_symbols() -> None:
         "Client",
         "Edge",
         "Endpoint",
+        "EventInjection",
         "LoadBalancer",
         "Server",
         "ServerResources",
@@ -57,6 +59,7 @@ def test_components_symbols_are_importable_classes() -> None:
         (Client, "Client"),
         (Edge, "Edge"),
         (Endpoint, "Endpoint"),
+        (EventInjection, "EventInjection"),
         (LoadBalancer, "LoadBalancer"),
         (Server, "Server"),
         (ServerResources, "ServerResources"),