0.3.4.0

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):