返回提交历史
Modified
g4f/Provider/PollinationsAI.py
+2
-5
Modified
g4f/Provider/needs_auth/OpenaiChat.py
+0
-1
Modified
g4f/api/stubs.py
+1
-0
XFEstudio/gpt4free
Small changes
5ae71adb
代码差异
3 个文件
+3
-6
@@ -16,7 +16,7 @@ from ..requests.raise_for_status import raise_for_status
16
16
from ..requests.aiohttp import get_connector
17
17
from ..image.copy_images import save_response_media
18
18
from ..image import use_aspect_ratio
19
from ..providers.response import FinishReason, Usage, ToolCalls
19
from ..providers.response import FinishReason, Usage, ToolCalls, ImageResponse
20
20
from .. import debug
21
21
22
22
DEFAULT_HEADERS = {
@@ -239,14 +239,11 @@ class PollinationsAI(AsyncGeneratorProvider, ProviderModelMixin):
239
239
}, aspect_ratio)
240
240
query = "&".join(f"{k}={quote_plus(str(v))}" for k, v in params.items() if v is not None)
241
241
url = f"{cls.image_api_endpoint}prompt/{quote_plus(prompt)}?{query}"
242
#yield ImagePreview(url, prompt)
243
242
244
243
async with ClientSession(headers=DEFAULT_HEADERS, connector=get_connector(proxy=proxy)) as session:
245
244
async with session.get(url, allow_redirects=True) as response:
246
245
await raise_for_status(response)
247
async for chunk in save_response_media(response, prompt):
248
yield chunk
249
return
246
yield ImageResponse(url, prompt)
250
247
251
248
@classmethod
252
249
async def _generate_text(
@@ -400,7 +400,6 @@ class OpenaiChat(AsyncAuthedProvider, ProviderModelMixin):
400
400
"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},
401
401
"paragen_cot_summary_display_override":"allow"
402
402
}
403
print(data)
404
403
if conversation.conversation_id is not None:
405
404
data["conversation_id"] = conversation.conversation_id
406
405
debug.log(f"OpenaiChat: Use conversation: {conversation.conversation_id}")
@@ -51,6 +51,7 @@ class ChatCompletionsConfig(BaseModel):
51
51
modalities: Optional[list[str]] = None
52
52
audio: Optional[dict] = None
53
53
response_format: Optional[dict] = None
54
extra_data: Optional[dict] = None
54
55
55
56
class ImageGenerationConfig(BaseModel):
56
57
prompt: str