ruff

jamesturk · Nov 25, 2023 · 7442085 · 7442085
1 parent c36ae0e
commit 7442085
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/scrapeghost/apicall.py b/src/scrapeghost/apicall.py
@@ -80,7 +80,10 @@ def __init__(
             self.postprocessors = postprocessors
 
     def _raw_api_request(
-        self, model: str, messages: list[dict[str, str]], response: Response,
+        self,
+        model: str,
+        messages: list[dict[str, str]],
+        response: Response,
     ) -> Response:
         """
         Make an OpenAPI request and return the raw response.
@@ -101,7 +104,10 @@ def _raw_api_request(
         )
         start_t = time.time()
         completion = client.chat.completions.create(
-            model=model, messages=messages, **self.model_params, **json_mode, # type: ignore
+            model=model,
+            messages=messages,
+            **self.model_params,
+            **json_mode,  # type: ignore
         )
         elapsed = time.time() - start_t
         if completion.usage:
@@ -136,7 +142,7 @@ def _raw_api_request(
                 f"(prompt_tokens={p_tokens}, "
                 f"completion_tokens={c_tokens})"
             )
-        response.data = choice.message.content # type: ignore
+        response.data = choice.message.content  # type: ignore
         return response
 
     def _api_request(self, html: str) -> Response: