diff options
Diffstat (limited to 'g4f/typing.py')
| -rw-r--r-- | g4f/typing.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/g4f/typing.py b/g4f/typing.py index c93a4bcf..c972f505 100644 --- a/g4f/typing.py +++ b/g4f/typing.py @@ -1,5 +1,6 @@ import sys -from typing import Any, AsyncGenerator, Generator, NewType, Tuple, Union, List, Dict, Type +from typing import Any, AsyncGenerator, Generator, NewType, Tuple, Union, List, Dict, Type, IO +from PIL.Image import Image if sys.version_info >= (3, 8): from typing import TypedDict @@ -10,6 +11,7 @@ SHA256 = NewType('sha_256_hash', str) CreateResult = Generator[str, None, None] AsyncResult = AsyncGenerator[str, None] Messages = List[Dict[str, str]] +ImageType = Union[str, bytes, IO, Image, None] __all__ = [ 'Any', |