summaryrefslogtreecommitdiffstats
path: root/g4f/stubs.py
diff options
context:
space:
mode:
authorHeiner Lohaus <hlohaus@users.noreply.github.com>2024-02-23 02:51:10 +0100
committerHeiner Lohaus <hlohaus@users.noreply.github.com>2024-02-23 02:51:10 +0100
commitd733930a2b1876340039d90f19ece81fab0d078d (patch)
tree208266ba04a3d4d056a85f20c104d324a1b625c8 /g4f/stubs.py
parentUse new client in inter api (diff)
downloadgpt4free-d733930a2b1876340039d90f19ece81fab0d078d.tar
gpt4free-d733930a2b1876340039d90f19ece81fab0d078d.tar.gz
gpt4free-d733930a2b1876340039d90f19ece81fab0d078d.tar.bz2
gpt4free-d733930a2b1876340039d90f19ece81fab0d078d.tar.lz
gpt4free-d733930a2b1876340039d90f19ece81fab0d078d.tar.xz
gpt4free-d733930a2b1876340039d90f19ece81fab0d078d.tar.zst
gpt4free-d733930a2b1876340039d90f19ece81fab0d078d.zip
Diffstat (limited to '')
-rw-r--r--g4f/stubs.py10
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