Apply PEP8 styling standard & reformat

bot/openai_helper.py

@@ -26,9 +26,11 @@ def default_max_tokens(model: str) -> int:
     """
     return 1200 if model in GPT_3_MODELS else 2400
 
+
 with open('translations.json', 'r', encoding='utf-8') as f:
     translations = json.load(f)
 
+
 def localized_text(key, bot_language):
     """
     Return translated text for a key in specified bot_language.
@@ -46,6 +48,7 @@ def localized_text(key, bot_language):
             # return key as text
             return key
 
+
 class OpenAIHelper:
     """
     ChatGPT helper class.
@@ -172,7 +175,7 @@ class OpenAIHelper:
                 frequency_penalty=self.config['frequency_penalty'],
                 stream=stream
             )
-        
+
         except openai.error.RateLimitError as e:
             raise Exception(f"⚠️ _{localized_text('openai_rate_limit', bot_language)}._ ⚠️\n{str(e)}") from e
 
@@ -198,7 +201,10 @@ class OpenAIHelper:
 
             if 'data' not in response or len(response['data']) == 0:
                 logging.error(f'No response from GPT: {str(response)}')
-                raise Exception(f"⚠️ _{localized_text('error', bot_language)}._ ⚠️\n{localized_text('try_again', bot_language)}.")
+                raise Exception(
+                    f"⚠️ _{localized_text('error', bot_language)}._ "
+                    f"⚠️\n{localized_text('try_again', bot_language)}."
+                )
 
             return response['data'][0]['url'], self.config['image_size']
         except Exception as e:
@@ -253,8 +259,8 @@ class OpenAIHelper:
         :return: The summary
         """
         messages = [
-            { "role": "assistant", "content": "Summarize this conversation in 700 characters or less" },
-            { "role": "user", "content": str(conversation) }
+            {"role": "assistant", "content": "Summarize this conversation in 700 characters or less"},
+            {"role": "user", "content": str(conversation)}
         ]
         response = await openai.ChatCompletion.acreate(
             model=self.config['model'],
@@ -304,7 +310,7 @@ class OpenAIHelper:
                     num_tokens += tokens_per_name
         num_tokens += 3  # every reply is primed with <|start|>assistant<|message|>
         return num_tokens
-    
+
     def get_billing_current_month(self):
         """Gets billed usage for current month from OpenAI API.
 
@@ -324,5 +330,5 @@ class OpenAIHelper:
         }
         response = requests.get("https://api.openai.com/dashboard/billing/usage", headers=headers, params=params)
         billing_data = json.loads(response.text)
-        usage_month = billing_data["total_usage"] / 100 # convert cent amount to dollars
-        return usage_month
+        usage_month = billing_data["total_usage"] / 100  # convert cent amount to dollars
+        return usage_month