♻️ make it pep8 compliant

tracex/extraction/prototype/create_prompts.py

@@ -5,8 +5,6 @@
 client = OpenAI(api_key=u.oaik)
 
 
-
-
 NEW_PROMPTS_CONTEXT = """
     You are an expert prompt engineer for gpt-3.5-turbo. You are tasked with creating the best possible prompts for given tasks.
     Your prompts should consist of three parts, one for the system, one for user and one for assistant.
@@ -22,10 +20,12 @@
     {"role": "user", "content": NEW_PROMPTS_PROMPT},
     {"role": "assistant", "content": NEW_PROMPTS_ANSWER},
 ]
-new_prompts = client.chat.completions.create(model=u.MODEL,
-messages=messages,
-max_tokens=u.MAX_TOKENS,
-temperature=u.TEMPERATURE_CREATION)
+new_prompts = client.chat.completions.create(
+    model=u.MODEL,
+    messages=messages,
+    max_tokens=u.MAX_TOKENS,
+    temperature=u.TEMPERATURE_CREATION,
+)
 output = new_prompts.choices[0].message.content
 with open(
     (u.output_path / "new_prompts.txt"),

tracex/extraction/prototype/input_inquiry.py

@@ -5,7 +5,6 @@
 from . import prompts as p
 
 
-
 def greeting():
     """Prints a greeting message."""
     print(

tracex/extraction/prototype/utils.py

@@ -52,7 +52,10 @@ def get_decision(question):
 #     output = response.choices[0].message.content
 #     return output
 
-def query_gpt(messages, tools=fc.TOOLS, tool_choice="none", temperature=TEMPERATURE_SUMMARIZING):
+
+def query_gpt(
+    messages, tools=fc.TOOLS, tool_choice="none", temperature=TEMPERATURE_SUMMARIZING
+):
     """Queries the GPT engine."""
     response = client.chat.completions.create(
         model=MODEL,
@@ -66,5 +69,5 @@ def query_gpt(messages, tools=fc.TOOLS, tool_choice="none", temperature=TEMPERAT
         output = response.choices[0].message.content
     else:
         api_response = response.choices[0].message.tool_calls[0].function.arguments
-        output = json.loads(api_response)['output'][0]
-    return output
+        output = json.loads(api_response)["output"][0]
+    return output