1 files changed, 7 insertions, 3 deletions

diff --git a/g4f/stubs.py b/g4f/stubs.py
index b9934b8c..49cf8a88 100644
--- a/g4f/stubs.py
+++ b/g4f/stubs.py
@@ -1,6 +1,8 @@
 
 from __future__ import annotations
 
+from typing import Union
+
 class Model():
     ...
 
@@ -52,7 +54,7 @@ class ChatCompletionChunk(Model):
         }
 
 class ChatCompletionMessage(Model):
-    def __init__(self, content: str | None):
+    def __init__(self, content: Union[str, None]):
         self.role = "assistant"
         self.content = content
 
@@ -72,7 +74,9 @@ class ChatCompletionChoice(Model):
         }
 
 class ChatCompletionDelta(Model):
-    def __init__(self, content: str | None):
+    content: Union[str, None] = None
+
+    def __init__(self, content: Union[str, None]):
         if content is not None:
             self.content = content
 
@@ -80,7 +84,7 @@ class ChatCompletionDelta(Model):
         return self.__dict__
 
 class ChatCompletionDeltaChoice(Model):
-    def __init__(self, delta: ChatCompletionDelta, finish_reason: str | None):
+    def __init__(self, delta: ChatCompletionDelta, finish_reason: Union[str, None]):
         self.delta = delta
         self.finish_reason = finish_reason