Format Python code with psf/black push

cogs/code_interpreter_service_cog.py

@@ -219,7 +219,9 @@ async def on_message(self, message):
             agent = self.chat_agents[message.channel.id]
             try:
                 # Start listening to STDOUT before this call. We wanna track all the output for this specific call below
-                self.usage_service.update_usage_memory(message.guild.name, "code_interpreter_message", 1)
+                self.usage_service.update_usage_memory(
+                    message.guild.name, "code_interpreter_message", 1
+                )
                 response, stdout_output = await capture_stdout(
                     self.bot.loop.run_in_executor, None, agent.run, prompt
                 )

cogs/commands.py

@@ -1028,7 +1028,9 @@ async def help(self, ctx: discord.ApplicationContext):
 
     @add_to_group("system")
     @discord.slash_command(
-        name="usage_metrics", description="Usage count by functionality", guild_ids=ALLOWED_GUILDS
+        name="usage_metrics",
+        description="Usage count by functionality",
+        guild_ids=ALLOWED_GUILDS,
     )
     @discord.guild_only()
     async def usage_metrics(self, ctx: discord.ApplicationContext):
@@ -1053,7 +1055,6 @@ async def usage_metrics(self, ctx: discord.ApplicationContext):
             for embed in embed_list:
                 await ctx.channel.send(embed=embed)
 
-
     @discord.slash_command(
         name="setup",
         description="Setup your API key for use with GPT3Discord",

cogs/image_service_cog.py

@@ -82,7 +82,6 @@ async def draw_command(
             )
             self.usage_service.update_usage_memory(ctx.guild.name, "image_drawn", 1)
 
-
         except Exception as e:
             print(e)
             traceback.print_exc()

cogs/search_service_cog.py

@@ -362,7 +362,9 @@ async def on_message(self, message):
             used_tools = []
             try:
                 # Start listening to STDOUT before this call. We wanna track all the output for this specific call below
-                self.usage_service.update_usage_memory(message.guild.name, "internet_chat_message", 1)
+                self.usage_service.update_usage_memory(
+                    message.guild.name, "internet_chat_message", 1
+                )
                 response, stdout_output = await capture_stdout(
                     self.bot.loop.run_in_executor, None, agent.run, prompt
                 )

cogs/text_service_cog.py

@@ -705,7 +705,9 @@ async def on_message(self, message):
             files=None if not file else message.attachments,
         ):
             print("Processing a conversation message in server", message.guild.name)
-            self.usage_service.update_usage_memory(message.guild.name, "conversation_message", 1)
+            self.usage_service.update_usage_memory(
+                message.guild.name, "conversation_message", 1
+            )
             original_message[message.author.id] = message.id
 
         # If the user tagged the bot and the tag wasn't an @here or @everyone, retrieve the message

models/index_model.py

@@ -429,7 +429,9 @@ async def index_chat_file(self, message: discord.Message, file: discord.Attachme
                         ),
                     )
                     print("Done Indexing")
-                    self.usage_service.update_usage_memory(message.guild.name, "index_chat_file", 1)
+                    self.usage_service.update_usage_memory(
+                        message.guild.name, "index_chat_file", 1
+                    )
 
                     summary = await index.as_query_engine(
                         response_mode="tree_summarize",
@@ -976,7 +978,9 @@ async def index_link(self, link, summarize=False, index_chat_ctx=None):
         if index_chat_ctx:
             try:
                 print("Getting transcript summary")
-                self.usage_service.update_usage_memory(index_chat_ctx.guild.name, "index_chat_link", 1)
+                self.usage_service.update_usage_memory(
+                    index_chat_ctx.guild.name, "index_chat_link", 1
+                )
                 summary = await index.as_query_engine(
                     response_mode="tree_summarize",
                     service_context=get_service_context_with_llm(

services/usage_service.py

@@ -26,7 +26,6 @@ def __init__(self, data_dir: Path):
         "curie": 0.002,
         "embedding": 0.0001,
         "gpt-turbo": 0.03,
-
     }
 
     MODEL_COST_MAP = {
@@ -133,7 +132,6 @@ def get_usage_memory(self, guild_name):
     def get_usage_memory_all(self):
         return self.usage
 
-
     @staticmethod
     def count_tokens_static(text):
         tokenizer = tiktoken.get_encoding("cl100k_base")