wirehack7/novelai-bot
1
0
Fork 0
Code Issues 1 Pull Requests 1 Actions Packages Releases Activity

better timeout handling
Some checks failed
Safety Check / safety-check (pull_request) Successful in -1s
Details
Code Quality Check / quality-check (pull_request) Failing after 14s
Details

Browse Source
This commit is contained in:
wirehack7 2025-05-01 19:25:48 +02:00
parent 6bf61fdbf0
commit 90c97d3f54
Signed by: wirehack7
GPG Key ID: C9F4CF85BA505ADC
1 changed files with 16 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/main.py
View File

@ -159,7 +159,7 @@ async def generate(
} }
start_time = time.monotonic() start_time = time.monotonic()
try:
async with aiohttp.ClientSession() as session: async with aiohttp.ClientSession() as session:
async with session.post( async with session.post(
"https://image.novelai.net/ai/generate-image", "https://image.novelai.net/ai/generate-image",
@ -169,6 +169,10 @@ async def generate(
) as response: ) as response:
response_bytes = await response.read() response_bytes = await response.read()
status_code = response.status status_code = response.status
except aiohttp.TimeoutError:
await interaction.followup.send("Error: the request timed out.")
except aiohttp.ClientError as e:
await interaction.followup.send(f"Error: {e}")
duration = time.monotonic() - start_time duration = time.monotonic() - start_time
logger.info("Image creation took %.2f seconds", duration) logger.info("Image creation took %.2f seconds", duration)