diff --git a/spacy/ml/richfeatureextractor.pyi b/spacy/ml/richfeatureextractor.pyi
index d89d11f19..f3bf57f3d 100644
--- a/spacy/ml/richfeatureextractor.pyi
+++ b/spacy/ml/richfeatureextractor.pyi
@@ -1,5 +1,14 @@
+from typing import List, Optional
+from thinc.types import Ints2d
+from thinc.api import Model
 from ..tokens import Doc
 
+def RichFeatureExtractor(
+    *,
+    case_sensitive: bool,
+    pref_lengths: Optional[List[int]] = None,
+    suff_lengths: Optional[List[int]] = None,
+) -> Model[List[Doc], List[Ints2d]]: ...
 def get_character_combination_hashes(
     *,
     doc: Doc,