:recycle: refactor(output): add __init__ method in Output class in order to generate output mehtod in initialization

2 files changed, 15 insertions, 6 deletions

diff --git a/src/infini/interceptor.py b/src/infini/interceptor.py
index 54387281..c227fef7 100644
--- a/src/infini/interceptor.py
+++ b/src/infini/interceptor.py
@@ -1,6 +1,6 @@
 from infini.input import Input
 from infini.output import Output
-from infini.typing import List, RouterType, Callable, Generic, T, overload
+from infini.typing import List, RouterType, Callable
 from infini.queue import EventQueue
 
 
diff --git a/src/infini/output.py b/src/infini/output.py
index ddb7f0a5..92913b1e 100644
--- a/src/infini/output.py
+++ b/src/infini/output.py
@@ -7,13 +7,22 @@ class Output:
     status: int
     block: bool
 
+    def __init__(
+        self,
+        type: Literal["null", "text", "workflow"],
+        name: str,
+        *,
+        status: int = 0,
+        block: bool = False,
+    ) -> None:
+        self.type = type
+        self.name = name
+        self.status = status
+        self.block = block
+
     @classmethod
     def empty(cls) -> "Output":
-        output = cls()
-        output.type = "null"
-        output.status = 0
-        output.block = True
-        return output
+        return cls("null", "null", status=0, block=True)
 
     def is_empty(self) -> bool:
         return self.type == "null"