From caa6fa847735e3db7c270d1d2d2d1c7ae51c2d8c Mon Sep 17 00:00:00 2001
From: krassowski <5832902+krassowski@users.noreply.github.com>
Date: Wed, 3 Apr 2024 12:15:12 +0100
Subject: [PATCH] Fix tests

---
 .../jupyter_ai/tests/__snapshots__/test_config_manager.ambr   | 4 ++++
 .../jupyter-ai/jupyter_ai/tests/completions/test_handlers.py  | 3 ++-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/packages/jupyter-ai/jupyter_ai/tests/__snapshots__/test_config_manager.ambr b/packages/jupyter-ai/jupyter_ai/tests/__snapshots__/test_config_manager.ambr
index cd254bed..b5d75bdc 100644
--- a/packages/jupyter-ai/jupyter_ai/tests/__snapshots__/test_config_manager.ambr
+++ b/packages/jupyter-ai/jupyter_ai/tests/__snapshots__/test_config_manager.ambr
@@ -3,6 +3,10 @@
   dict({
     'api_keys': list([
     ]),
+    'completions_embeddings_provider_id': None,
+    'completions_fields': dict({
+    }),
+    'completions_model_provider_id': None,
     'embeddings_provider_id': None,
     'fields': dict({
     }),
diff --git a/packages/jupyter-ai/jupyter_ai/tests/completions/test_handlers.py b/packages/jupyter-ai/jupyter_ai/tests/completions/test_handlers.py
index 1b950af7..0028356a 100644
--- a/packages/jupyter-ai/jupyter_ai/tests/completions/test_handlers.py
+++ b/packages/jupyter-ai/jupyter_ai/tests/completions/test_handlers.py
@@ -28,7 +28,8 @@ def __init__(self):
         self.messages = []
         self.tasks = []
         self.settings["jai_config_manager"] = SimpleNamespace(
-            lm_provider=MockProvider, lm_provider_params={"model_id": "model"}
+            completions_lm_provider=MockProvider,
+            completions_lm_provider_params={"model_id": "model"},
         )
         self.settings["jai_event_loop"] = SimpleNamespace(
             create_task=lambda x: self.tasks.append(x)