fix: mypy issues

This commit is contained in:
Yeuoly
2025-01-09 16:53:30 +08:00
parent 76e24d91c0
commit f748d6c7c4
49 changed files with 157 additions and 133 deletions

View File

@ -71,13 +71,13 @@ class Tool(ABC):
if isinstance(result, ToolInvokeMessage):
def single_generator():
def single_generator() -> Generator[ToolInvokeMessage, None, None]:
yield result
return single_generator()
elif isinstance(result, list):
def generator():
def generator() -> Generator[ToolInvokeMessage, None, None]:
yield from result
return generator()