From 0250c8373c8ea12c6d12cf399a15c57f8690b032 Mon Sep 17 00:00:00 2001 From: 苏向夜 Date: Fri, 26 Jan 2024 15:04:21 +0800 Subject: :recycle: refactor(output): add __init__ method in Output class in order to generate output mehtod in initialization --- src/infini/interceptor.py | 2 +- src/infini/output.py | 19 ++++++++++++++----- 2 files changed, 15 insertions(+), 6 deletions(-) (limited to 'src') 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" -- cgit v1.2.3-70-g09d2