Skip to content

Commit aca2beb

Browse files
committed
Fix tests
1 parent edcf55a commit aca2beb

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

packages/jupyter-ai/jupyter_ai/tests/__snapshots__/test_config_manager.ambr

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,10 @@
33
dict({
44
'api_keys': list([
55
]),
6+
'completions_embeddings_provider_id': None,
7+
'completions_fields': dict({
8+
}),
9+
'completions_model_provider_id': None,
610
'embeddings_provider_id': None,
711
'fields': dict({
812
}),

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

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,8 @@ def __init__(self):
2828
self.messages = []
2929
self.tasks = []
3030
self.settings["jai_config_manager"] = SimpleNamespace(
31-
lm_provider=MockProvider, lm_provider_params={"model_id": "model"}
31+
completions_lm_provider=MockProvider,
32+
completions_lm_provider_params={"model_id": "model"},
3233
)
3334
self.settings["jai_event_loop"] = SimpleNamespace(
3435
create_task=lambda x: self.tasks.append(x)

0 commit comments

Comments
 (0)