diff --git a/cogs/gpt_3_commands_and_converser.py b/cogs/gpt_3_commands_and_converser.py index 05b8744..40afeb9 100644 --- a/cogs/gpt_3_commands_and_converser.py +++ b/cogs/gpt_3_commands_and_converser.py @@ -98,6 +98,7 @@ class GPT3ComCon(commands.Cog, name="GPT3ComCon"): async def on_member_join(self, member): if self.model.welcome_message_enabled: query = f"Please generate a welcome message for {member.name} who has just joined the server." + try: welcome_message = self.model.send_request(query, tokens=self.usage_service.count_tokens(query)) except: diff --git a/models/env_service_model.py b/models/env_service_model.py index c796707..cf94c9e 100644 --- a/models/env_service_model.py +++ b/models/env_service_model.py @@ -53,12 +53,12 @@ class EnvService: allowed_roles.split(",") if "," in allowed_roles else [allowed_roles] ) return allowed_roles - + @staticmethod def get_welcome_message(): # WELCOME_MESSAGE is a default string used to welcome new members to the server if GPT3 is not available. - #The string can be blank but this is not advised. If a string cannot be found in the .env file, the below string is used. - #The string is DMd to the new server member as part of an embed. + # The string can be blank but this is not advised. If a string cannot be found in the .env file, the below string is used. + # The string is DMd to the new server member as part of an embed. try: welcome_message = os.getenv("WELCOME_MESSAGE") except: diff --git a/models/openai_model.py b/models/openai_model.py index 1a012b3..5ac1ac6 100644 --- a/models/openai_model.py +++ b/models/openai_model.py @@ -78,7 +78,7 @@ class Model: self.openai_key = os.getenv("OPENAI_TOKEN") # Use the @property and @setter decorators for all the self fields to provide value checking - + @property def welcome_message_enabled(self): return self._welcome_message_enabled @@ -91,8 +91,7 @@ class Model: self._welcome_message_enabled = False else: raise ValueError("Value must be either true or false!") - - + @property def summarize_threshold(self): return self._summarize_threshold