fix provider type hint

djcopley · Feb 4, 2024 · b2bc495 · b2bc495
1 parent 22b0d3e
commit b2bc495
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/shelloracle/providers/__init__.py b/src/shelloracle/providers/__init__.py
@@ -39,7 +39,7 @@ def generate(self, prompt: str) -> AsyncIterator[str]:
         # https://mypy.readthedocs.io/en/stable/more_types.html#asynchronous-iterators
 
 
-def _providers() -> dict[str, Provider]:
+def _providers() -> dict[str, type[Provider]]:
     from .ollama import Ollama
     from .openai import OpenAI
     from .localai import LocalAI