chore: typing

phil65 · Jan 27, 2025 · e01acd1 · e01acd1
1 parent a19e6d0
commit e01acd1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/slashed/prompt_toolkit_completer.py b/src/slashed/prompt_toolkit_completer.py
@@ -14,7 +14,7 @@
 
 
 if TYPE_CHECKING:
-    from collections.abc import AsyncIterable, Iterable
+    from collections.abc import AsyncGenerator, Iterable
 
     from prompt_toolkit.document import Document
 
@@ -57,7 +57,7 @@ def get_completions(
 
     async def get_completions_async(
         self, document: Document, complete_event: Any
-    ) -> AsyncIterable[Completion]:
+    ) -> AsyncGenerator[Completion, None]:
         """Get completions for the current context."""
         text = document.text.lstrip()