mirrors/gpt4free
1
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2025-12-15 14:51:19 -08:00
Code Issues Projects Releases Packages Wiki Activity

Small changes

Browse source
This commit is contained in:
hlohaus 2025-03-24 11:35:41 +01:00
parent ff365ff55c
commit 5ae71adbae
3 changed files with 3 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

7
g4f/Provider/PollinationsAI.py
View file

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

1
g4f/Provider/needs_auth/OpenaiChat.py
View file

@ -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}, "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" "paragen_cot_summary_display_override":"allow"
} }
print(data)
if conversation.conversation_id is not None: if conversation.conversation_id is not None:
data["conversation_id"] = conversation.conversation_id data["conversation_id"] = conversation.conversation_id
debug.log(f"OpenaiChat: Use conversation: {conversation.conversation_id}") debug.log(f"OpenaiChat: Use conversation: {conversation.conversation_id}")

1
g4f/api/stubs.py
View file

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