Spaces:
Running
Running
Fix langchain chains and retrievers imports for 0.3.x
Browse files
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.
|
| 5 |
-
from langchain.retrievers
|
| 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 |
+
|