Abeshith commited on
Commit
945dc2a
·
1 Parent(s): 6215836

Fix langchain chains and retrievers imports for 0.3.x

Browse files
Files changed (1) hide show
  1. project/model/retriever.py +3 -2
project/model/retriever.py CHANGED
@@ -1,8 +1,8 @@
1
  from typing import List, Optional
2
  from langchain_core.documents import Document
3
  from langchain_community.vectorstores import FAISS
4
- from langchain.chains.query_constructor.base import AttributeInfo
5
- from langchain.retrievers.self_query.base import SelfQueryRetriever
6
  from project.utils.model_loader import ModelLoader
7
  from project.utils.config_loader import load_config
8
  from project.logger.logging import get_logger
@@ -91,3 +91,4 @@ class DocumentRetriever:
91
 
92
  logger.info(f"Base retriever configured with {search_type} search")
93
  return self.retriever
 
 
1
  from typing import List, Optional
2
  from langchain_core.documents import Document
3
  from langchain_community.vectorstores import FAISS
4
+ from langchain.chains.query_constructor.schema import AttributeInfo
5
+ from langchain.retrievers import SelfQueryRetriever
6
  from project.utils.model_loader import ModelLoader
7
  from project.utils.config_loader import load_config
8
  from project.logger.logging import get_logger
 
91
 
92
  logger.info(f"Base retriever configured with {search_type} search")
93
  return self.retriever
94
+