[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

packages/jupyter-ai-magics/jupyter_ai_magics/models/completion.py

@@ -28,14 +28,17 @@ class InlineCompletionRequest(BaseModel):
     # previous cells and following cells can be used to learn the wider context
     cell_id: Optional[str]
 
+
 class InlineCompletionError(BaseModel):
     message: str
 
+
 class InlineCompletionItem(BaseModel):
     """The inline completion suggestion to be displayed on the frontend.
 
     See JupyterLab `InlineCompletionItem` documentation for the details.
     """
+
     error: Optional[InlineCompletionError]
     insertText: str
     filterText: Optional[str]

packages/jupyter-ai/jupyter_ai/completions/handlers/base.py

@@ -138,7 +138,9 @@ async def handle_exc(self, e: Exception, request: InlineCompletionRequest):
         )
         self.reply(
             InlineCompletionReply(
-                list=InlineCompletionList(items=[{"error":{"message":title},"insertText":""}]),
+                list=InlineCompletionList(
+                    items=[{"error": {"message": title}, "insertText": ""}]
+                ),
                 error=error,
                 reply_to=request.number,
             )

packages/jupyter-ai/jupyter_ai/tests/completions/test_handlers.py

@@ -1,16 +1,16 @@
 import json
+import traceback
 from types import SimpleNamespace
 from typing import Union
-import traceback
 
 import pytest
 from jupyter_ai.completions.handlers.default import DefaultInlineCompletionHandler
 from jupyter_ai.completions.models import (
+    CompletionError,
+    InlineCompletionList,
     InlineCompletionReply,
     InlineCompletionRequest,
     InlineCompletionStreamChunk,
-    CompletionError,
-    InlineCompletionList,
 )
 from jupyter_ai_magics import BaseProvider
 from langchain_community.llms import FakeListLLM
@@ -63,7 +63,9 @@ async def handle_exc(self, e: Exception, _request: InlineCompletionRequest):
         )
         self.reply(
             InlineCompletionReply(
-                list=InlineCompletionList(items=[{"error":{"message":title},"insertText":""}]),
+                list=InlineCompletionList(
+                    items=[{"error": {"message": title}, "insertText": ""}]
+                ),
                 error=error,
                 reply_to=_request.number,
             )
@@ -207,6 +209,7 @@ async def test_handle_stream_request():
     assert third.response.insertText == "test"
     assert third.done is True
 
+
 async def test_handle_request_with_error(inline_handler):
     inline_handler = MockCompletionHandler(
         lm_provider=MockProvider,
@@ -220,5 +223,12 @@ async def test_handle_request_with_error(inline_handler):
     )
     await inline_handler.on_message(json.dumps(dict(dummy_request)))
     await inline_handler.tasks[0]
-    error_message = inline_handler.messages[-1].dict().get('list', {}).get('items', [{}])[0].get('error', {}).get('message', None)
+    error_message = (
+        inline_handler.messages[-1]
+        .dict()
+        .get("list", {})
+        .get("items", [{}])[0]
+        .get("error", {})
+        .get("message", None)
+    )
     assert error_message is not None