summaryrefslogtreecommitdiffstats
path: root/g4f/client/stubs.py
diff options
context:
space:
mode:
Diffstat (limited to 'g4f/client/stubs.py')
-rw-r--r--g4f/client/stubs.py13
1 files changed, 12 insertions, 1 deletions
diff --git a/g4f/client/stubs.py b/g4f/client/stubs.py
index 8cf2bcba..b38c9f6c 100644
--- a/g4f/client/stubs.py
+++ b/g4f/client/stubs.py
@@ -1,6 +1,7 @@
from __future__ import annotations
from typing import Union
+from time import time
class Model():
...
@@ -108,8 +109,18 @@ class Image(Model):
return self.__dict__
class ImagesResponse(Model):
- def __init__(self, data: list[Image], created: int = 0) -> None:
+ data: list[Image]
+ model: str
+ provider: str
+ created: int
+
+ def __init__(self, data: list[Image], created: int = None, model: str = None, provider: str = None) -> None:
self.data = data
+ if created is None:
+ created = int(time())
+ self.model = model
+ if provider is not None:
+ self.provider = provider
self.created = created
def to_json(self):