Skip to content

Commit 6d86dd0

Browse files
pre-commit-ci[bot]dlqqq
authored andcommitted
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 6be48f8 commit 6d86dd0

File tree

2 files changed

+7
-4
lines changed

2 files changed

+7
-4
lines changed

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

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,11 +28,13 @@ class InlineCompletionRequest(BaseModel):
2828
# previous cells and following cells can be used to learn the wider context
2929
cell_id: Optional[str] = None
3030

31+
3132
class InlineCompletionItem(BaseModel):
3233
"""The inline completion suggestion to be displayed on the frontend.
3334
3435
See JupyterLab `InlineCompletionItem` documentation for the details.
3536
"""
37+
3638
insertText: str
3739
filterText: Optional[str] = None
3840
isIncomplete: Optional[bool] = None

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

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ def __init__(self, **kwargs):
2727
if "responses" not in kwargs:
2828
kwargs["responses"] = ["Test response"]
2929
super().__init__(**kwargs)
30-
30+
3131
def _acall(self, *args, **kwargs):
3232
if self.raise_exc:
3333
raise Exception("Test exception")
@@ -57,6 +57,7 @@ def reply(
5757
) -> None:
5858
self.messages.append(message)
5959

60+
6061
@fixture
6162
def inline_handler() -> MockCompletionHandler:
6263
return MockCompletionHandler()
@@ -201,13 +202,13 @@ async def test_handle_request_with_error(inline_handler):
201202
lm_provider_params={
202203
"model_id": "model",
203204
"responses": ["test"],
204-
"raise_exc": True
205-
}
205+
"raise_exc": True,
206+
},
206207
)
207208
dummy_request = InlineCompletionRequest(
208209
number=1, prefix="", suffix="", mime="", stream=True
209210
)
210211
await inline_handler.on_message(json.dumps(dict(dummy_request)))
211212
await inline_handler.tasks[0]
212-
error = inline_handler.messages[-1].dict().get('error', None)
213+
error = inline_handler.messages[-1].dict().get("error", None)
213214
assert error is not None

0 commit comments

Comments
 (0)