chore: style black

dsdanielpark · Aug 9, 2023 · cd759f5 · cd759f5
1 parent ecbbeac
commit cd759f5
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 11 deletions.
diff --git a/bardapi/core.py b/bardapi/core.py
@@ -111,7 +111,7 @@ def get_answer(self, input_text: str) -> dict:
                 api_key=self.google_translator_api_key
             )
 
-        # [Optional] Set language 
+        # [Optional] Set language
         if (
             self.language is not None
             and self.language not in ALLOWED_LANGUAGES
@@ -158,7 +158,7 @@ def get_answer(self, input_text: str) -> dict:
             }
         resp_json = json.loads(resp_dict)
 
-        # [Optional] gather image links 
+        # [Optional] gather image links
         images = set()
         try:
             if len(resp_json) >= 3:
@@ -171,7 +171,7 @@ def get_answer(self, input_text: str) -> dict:
         # Parsed Answer Object
         parsed_answer = json.loads(resp_dict)
 
-        # [Optional] translated by google translator 
+        # [Optional] translated by google translator
         # Unofficial
         if (
             self.language is not None
@@ -200,9 +200,11 @@ def get_answer(self, input_text: str) -> dict:
                 for x in parsed_answer[4]
             ]
 
-        # [Optional] get program_lang & code 
+        # [Optional] get program_lang & code
         try:
-            program_lang = parsed_answer[4][0][1][0].split("```")[1].split("\n")[0].strip()
+            program_lang = (
+                parsed_answer[4][0][1][0].split("```")[1].split("\n")[0].strip()
+            )
             code = parsed_answer[4][0][1][0].split("```")[1][len(program_lang) :]
         except Exception:
             program_lang, code = None, None
@@ -409,7 +411,7 @@ def ask_about_image(self, input_text: str, image: bytes, lang: str = None) -> di
         else:
             translator_to_eng = GoogleTranslator(source="auto", target="en")
 
-        # [Optional] Set language 
+        # [Optional] Set language
         if (
             (self.language is not None or lang is not None)
             and self.language not in ALLOWED_LANGUAGES

diff --git a/bardapi/core_async.py b/bardapi/core_async.py
@@ -113,7 +113,7 @@ async def get_answer(self, input_text: str) -> dict:
                 api_key=self.google_translator_api_key
             )
 
-        # [Optional] Set language 
+        # [Optional] Set language
         if (
             self.language is not None
             and self.language not in ALLOWED_LANGUAGES
@@ -158,7 +158,7 @@ async def get_answer(self, input_text: str) -> dict:
             return {"content": f"Response Error: {resp.content}."}
         resp_json = json.loads(resp_dict)
 
-        # [Optional] Gather image links 
+        # [Optional] Gather image links
         try:
             images = set()
             if len(resp_json) >= 3:
@@ -174,7 +174,7 @@ async def get_answer(self, input_text: str) -> dict:
             pass
         parsed_answer = json.loads(resp_dict)
 
-        # [Optional] Translated by google translator 
+        # [Optional] Translated by google translator
         ## Unofficial for testing
         if (
             self.language is not None
@@ -204,7 +204,9 @@ async def get_answer(self, input_text: str) -> dict:
 
         # Get code
         try:
-            program_lang = parsed_answer[4][0][1][0].split("```")[1].split("\n")[0].strip()
+            program_lang = (
+                parsed_answer[4][0][1][0].split("```")[1].split("\n")[0].strip()
+            )
             code = parsed_answer[4][0][1][0].split("```")[1][len(program_lang) :]
         except Exception:
             program_lang, code = None, None
@@ -571,7 +573,7 @@ async def ask_about_image(
         else:
             translator_to_eng = GoogleTranslator(source="auto", target="en")
 
-        # [Optional] Set language 
+        # [Optional] Set language
         if (
             (self.language is not None or lang is not None)
             and self.language not in ALLOWED_LANGUAGES