Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 13 additions & 3 deletions paddlex/inference/pipelines/components/retriever/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -21,13 +21,23 @@
from .....utils.deps import class_requires_deps, is_dep_available
from .....utils.subclass_register import AutoRegisterABCMetaClass

if is_dep_available("langchain"):
from langchain.docstore.document import Document
from langchain.text_splitter import RecursiveCharacterTextSplitter

if is_dep_available("langchain-community"):
from langchain_community import vectorstores
from langchain_community.vectorstores import FAISS

# Document import capability
try:
from langchain_core.documents import Document
except ImportError:
from langchain.docstore.document import Document

# Text splitter compatibility
try:
from langchain_text_splitters import RecursiveCharacterTextSplitter
except ImportError:
from langchain.text_splitter import RecursiveCharacterTextSplitter


@class_requires_deps("langchain", "langchain-community")
class BaseRetriever(ABC, metaclass=AutoRegisterABCMetaClass):
Expand Down
7 changes: 4 additions & 3 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -40,10 +40,11 @@
"jieba": "",
"Jinja2": "",
"joblib": "",
"langchain": ">= 0.2, < 1.0",
"langchain-community": ">= 0.2, < 1.0",
"langchain": ">= 0.2",
"langchain-community": ">= 0.2",
"langchain-core": "",
"langchain-openai": ">= 0.1, < 1.0",
"langchain-openai": ">= 0.1",
"langchain_text_splitters": "",
"lxml": "",
"matplotlib": "",
"modelscope": ">=1.28.0",
Expand Down