update the folder of oepnvino rerank

update the folder of oepnvino rerank update the folder of oepnvino rerank update the folder of oepnvino rerank
langchain-ai · Mar 31, 2024 · d3d03ab · d3d03ab
1 parent 809fa81
commit d3d03ab
Show file tree

Hide file tree

Showing 8 changed files with 5 additions and 6 deletions.
diff --git a/docs/docs/integrations/llms/openvino.ipynb b/docs/docs/integrations/llms/openvino.ipynb
@@ -5,7 +5,7 @@
    "id": "959300d4",
    "metadata": {},
    "source": [
-    "# OpenVINO Local Pipelines\n",
+    "# OpenVINO\n",
     "\n",
     "[OpenVINO™](https://github.com/openvinotoolkit/openvino) is an open-source toolkit for optimizing and deploying AI inference. OpenVINO™ Runtime can enable running the same model optimized across various hardware [devices](https://github.com/openvinotoolkit/openvino?tab=readme-ov-file#supported-hardware-matrix). Accelerate your deep learning performance across use cases like: language + LLMs, computer vision, automatic speech recognition, and more.\n",
     "\n",

diff --git a/docs/docs/integrations/retrievers/openvino_rerank.ipynb b/docs/docs/integrations/retrievers/openvino_rerank.ipynb
@@ -437,7 +437,7 @@
    ],
    "source": [
     "from langchain.retrievers import ContextualCompressionRetriever\n",
-    "from langchain_community.retrievers import OpenVINOReranker\n",
+    "from langchain.retrievers.document_compressors.openvino_rerank import OpenVINOReranker\n",
     "\n",
     "model_name = \"BAAI/bge-reranker-large\"\n",
     "\n",

diff --git a/docs/docs/integrations/text_embedding/openvino.ipynb b/docs/docs/integrations/text_embedding/openvino.ipynb
@@ -5,7 +5,7 @@
    "id": "ed47bb62",
    "metadata": {},
    "source": [
-    "# OpenVINO Local Pipelines\n",
+    "# OpenVINO\n",
     "[OpenVINO™](https://github.com/openvinotoolkit/openvino) is an open-source toolkit for optimizing and deploying AI inference. The OpenVINO™ Runtime supports various hardware [devices](https://github.com/openvinotoolkit/openvino?tab=readme-ov-file#supported-hardware-matrix) including x86 and ARM CPUs, and Intel GPUs. It can help to boost deep learning performance in Computer Vision, Automatic Speech Recognition, Natural Language Processing and other common tasks.\n",
     "\n",
     "Hugging Face embedding model can be supported by OpenVINO through ``OpenVINOEmbeddings`` class. If you have an Intel GPU, you can specify `model_kwargs={\"device\": \"GPU\"}` to run inference on it."

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/community/tests/unit_tests/retrievers/test_imports.py b/libs/community/tests/unit_tests/retrievers/test_imports.py
@@ -23,7 +23,6 @@
     "MetalRetriever",
     "MilvusRetriever",
     "OutlineRetriever",
-    "OpenVINOReranker",
     "PineconeHybridSearchRetriever",
     "PubMedRetriever",
     "QdrantSparseVectorRetriever",

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",