Small changes

g4f/Provider/PollinationsAI.py

@@ -16,7 +16,7 @@ from ..requests.raise_for_status import raise_for_status
 from ..requests.aiohttp import get_connector
 from ..image.copy_images import save_response_media
 from ..image import use_aspect_ratio
-from ..providers.response import FinishReason, Usage, ToolCalls
+from ..providers.response import FinishReason, Usage, ToolCalls, ImageResponse
 from .. import debug
 
 DEFAULT_HEADERS = {
@@ -239,14 +239,11 @@ class PollinationsAI(AsyncGeneratorProvider, ProviderModelMixin):
         }, aspect_ratio)
         query = "&".join(f"{k}={quote_plus(str(v))}" for k, v in params.items() if v is not None)
         url = f"{cls.image_api_endpoint}prompt/{quote_plus(prompt)}?{query}"
-        #yield ImagePreview(url, prompt)
 
         async with ClientSession(headers=DEFAULT_HEADERS, connector=get_connector(proxy=proxy)) as session:
             async with session.get(url, allow_redirects=True) as response:
                 await raise_for_status(response)
-                async for chunk in save_response_media(response, prompt):
-                    yield chunk
-                    return
+                yield ImageResponse(url, prompt)
 
     @classmethod
     async def _generate_text(

g4f/Provider/needs_auth/OpenaiChat.py

@@ -400,7 +400,6 @@ class OpenaiChat(AsyncAuthedProvider, ProviderModelMixin):
                     "client_contextual_info":{"is_dark_mode":False,"time_since_loaded":random.randint(20, 500),"page_height":578,"page_width":1850,"pixel_ratio":1,"screen_height":1080,"screen_width":1920},
                     "paragen_cot_summary_display_override":"allow"
                 }
-                print(data)
                 if conversation.conversation_id is not None:
                     data["conversation_id"] = conversation.conversation_id
                     debug.log(f"OpenaiChat: Use conversation: {conversation.conversation_id}")

g4f/api/stubs.py

@@ -51,6 +51,7 @@ class ChatCompletionsConfig(BaseModel):
     modalities: Optional[list[str]] = None
     audio: Optional[dict] = None
     response_format: Optional[dict] = None
+    extra_data: Optional[dict] = None
 
 class ImageGenerationConfig(BaseModel):
     prompt: str