update the folder of oepnvino rerank

langchain-ai · Mar 31, 2024 · 6a3abd8 · 6a3abd8
1 parent 809fa81
commit 6a3abd8
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 2 deletions.
diff --git a/libs/community/langchain_community/document_compressors/__init__.py b/libs/community/langchain_community/document_compressors/__init__.py
@@ -3,6 +3,7 @@
 
 _module_lookup = {
     "LLMLinguaCompressor": "langchain_community.document_compressors.llmlingua_filter",
+    "OpenVINOReranker": "langchain_community.document_compressors.openvino_rerank",
 }
 
 

diff --git a/...n_community/retrievers/openvino_rerank.py → ...y/document_compressors/openvino_rerank.py b/...n_community/retrievers/openvino_rerank.py → ...y/document_compressors/openvino_rerank.py
diff --git a/libs/community/langchain_community/retrievers/__init__.py b/libs/community/langchain_community/retrievers/__init__.py
@@ -45,7 +45,6 @@
     "LlamaIndexRetriever": "langchain_community.retrievers.llama_index",
     "MetalRetriever": "langchain_community.retrievers.metal",
     "MilvusRetriever": "langchain_community.retrievers.milvus",
-    "OpenVINOReranker": "langchain_community.retrievers.openvino_rerank",
     "OutlineRetriever": "langchain_community.retrievers.outline",
     "PineconeHybridSearchRetriever": "langchain_community.retrievers.pinecone_hybrid_search",  # noqa: E501
     "PubMedRetriever": "langchain_community.retrievers.pubmed",

diff --git a/libs/langchain/langchain/retrievers/document_compressors/openvino_rerank.py b/libs/langchain/langchain/retrievers/document_compressors/openvino_rerank.py
@@ -1,4 +1,4 @@
-from langchain_community.retrievers.openvino_rerank import OpenVINOReranker
+from langchain_community.document_compressors.openvino_rerank import OpenVINOReranker
 
 __all__ = [
     "OpenVINOReranker",