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

better timeout handling #14

Merged
wirehack7 merged 3 commits from dev into main 2025-05-01 19:29:55 +02:00
Conversation 1 Commits 3 Files Changed 1 +16 -12
1 changed files with 16 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 90c97d3f54 - Show all commits

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)