Provide type hints for pass-through Langchain and Llama-index objects (#95)

knowledgehub/indices/extractors/doc_parsers.py

@@ -6,6 +6,14 @@ class BaseDocParser(DocTransformer):
 
 
 class TitleExtractor(LlamaIndexDocTransformerMixin, BaseDocParser):
+    def __init__(
+        self,
+        llm=None,
+        nodes: int = 5,
+        **params,
+    ):
+        super().__init__(llm=llm, nodes=nodes, **params)
+
     def _get_li_class(self):
         from llama_index.extractors import TitleExtractor
 
@@ -13,6 +21,14 @@ class TitleExtractor(LlamaIndexDocTransformerMixin, BaseDocParser):
 
 
 class SummaryExtractor(LlamaIndexDocTransformerMixin, BaseDocParser):
+    def __init__(
+        self,
+        llm=None,
+        summaries: list[str] = ["self"],
+        **params,
+    ):
+        super().__init__(llm=llm, summaries=summaries, **params)
+
     def _get_li_class(self):
         from llama_index.extractors import SummaryExtractor