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',