Merge pull request #26 from NASA-IMPACT/bugfix/document-retriever

Bugfix `larch.indexing.DocumentIndexer.as_langchain_retriever(...)`
NASA-IMPACT · Feb 14, 2024 · 8e8489b · 8e8489b
2 parents 3440f77 + 15ecf3d
commit 8e8489b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/larch/indexing/_base.py b/larch/indexing/_base.py
@@ -122,6 +122,6 @@ def query(self, query: str, **kwargs) -> Response:
 
     def as_langchain_retriever(self) -> Type[BaseRetriever]:
         # to avoid circular import
-        from .search.chains import DocumentIndexerAsRetriever
+        from ..search.chains import DocumentIndexerAsRetriever
 
         return DocumentIndexerAsRetriever(document_indexer=self)
diff --git a/larch/search/chains.py b/larch/search/chains.py
@@ -1,6 +1,6 @@
 #!/usr/bin/env python3
 
-from typing import List, Type
+from typing import List
 
 from langchain.callbacks.manager import CallbackManagerForRetrieverRun
 from langchain.schema.retriever import BaseRetriever
@@ -10,7 +10,7 @@
 
 
 class DocumentIndexerAsRetriever(BaseRetriever):
-    document_indexer: Type[DocumentIndexer]
+    document_indexer: DocumentIndexer
 
     def _get_relevant_documents(
         self,