From d733930a2b1876340039d90f19ece81fab0d078d Mon Sep 17 00:00:00 2001
From: Heiner Lohaus <hlohaus@users.noreply.github.com>
Date: Fri, 23 Feb 2024 02:51:10 +0100
Subject: Fix unittests, use Union typing

---
 g4f/stubs.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

(limited to 'g4f/stubs.py')

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
 
-- 
cgit v1.2.3