open-source-mirrors
/
GPT3Discord
mirror of https://github.com/Kav-K/GPT3Discord
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

make /index responses look nicer

Browse Source
Kaveen Kumarasinghe 2 years ago
parent 22e9010481
commit 3c54cb97b6
4 changed files with 169 additions and 38 deletions
Show Stats Download Patch File Download Diff File

12
cogs/commands.py
Unescape View File

@ -533,13 +533,13 @@ class Commands(discord.Cog, name="Commands"):
@discord.option(
name="user_index",
description="Which user index to rename",
required=False,
required=True,
autocomplete=File_autocompleter.get_user_indexes,
)
@discord.option(
name="new_name",
description="The new name",
required=False,
required=True,
type=discord.SlashCommandOptionType.string,
)
async def rename_user_index(
@ -561,13 +561,13 @@ class Commands(discord.Cog, name="Commands"):
@discord.option(
name="server_index",
description="Which server index to rename",
required=False,
required=True,
autocomplete=File_autocompleter.get_server_indexes,
)
@discord.option(
name="new_name",
description="The new name",
required=False,
required=True,
type=discord.SlashCommandOptionType.string,
)
async def rename_server_index(
@ -589,13 +589,13 @@ class Commands(discord.Cog, name="Commands"):
@discord.option(
name="search_index",
description="Which search index to rename",
required=False,
required=True,
autocomplete=File_autocompleter.get_user_search_indexes,
)
@discord.option(
name="new_name",
description="The new name",
required=False,
required=True,
type=discord.SlashCommandOptionType.string,
)
async def rename_search_index(

19
cogs/index_service_cog.py
Unescape View File

@ -3,6 +3,7 @@ from pathlib import Path
import discord
from models.embed_statics_model import EmbedStatics
from services.environment_service import EnvService
from services.moderations_service import Moderation
from services.text_service import TextService
@ -29,7 +30,7 @@ class IndexService(discord.Cog, name="IndexService"):
"""Command handler to rename a user index"""
if not new_name:
await ctx.respond("Please provide a new name for this index")
await ctx.respond(await EmbedStatics.get_index_rename_failure_embed(user_index.split("/")[-1], "None", "Please provide a new name for this index"))
return
if await self.index_handler.rename_index(
@ -37,15 +38,15 @@ class IndexService(discord.Cog, name="IndexService"):
f"indexes/{ctx.user.id}/{user_index}",
f"indexes/{ctx.user.id}/{new_name}",
):
await ctx.respond(f"Your index has been renamed to `{new_name}`")
await ctx.respond(embed=EmbedStatics.get_index_rename_success_embed(user_index.split("/")[-1], new_name))
else:
await ctx.respond("Something went wrong while renaming your index")
await ctx.respond(embed=EmbedStatics.get_index_rename_failure_embed(user_index.split("/")[-1], new_name, "Please check the server console for more details."))
async def rename_server_index_command(self, ctx, server_index, new_name):
"""Command handler to rename a user index"""
if not new_name:
await ctx.respond("Please provide a new name for this index")
await ctx.respond(await EmbedStatics.get_index_rename_failure_embed(server_index.split("/")[-1], "None", "Please provide a new name for this index"))
return
if await self.index_handler.rename_index(
@ -53,13 +54,13 @@ class IndexService(discord.Cog, name="IndexService"):
f"indexes/{ctx.guild.id}/{server_index}",
f"indexes/{ctx.guild.id}/{new_name}",
):
await ctx.respond(f"Your index has been renamed to `{new_name}`")
await ctx.respond(embed=EmbedStatics.get_index_rename_success_embed(server_index.split("/")[-1], new_name))
else:
await ctx.respond("Something went wrong while renaming your index")
await ctx.respond(embed=EmbedStatics.get_index_rename_failure_embed(server_index.split("/")[-1], new_name, "Please check the server console for more details."))
async def rename_search_index_command(self, ctx, search_index, new_name):
if not new_name:
await ctx.respond("Please provide a new name for this index")
await ctx.respond(await EmbedStatics.get_index_rename_failure_embed(search_index.split("/")[-1], "None", "Please provide a new name for this index"))
return
if await self.index_handler.rename_index(
@ -67,9 +68,9 @@ class IndexService(discord.Cog, name="IndexService"):
f"indexes/{ctx.user.id}_search/{search_index}",
f"indexes/{ctx.user.id}_search/{new_name}",
):
await ctx.respond(f"Your index has been renamed to `{new_name}`")
await ctx.respond(embed=EmbedStatics.get_index_rename_success_embed(search_index.split("/")[-1], new_name))
else:
await ctx.respond("Something went wrong while renaming your index")
await ctx.respond(embed=EmbedStatics.get_index_rename_failure_embed(search_index.split("/")[-1], new_name, "Please check the server console for more details."))
async def set_index_command(
self, ctx, file: discord.Attachment = None, link: str = None

127
models/embed_statics_model.py
Unescape View File

@ -82,3 +82,130 @@ class EmbedStatics:
color=0x808080,
)
return embed
@staticmethod
def get_index_set_success_embed():
embed = discord.Embed(
title="Index Added",
color=discord.Color.green(),
)
# thumbnail of https://i.imgur.com/7JF0oGD.png
embed.set_thumbnail(
url="https://i.imgur.com/7JF0oGD.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/VLJ32x7.png
embed.set_thumbnail(
url="https://i.imgur.com/VLJ32x7.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/7JF0oGD.png
embed.set_thumbnail(
url="https://i.imgur.com/7JF0oGD.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/VLJ32x7.png
embed.set_thumbnail(
url="https://i.imgur.com/VLJ32x7.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/VLJ32x7.png
embed.set_thumbnail(
url="https://i.imgur.com/VLJ32x7.png"
)
return embed
@staticmethod
def get_index_compose_success_embed():
embed = discord.Embed(
title="Indexes Composed",
color=discord.Color.green(),
)
# thumbnail of https://i.imgur.com/7JF0oGD.png
embed.set_thumbnail(
url="https://i.imgur.com/7JF0oGD.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/VLJ32x7.png
embed.set_thumbnail(
url="https://i.imgur.com/VLJ32x7.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/VLJ32x7.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/7JF0oGD.png
embed.set_thumbnail(
url="https://i.imgur.com/7JF0oGD.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/VLJ32x7.png
embed.set_thumbnail(
url="https://i.imgur.com/VLJ32x7.png"
)
return embed

49
models/index_model.py
Unescape View File

@ -44,6 +44,7 @@ from gpt_index.readers.web import DEFAULT_WEBSITE_EXTRACTOR
from gpt_index.composability import ComposableGraph
from models.embed_statics_model import EmbedStatics
from services.environment_service import EnvService, app_root_path
SHORT_TO_LONG_CACHE = {}
@ -168,6 +169,7 @@ class Index_handler:
Path(original_path).rename(rename_path)
return True
except Exception as e:
traceback.print_exc()
return False
async def paginate_embed(self, response_text):
@ -357,7 +359,7 @@ class Index_handler:
pass # No suffix change
else:
await ctx.respond(
"Only accepts text, pdf, images, spreadheets, powerpoint, and audio/video files."
embed=EmbedStatics.get_index_set_failure_embed("Only accepts text, pdf, images, spreadheets, powerpoint, and audio/video files.")
)
return
async with aiofiles.tempfile.TemporaryDirectory() as temp_path:
@ -375,9 +377,9 @@ class Index_handler:
file_name = file.filename
self.index_storage[ctx.user.id].add_index(index, ctx.user.id, file_name)
await ctx.respond("Index added to your indexes.")
except Exception:
await ctx.respond("Failed to set index")
await ctx.respond(embed=EmbedStatics.get_index_set_success_embed())
except Exception as e:
await ctx.respond(embed=EmbedStatics.get_index_set_failure_embed(str(e)))
traceback.print_exc()
async def set_link_index(
@ -399,11 +401,12 @@ class Index_handler:
if response.status == 200:
content_type = response.headers.get("content-type")
else:
await ctx.respond("Failed to get link")
await ctx.respond(embed=EmbedStatics.get_index_set_failure_embed("Invalid URL or could not connect to the provided URL."))
return
except Exception:
except Exception as e:
traceback.print_exc()
await ctx.respond("Failed to get link")
await ctx.respond(embed=EmbedStatics.get_index_set_failure_embed(
"Invalid URL or could not connect to the provided URL. "+str(e)))
return
# Check if the link contains youtube in it
@ -437,16 +440,16 @@ class Index_handler:
self.index_storage[ctx.user.id].add_index(index, ctx.user.id, file_name)
except ValueError as e:
await ctx.respond(str(e))
await ctx.respond(embed=EmbedStatics.get_index_set_failure_embed(str(e)))
traceback.print_exc()
return
except Exception:
await ctx.respond("Failed to set index")
except Exception as e:
await ctx.respond(embed=EmbedStatics.get_index_set_failure_embed(str(e)))
traceback.print_exc()
return
await ctx.respond("Index set")
await ctx.respond(embed=EmbedStatics.get_index_set_success_embed())
async def set_discord_index(
self,
@ -472,9 +475,9 @@ class Index_handler:
embedding_model.last_token_usage, embeddings=True
)
self.index_storage[ctx.user.id].add_index(index, ctx.user.id, channel.name)
await ctx.respond("Index set")
except Exception:
await ctx.respond("Failed to set index")
await ctx.respond(embed=EmbedStatics.get_index_set_success_embed())
except Exception as e:
await ctx.respond(embed=EmbedStatics.get_index_set_failure_embed(str(e)))
traceback.print_exc()
async def load_index(
@ -502,10 +505,10 @@ class Index_handler:
None, partial(self.index_load_file, index_file)
)
self.index_storage[ctx.user.id].queryable_index = index
await ctx.respond("Loaded index")
await ctx.respond(embed=EmbedStatics.get_index_load_success_embed())
except Exception as e:
traceback.print_exc()
await ctx.respond(e)
await ctx.respond(embed=EmbedStatics.get_index_load_failure_embed(str(e)))
async def index_to_docs(
self, old_index, chunk_size: int = 4000, chunk_overlap: int = 200
@ -664,9 +667,9 @@ class Index_handler:
/ f"{ctx.guild.name.replace(' ', '-')}_{date.today().month}_{date.today().day}.json"
)
await ctx.respond("Backup saved")
except Exception:
await ctx.respond("Failed to save backup")
await ctx.respond(embed=EmbedStatics.get_index_set_success_embed())
except Exception as e:
await ctx.respond(embed=EmbedStatics.get_index_set_failure_embed((str(e))))
traceback.print_exc()
async def query(
@ -720,7 +723,7 @@ class Index_handler:
except Exception:
traceback.print_exc()
await ctx.respond(
"Failed to send query. You may not have an index set, load an index with /index load",
embed=EmbedStatics.get_index_query_failure_embed("Failed to send query. You may not have an index set, load an index with /index load"),
delete_after=10,
)
@ -809,7 +812,7 @@ class Index_handler:
os.environ["OPENAI_API_KEY"] = user_api_key
if not self.index_storage[ctx.user.id].has_indexes(ctx.user.id):
await ctx.respond("You must load at least one indexes before composing")
await ctx.respond(embed=EmbedStatics.get_index_compose_failure_embed("You must have at least one index to compose."))
return
await ctx.respond(
@ -947,14 +950,14 @@ class ComposeModal(discord.ui.View):
return False
except Exception as e:
await interaction.followup.send(
"An error occurred while composing the indexes: " + str(e),
embed=EmbedStatics.get_index_compose_failure_embed("An error occurred while composing the indexes: " + str(e)),
ephemeral=True,
delete_after=180,
)
return False
await interaction.followup.send(
"Composed indexes", ephemeral=True, delete_after=180
embed=EmbedStatics.get_index_compose_success_embed(), ephemeral=True, delete_after=180
)
# Try to direct message the user that their composed index is ready

Write Preview
Loading…
Cancel
Save