Merge pull request #135 from GreyDGL/deepsource-transform-0589ab0c

format code with black

Author: GreyDGL

pentestgpt/utils/APIs/chatgpt_api.py

@@ -42,10 +42,7 @@ def __eq__(self, other):
 
 
 class ChatGPTAPI(LLMAPI):
-    def __init__(
-        self,
-        config_class
-    ):
+    def __init__(self, config_class):
         openai.api_key = os.getenv("OPENAI_KEY", None)
         openai.api_base = config_class.api_base
         self.model = config_class.model

pentestgpt/utils/APIs/module_import.py

@@ -34,7 +34,6 @@ class GPT4ConfigClass:
     is_debugging: bool = False
 
 
-
 @dataclasses.dataclass
 class GPT35Turbo16kConfigClass:
     model: str = "gpt-3.5-turbo-16k"
@@ -47,10 +46,9 @@ class GPT35Turbo16kConfigClass:
     is_debugging: bool = False
 
 
-
 @dataclasses.dataclass
 class GPT4ALLConfigClass:
-    model:str = "orca-mini-3b.ggmlv3.q4_0.bin"
+    model: str = "orca-mini-3b.ggmlv3.q4_0.bin"
 
 
 def dynamic_import(module_name, log_dir) -> object: