diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/infini/handler.py | 2 | ||||
| -rw-r--r-- | src/infini/typing.py | 1 |
2 files changed, 2 insertions, 1 deletions
diff --git a/src/infini/handler.py b/src/infini/handler.py index 5a4e1649..79686485 100644 --- a/src/infini/handler.py +++ b/src/infini/handler.py @@ -12,7 +12,7 @@ class Handler: yield Output.empty() return while not queue.is_empty(): - if isinstance(stream := queue.pop()(input), Generator): # TODO 新增测试 + if isinstance(stream := queue.pop()(input), Generator): for output in stream: yield output if output.block: diff --git a/src/infini/typing.py b/src/infini/typing.py index d9db1e61..05eea05e 100644 --- a/src/infini/typing.py +++ b/src/infini/typing.py @@ -7,6 +7,7 @@ from typing import ( Literal as Literal, Sequence as Sequence, Generator as Generator, + overload as overload, TypeVar, TypedDict, Union, |