Add a test for slow-starting completion provider

krassowski · krassowski · commit a64ad8dfd5a7 · 2025-05-30T16:07:07.000+01:00
diff --git a/tests/fake_llm.py b/tests/fake_llm.py
@@ -1,4 +1,5 @@
 import asyncio
+from time import sleep
 
 try:
     from jupyter_ai_magics import BaseProvider
@@ -87,3 +88,16 @@ async def _stream(self, sentence, request_number, token, start_with=""):
             reply_to=request_number,
             done=True,
         )
+
+
+class SlowStartingCompletionProvider(BaseProvider, FakeListLLM):  # type: ignore[misc, valid-type]
+    id = "slow_provider"
+    name = "Slow Provider"
+    model_id_key = "model"
+    models = ["model_a"]
+
+    def __init__(self, **kwargs):
+        kwargs["responses"] = ["This fake response will be used for completion"]
+        kwargs["model_id"] = "model_a"
+        sleep(10)
+        super().__init__(**kwargs)
diff --git a/tests/test_shortcuts.py b/tests/test_shortcuts.py
@@ -1,4 +1,5 @@
 import pytest
+import time
 from IPython.terminal.interactiveshell import PtkHistoryAdapter
 from IPython.terminal.shortcuts.auto_suggest import (
     accept,
@@ -61,6 +62,17 @@ async def test_llm_autosuggestion():
     assert event.current_buffer.suggestion.text == FIBONACCI[len(text) :]
 
 
+def test_slow_llm_provider_should_not_block_init():
+    ip = get_ipython()
+    provider = NavigableAutoSuggestFromHistory()
+    ip.auto_suggest = provider
+    start = time.perf_counter()
+    ip.llm_provider_class = "tests.fake_llm.SlowStartingCompletionProvider"
+    end = time.perf_counter()
+    elapsed = end - start
+    assert elapsed < 0.1
+
+
 @pytest.mark.parametrize(
     "text, suggestion, expected",
     [
