ruff

jamesturk · Nov 25, 2023 · 4fe7e72 · 4fe7e72
1 parent 07b72d9
commit 4fe7e72
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tests/testutils.py b/tests/testutils.py
@@ -10,15 +10,18 @@ def _mock_response(**kwargs):
             openai.types.completion.CompletionChoice(
                 index=0,
                 text="",
-                message=openai.types.chat.ChatCompletionMessage(content=kwargs.get("content", "hello world"), role="assistant"),
-                finish_reason= kwargs.get("finish_reason", "stop"),
+                message=openai.types.chat.ChatCompletionMessage(
+                    content=kwargs.get("content", "hello world"), role="assistant"
+                ),
+                finish_reason=kwargs.get("finish_reason", "stop"),
                 logprobs={},
             )
         ],
         usage={
             "prompt_tokens": kwargs.get("prompt_tokens", 1),
             "completion_tokens": kwargs.get("completion_tokens", 1),
-            "total_tokens": kwargs.get("prompt_tokens", 1) + kwargs.get("completion_tokens", 1),
+            "total_tokens": kwargs.get("prompt_tokens", 1)
+            + kwargs.get("completion_tokens", 1),
         },
         created=1629200000,
         id="cmpl-xxxxxxxxxxxxxxxxxxxx",