aboutsummaryrefslogtreecommitdiffstatshomepage
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/infini/interceptor.py2
-rw-r--r--src/infini/output.py19
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"