/
githubmirror
/
GPTDiscord
Обзор
Документация
Войти
/
githubmirror
/
GPTDiscord
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
models/embed_statics_model.py
379 строк
13 KB
github-actions
Format Python code with psf/black push
09 ноя 2023, 04:44
09 ноя 2023, 04:44
201e9e4
Код
Авторство
О чём код?
import discord from services.environment_service import EnvService BOT_NAME = EnvService.get_custom_bot_name() class EmbedStatics: def __init__(self): pass @staticmethod def paginate_chat_embed(response_text): """Given a response text make embed pages and return a list of the pages.""" response_text = [ response_text[i : i + 3500] for i in range(0, len(response_text), 7000) ] pages = [] first = False # Send each chunk as a message for count, chunk in enumerate(response_text, start=1): if not first: page = discord.Embed( title=f"{count}" if len(response_text) > 1 else "", description=chunk, ) first = True else: page = discord.Embed( title=f"{count}", description=chunk, ) pages.append(page) return pages @staticmethod def get_api_timeout_embed(): embed = discord.Embed( title="The API timed out. Try again later.", description=f"*This is an issue with the OpenAI APIs, not with the bot instance.*", color=discord.Color.red(), ) return embed @staticmethod def get_invalid_api_response_embed(e): embed = discord.Embed( title="The API returned an invalid response", description=f"**{e.status}: {e.message}**", color=discord.Color.red(), ) return embed @staticmethod def get_invalid_value_embed(e): embed = discord.Embed( title="Invalid value", description=f"**{str(e)}**", color=discord.Color.red(), ) return embed @staticmethod def get_general_error_embed(e): embed = discord.Embed( title="An error occurred", description=f"**{str(e)}**", color=discord.Color.red(), ) return embed @staticmethod def generate_end_embed(): embed = discord.Embed( title="Conversation Ended", description=f"This conversation has ended. You can start a new one with `/gpt converse`", color=0x808080, ) embed.set_thumbnail(url="https://i.imgur.com/asA13vI.png") footer_text = "Conversation ended" embed.set_footer(text=footer_text, icon_url="https://i.imgur.com/asA13vI.png") return embed @staticmethod def generate_conversation_embed(conversation_threads, thread, opener, overrides): # Generate a nice looking embed for the above text embed = discord.Embed( title="Conversation started", description=f"Conversation started with {BOT_NAME}", color=0x808080, ) embed.add_field( name="Model", value=f"The model used is **{conversation_threads[thread.id].model}**", ) embed.add_field( name="Overrides", value=f"**temp={overrides['temperature']}**, **top_p={overrides['top_p']}**" f", **freq. penalty={overrides['frequency_penalty']}**, **pres. penalty={overrides['presence_penalty']}**\n", ) embed.add_field( name="End the conversation", value="End the conversation by saying `end`, or clicking the red 'End Conversation' button\n\n", inline=False, ) embed.add_field( name="Ignoring Messages", value="If you want GPT to ignore your messages, start your messages with `~`\n\n", inline=False, ) return embed @staticmethod def generate_opener_embed(opener): embed = discord.Embed( title="Opening Prompt", description=f"{opener}", color=0x808080, ) return embed @staticmethod def get_index_set_success_embed(price="Unknown"): embed = discord.Embed( title="Index Added", description=f"This index can now be queried and loaded with `/index query` and `/index load`\n\n||Total cost: {round(float(price), 6) if price != 'Unknown' else 'Unknown'}||", color=discord.Color.green(), ) # thumbnail of https://i.imgur.com/I5dIdg6.png embed.set_thumbnail(url="https://i.imgur.com/I5dIdg6.png") return embed @staticmethod def get_index_set_failure_embed(message): embed = discord.Embed( title="Index Add", description=f"Index add failed. {message}", color=discord.Color.red(), ) # thumbnail of https://i.imgur.com/hbdBZfG.png embed.set_thumbnail(url="https://i.imgur.com/hbdBZfG.png") return embed @staticmethod def get_index_load_success_embed(name=None): embed = discord.Embed( title="Index Loaded" if not name else f"Index {name} loaded", color=discord.Color.green(), ) # thumbnail of https://i.imgur.com/I5dIdg6.png embed.set_thumbnail(url="https://i.imgur.com/I5dIdg6.png") return embed @staticmethod def get_index_load_failure_embed(message): embed = discord.Embed( title="Index load", description=f"Index load failed. {message}", color=discord.Color.red(), ) # thumbnail of https://i.imgur.com/hbdBZfG.png embed.set_thumbnail(url="https://i.imgur.com/hbdBZfG.png") return embed @staticmethod def get_index_query_failure_embed(message): embed = discord.Embed( title="Index query", description=f"Index query failed. {message}", color=discord.Color.red(), ) # thumbnail of https://i.imgur.com/hbdBZfG.png embed.set_thumbnail(url="https://i.imgur.com/hbdBZfG.png") return embed @staticmethod def get_index_compose_success_embed(price="Unknown"): embed = discord.Embed( title="Indexes Composed", description=f"Indexes composed successfully, you can query and load this index with `/index query` and `/index load`\n\n||Total cost: {round(float(price), 6) if price != 'Unknown' else 'Unknown'}||", color=discord.Color.green(), ) # thumbnail of https://i.imgur.com/I5dIdg6.png embed.set_thumbnail(url="https://i.imgur.com/I5dIdg6.png") return embed @staticmethod def get_index_compose_failure_embed(message): embed = discord.Embed( title="Index Compose", description=f"Index compose failed. {message}", color=discord.Color.red(), ) # thumbnail of https://i.imgur.com/hbdBZfG.png embed.set_thumbnail(url="https://i.imgur.com/hbdBZfG.png") return embed @staticmethod def get_index_compose_progress_embed(): embed = discord.Embed( title="Index Compose", description=f"Your index composition is running, this may take a while.", color=discord.Color.blurple(), ) # thumbnail of https://i.imgur.com/hbdBZfG.png embed.set_thumbnail(url="https://i.imgur.com/txHhNzL.png") return embed @staticmethod def get_index_chat_preparation_message(): embed = discord.Embed( title="Index Chat", description=f"Your index chat is preparing, this might take a moment.", color=discord.Color.blurple(), ) # thumbnail of https://i.imgur.com/hbdBZfG.png embed.set_thumbnail(url="https://i.imgur.com/txHhNzL.png") return embed @staticmethod def get_index_rename_success_embed(original, renamed): embed = discord.Embed( title=f"Index Rename", description=f"Index {original} renamed to {renamed}", color=discord.Color.green(), ) # thumbnail of https://i.imgur.com/I5dIdg6.png embed.set_thumbnail(url="https://i.imgur.com/I5dIdg6.png") return embed @staticmethod def get_index_rename_failure_embed(original, renamed, message): embed = discord.Embed( title="Index Rename", description=f"Index rename from {original} to {renamed} failed. {message}", color=discord.Color.red(), ) # thumbnail of https://i.imgur.com/hbdBZfG.png embed.set_thumbnail(url="https://i.imgur.com/hbdBZfG.png") return embed @staticmethod def get_edit_command_output_embed(response_text): embed = discord.Embed( title="GPT Edits", description=f"{response_text}", color=discord.Color.light_grey(), ) return embed @staticmethod def get_search_failure_embed(message): embed = discord.Embed( title="AI-Assisted Search", description=f"An error occured while performing search: {message}", color=discord.Color.red(), ) # thumbnail of https://i.imgur.com/hbdBZfG.png embed.set_thumbnail(url="https://i.imgur.com/hbdBZfG.png") return embed @staticmethod def get_internet_chat_failure_embed(message): embed = discord.Embed( title="Internet-Connected Chat", description=f"An error occured while using internet connected chat: {message}", color=discord.Color.red(), ) # thumbnail of https://i.imgur.com/hbdBZfG.png embed.set_thumbnail(url="https://i.imgur.com/hbdBZfG.png") return embed @staticmethod def get_code_chat_failure_embed(message): embed = discord.Embed( title="Code Interpreter Chat", description=f"An error occured while using code interpreter chat: {message}", color=discord.Color.red(), ) # thumbnail of https://i.imgur.com/hbdBZfG.png embed.set_thumbnail(url="https://i.imgur.com/hbdBZfG.png") return embed @staticmethod def get_search_redo_progress_embed(): embed = discord.Embed( title="AI-Assisted Search", description=f"Your original search request is being redone. This may take a while.", color=discord.Color.blurple(), ) # thumbnail of https://i.imgur.com/hbdBZfG.png embed.set_thumbnail(url="https://i.imgur.com/txHhNzL.png") return embed @staticmethod def get_conversation_shared_embed(url): embed = discord.Embed( title="Conversation Shared", description=f"You can access your shared conversation at: {url}", color=discord.Color.blurple(), ) # thumbnail of https://i.imgur.com/hbdBZfG.png embed.set_thumbnail(url="https://i.imgur.com/8OIZc1A.png") return embed @staticmethod def get_conversation_share_failed_embed(message): embed = discord.Embed( title="Conversation Sharing", description=f"Conversation sharing failed: " + message, color=discord.Color.red(), ) # thumbnail of https://i.imgur.com/hbdBZfG.png embed.set_thumbnail(url="https://i.imgur.com/hbdBZfG.png") return embed @staticmethod def build_index_progress_embed(): embed = discord.Embed( title="Index Service", description="Indexing...", color=discord.Color.blurple(), ) embed.set_thumbnail(url="https://i.imgur.com/txHhNzL.png") return embed @staticmethod def build_index_query_progress_embed(query): embed = discord.Embed( title="Index Service", description=f"Query:\n`{query}`\nQuerying...", color=discord.Color.blurple(), ) embed.set_thumbnail(url="https://i.imgur.com/txHhNzL.png") return embed @staticmethod def build_index_query_success_embed(query, price="Unknown"): embed = discord.Embed( title="Index Service", description=f"Query:\n`{query}`\nThe index query was successful.\n\n||Total cost: {round(float(price), 6) if price != 'Unknown' else 'Unknown'}||", color=discord.Color.green(), ) # thumbnail of https://i.imgur.com/I5dIdg6.png embed.set_thumbnail(url="https://i.imgur.com/I5dIdg6.png") return embed @staticmethod def build_transcribe_progress_embed(): embed = discord.Embed( title="Transcriber", description=f"Your transcription request has been sent, this may take a while.", color=discord.Color.blurple(), ) embed.set_thumbnail(url="https://i.imgur.com/txHhNzL.png") return embed @staticmethod def build_transcribe_success_embed(transcribed_text): embed = discord.Embed( title="Transcriber", description=f"Transcribed successfully:\n`{transcribed_text}`", color=discord.Color.green(), ) # thumbnail of https://i.imgur.com/I5dIdg6.png embed.set_thumbnail(url="https://i.imgur.com/I5dIdg6.png") return embed @staticmethod def build_transcribe_failed_embed(message): embed = discord.Embed( title="Transcriber", description=f"Transcription failed: " + message, color=discord.Color.red(), ) embed.set_thumbnail(url="https://i.imgur.com/hbdBZfG.png") return embed