mirror of
https://github.com/xtekky/gpt4free.git
synced 2025-10-18 22:24:36 +08:00
Check for exception in generate of G4F
This commit is contained in:
@@ -83,14 +83,11 @@ class G4F(Janus_Pro_7B):
|
||||
"x-zerogpu-uuid": zerogpu_uuid,
|
||||
}
|
||||
async def generate():
|
||||
try:
|
||||
async with session.post(cls.url_flux, json=payload, proxy=proxy, headers=headers) as response:
|
||||
await raise_for_status(response)
|
||||
response_data = await response.json()
|
||||
image_url = response_data["data"][0]['url']
|
||||
return ImageResponse(images=[image_url], alt=prompt)
|
||||
except Exception as e:
|
||||
return Reasoning(status=f"Error: {e.__class__.__name__}: {e}")
|
||||
background_tasks = set()
|
||||
started = time.time()
|
||||
task = asyncio.create_task(generate())
|
||||
|
@@ -147,6 +147,8 @@ class Backend_Api(Api):
|
||||
limiter_check()
|
||||
if "images" in json_data:
|
||||
kwargs["images"] = json_data["images"]
|
||||
if "zerogpu_token" in json_data:
|
||||
kwargs["zerogpu_token"] = json_data["zerogpu_token"]
|
||||
kwargs = self._prepare_conversation_kwargs(json_data, kwargs)
|
||||
return self.app.response_class(
|
||||
self._create_response_stream(
|
||||
|
Reference in New Issue
Block a user