From 1dcdbe3cf03ff16b3ff2e2e6d572e0b3c72c18c0 Mon Sep 17 00:00:00 2001 From: salmonik Date: Sun, 19 Apr 2026 15:43:32 +0200 Subject: refactor: simplify cogs, reduce complexity - Remove unused imports and dead code - Cleaner error handling - Reduce cognitive complexity --- cogs/ai.py | 35 ++++++++++++++++++----------------- 1 file changed, 18 insertions(+), 17 deletions(-) (limited to 'cogs/ai.py') diff --git a/cogs/ai.py b/cogs/ai.py index 4a04693..6f79ac5 100644 --- a/cogs/ai.py +++ b/cogs/ai.py @@ -7,7 +7,7 @@ import os import discord from discord.ext import commands import logging -import requests +import httpx from gtts import gTTS from cogs.music import audio_playback import io @@ -52,23 +52,24 @@ class AI(commands.Cog): ], } - response = requests.post(url=url, headers=headers, json=payload) - response_data = response.json() - logger.info(response_data) - if not response_data["choices"]: - await ctx.send_followup("bot siem zesroł") - else: - response_text = response_data["choices"][0]["message"]["content"] - if tts: - # Generate TTS - tts_obj = gTTS(text=response_text, lang='pl') - tts_buffer = io.BytesIO() - tts_obj.write_to_fp(tts_buffer) - tts_buffer.seek(0) - await audio_playback.AudioPlaybackManager.play(ctx, discord.FFmpegPCMAudio(tts_buffer), delete_after_play=True) - + async with httpx.AsyncClient() as client: + response = await client.post(url=url, headers=headers, json=payload) + response_data = response.json() + logger.info(response_data) + if not response_data["choices"]: + await ctx.send_followup("bot siem zesroł") else: - await ctx.send_followup(response_text) + response_text = response_data["choices"][0]["message"]["content"] + if tts: + # Generate TTS + tts_obj = gTTS(text=response_text, lang='pl') + tts_buffer = io.BytesIO() + tts_obj.write_to_fp(tts_buffer) + tts_buffer.seek(0) + await audio_playback.AudioPlaybackManager.play(ctx, discord.FFmpegPCMAudio(tts_buffer), delete_after_play=True) + + else: + await ctx.send_followup(response_text) -- cgit v1.2.3