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

Merge pull request 'better timeout handling' (#14) from dev into main
Some checks failed
Code Quality Check / quality-check (push) Successful in 13s
Details
Safety Check / safety-check (push) Successful in 4m38s
Details
Cleanup old Docker images / cleanup (push) Failing after 4m27s
Details

Browse Source 
Reviewed-on: #14
This commit is contained in:
wirehack7 2025-05-01 19:29:54 +02:00
commit 44d0315059
1 changed files with 19 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/main.py
View File

@ -58,7 +58,7 @@ async def on_ready():
orientation="portrait or landscape (Standard: portrait)", orientation="portrait or landscape (Standard: portrait)",
seed="Optional seed (integer). If empty, a random one will be generated", seed="Optional seed (integer). If empty, a random one will be generated",
) )
# pylint: disable=too-many-locals,too-many-return-statements # pylint: disable=too-many-locals,too-many-return-statements,too-many-statements
async def generate( async def generate(
interaction: discord.Interaction, interaction: discord.Interaction,
prompt: str, prompt: str,
@ -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)
@ -214,7 +218,9 @@ async def generate(
except Exception: # pylint: disable=W0718 except Exception: # pylint: disable=W0718
error_data = "error message unreadable" error_data = "error message unreadable"
error_message = f"Error {status_code} at API-request.\n" f"Answer: {error_data}\n" error_message = (
f"Error {status_code} at API-request.\n" f"Answer: {error_data}\n"
)
await interaction.followup.send(error_message) await interaction.followup.send(error_message)