From 0b2d4c95b711704dbd3af3e56bb7db36275968eb Mon Sep 17 00:00:00 2001 From: Kaveen Kumarasinghe Date: Mon, 9 Jan 2023 23:50:40 -0500 Subject: [PATCH] fix bugs with conversations, some slight improvements --- cogs/gpt_3_commands_and_converser.py | 8 ++------ gpt3discord.py | 6 +++++- models/autocomplete_model.py | 2 +- 3 files changed, 8 insertions(+), 8 deletions(-) diff --git a/cogs/gpt_3_commands_and_converser.py b/cogs/gpt_3_commands_and_converser.py index 1245743..af30705 100644 --- a/cogs/gpt_3_commands_and_converser.py +++ b/cogs/gpt_3_commands_and_converser.py @@ -729,11 +729,7 @@ class GPT3ComCon(discord.Cog, name="GPT3ComCon"): new_prompt = prompt.encode("ascii", "ignore").decode() prompt_less_author = f"{new_prompt} <|endofstatement|>\n" - user_displayname = ( - ctx.user.name - if isinstance(ctx, discord.ApplicationContext) - else ctx.author.display_name - ) + user_displayname = ctx.author.display_name new_prompt = ( f"\n'{user_displayname}': {new_prompt} <|endofstatement|>\n" @@ -1195,7 +1191,7 @@ class GPT3ComCon(discord.Cog, name="GPT3ComCon"): if not self.pinecone_service: self.conversation_threads[thread.id].history.append( - f"\n'{ctx.user.name}': {opener} <|endofstatement|>\n" + f"\n'{ctx.author.display_name}': {opener} <|endofstatement|>\n" ) self.conversation_threads[thread.id].count += 1 diff --git a/gpt3discord.py b/gpt3discord.py index 35271f9..a591366 100644 --- a/gpt3discord.py +++ b/gpt3discord.py @@ -37,7 +37,11 @@ except: pinecone_service = None if PINECONE_TOKEN: pinecone.init(api_key=PINECONE_TOKEN, environment="us-west1-gcp") - PINECONE_INDEX = "conversation-embeddings" # This will become unfixed later. + PINECONE_INDEX = "conversation-embeddings" + if PINECONE_INDEX not in pinecone.list_indexes(): + print("Creating pinecone index. Please wait...") + pinecone.create_index('conversation-embeddings', dimension=1536, metric='dotproduct', pod_type='s1') + pinecone_service = PineconeService(pinecone.Index(PINECONE_INDEX)) print("Got the pinecone service") diff --git a/models/autocomplete_model.py b/models/autocomplete_model.py index 3fa9b6e..f007862 100644 --- a/models/autocomplete_model.py +++ b/models/autocomplete_model.py @@ -41,7 +41,7 @@ class Settings_autocompleter: "summarize_threshold": [str(num) for num in range(800, 3500, 50)], } if ctx.options["parameter"] in values.keys(): - return [value for value in values[ctx.options["parameter"]]] + return [value for value in values[ctx.options["parameter"]] if value.startswith(ctx.value.lower())] else: await ctx.interaction.response.defer() # defer so the autocomplete in int values doesn't error but rather just says not found return []