langchain_cohere: add cohere as a partner package

docs/docs/integrations/chat/cohere.ipynb

@@ -32,7 +32,7 @@
     "The integration lives in the `langchain-community` package. We also need to install the `cohere` package itself. We can install these with:\n",
     "\n",
     "```bash\n",
-    "pip install -U langchain-community cohere\n",
+    "pip install -U langchain-community langchain-cohere\n",
     "```\n",
     "\n",
     "We'll also need to get a [Cohere API key](https://cohere.com/) and set the `COHERE_API_KEY` environment variable:"
@@ -89,7 +89,7 @@
    },
    "outputs": [],
    "source": [
-    "from langchain_community.chat_models import ChatCohere\n",
+    "from langchain_cohere import ChatCohere\n",
     "from langchain_core.messages import HumanMessage"
    ]
   },

docs/docs/integrations/llms/cohere.ipynb

@@ -24,7 +24,7 @@
     "The integration lives in the `langchain-community` package. We also need to install the `cohere` package itself. We can install these with:\n",
     "\n",
     "```bash\n",
-    "pip install -U langchain-community cohere\n",
+    "pip install -U langchain-community langchain-cohere\n",
     "```\n",
     "\n",
     "We'll also need to get a [Cohere API key](https://cohere.com/) and set the `COHERE_API_KEY` environment variable:"
@@ -39,7 +39,7 @@
    },
    "outputs": [
     {
-     "name": "stdin",
+     "name": "stdout",
      "output_type": "stream",
      "text": [
       " ········\n"
@@ -91,7 +91,7 @@
    },
    "outputs": [],
    "source": [
-    "from langchain_community.llms import Cohere\n",
+    "from langchain_cohere import Cohere\n",
     "from langchain_core.messages import HumanMessage"
    ]
   },
@@ -255,7 +255,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.11.1"
+   "version": "3.11.7"
   }
  },
  "nbformat": 4,

docs/docs/integrations/retrievers/cohere.ipynb

@@ -32,7 +32,7 @@
    },
    "outputs": [],
    "source": [
-    "from langchain_community.chat_models import ChatCohere\n",
+    "from langchain_cohere import ChatCohere\n",
     "from langchain_community.retrievers import CohereRagRetriever\n",
     "from langchain_core.documents import Document"
    ]

docs/docs/integrations/text_embedding/cohere.ipynb

@@ -30,7 +30,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain_community.embeddings import CohereEmbeddings"
+    "from langchain_cohere import CohereEmbeddings"
    ]
   },
   {

libs/community/langchain_community/chat_models/cohere.py

@@ -1,5 +1,6 @@
 from typing import Any, AsyncIterator, Dict, Iterator, List, Optional
 
+from langchain_core._api.deprecation import deprecated
 from langchain_core.callbacks import (
     AsyncCallbackManagerForLLMRun,
     CallbackManagerForLLMRun,
@@ -94,6 +95,9 @@ def get_cohere_chat_request(
     return {k: v for k, v in req.items() if v is not None}
 
 
+@deprecated(
+    since="0.0.30", removal="0.2.0", alternative_import="langchain_cohere.ChatCohere"
+)
 class ChatCohere(BaseChatModel, BaseCohere):
     """`Cohere` chat large language models.
 

libs/community/langchain_community/embeddings/cohere.py

@@ -1,12 +1,18 @@
 from typing import Any, Dict, List, Optional
 
+from langchain_core._api.deprecation import deprecated
 from langchain_core.embeddings import Embeddings
 from langchain_core.pydantic_v1 import BaseModel, Extra, root_validator
 from langchain_core.utils import get_from_dict_or_env
 
 from langchain_community.llms.cohere import _create_retry_decorator
 
 
+@deprecated(
+    since="0.0.30",
+    removal="0.2.0",
+    alternative_import="langchain_cohere.CohereEmbeddings",
+)
 class CohereEmbeddings(BaseModel, Embeddings):
     """Cohere embedding models.
 

libs/community/langchain_community/llms/cohere.py

@@ -3,6 +3,7 @@
 import logging
 from typing import Any, Callable, Dict, List, Optional
 
+from langchain_core._api.deprecation import deprecated
 from langchain_core.callbacks import (
     AsyncCallbackManagerForLLMRun,
     CallbackManagerForLLMRun,
@@ -69,6 +70,9 @@ async def _completion_with_retry(**kwargs: Any) -> Any:
     return _completion_with_retry(**kwargs)
 
 
+@deprecated(
+    since="0.0.30", removal="0.2.0", alternative_import="langchain_cohere.BaseCohere"
+)
 class BaseCohere(Serializable):
     """Base class for Cohere models."""
 
@@ -117,6 +121,9 @@ def validate_environment(cls, values: Dict) -> Dict:
         return values
 
 
+@deprecated(
+    since="0.1.14", removal="0.2.0", alternative_import="langchain_cohere.Cohere"
+)
 class Cohere(LLM, BaseCohere):
     """Cohere large language models.
 

libs/community/langchain_community/retrievers/cohere_rag_retriever.py

@@ -2,6 +2,7 @@
 
 from typing import TYPE_CHECKING, Any, Dict, List
 
+from langchain_core._api.deprecation import deprecated
 from langchain_core.callbacks import (
     AsyncCallbackManagerForRetrieverRun,
     CallbackManagerForRetrieverRun,
@@ -40,6 +41,11 @@ def _get_docs(response: Any) -> List[Document]:
     return docs
 
 
+@deprecated(
+    since="0.0.30",
+    removal="0.2.0",
+    alternative_import="langchain_cohere.CohereRagRetriever",
+)
 class CohereRagRetriever(BaseRetriever):
     """Cohere Chat API with RAG."""
 

libs/langchain/langchain/retrievers/document_compressors/cohere_rerank.py

@@ -3,6 +3,7 @@
 from copy import deepcopy
 from typing import Any, Dict, List, Optional, Sequence, Union
 
+from langchain_core._api.deprecation import deprecated
 from langchain_core.documents import Document
 from langchain_core.pydantic_v1 import Extra, root_validator
 
@@ -11,6 +12,9 @@
 from langchain.utils import get_from_dict_or_env
 
 
+@deprecated(
+    since="0.0.30", removal="0.2.0", alternative_import="langchain_cohere.CohereRerank"
+)
 class CohereRerank(BaseDocumentCompressor):
     """Document compressor that uses `Cohere Rerank API`."""
 

libs/partners/cohere/.gitignore

@@ -0,0 +1 @@
+__pycache__

libs/partners/cohere/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 LangChain, Inc.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

libs/partners/cohere/Makefile

@@ -0,0 +1,57 @@
+.PHONY: all format lint test tests integration_tests docker_tests help extended_tests
+
+# Default target executed when no arguments are given to make.
+all: help
+
+# Define a variable for the test file path.
+TEST_FILE ?= tests/unit_tests/
+integration_test integration_tests: TEST_FILE=tests/integration_tests/
+
+test tests integration_test integration_tests:
+	poetry run pytest $(TEST_FILE)
+
+
+######################
+# LINTING AND FORMATTING
+######################
+
+# Define a variable for Python and notebook files.
+PYTHON_FILES=.
+MYPY_CACHE=.mypy_cache
+lint format: PYTHON_FILES=.
+lint_diff format_diff: PYTHON_FILES=$(shell git diff --relative=libs/partners/cohere --name-only --diff-filter=d master | grep -E '\.py$$|\.ipynb$$')
+lint_package: PYTHON_FILES=langchain_cohere
+lint_tests: PYTHON_FILES=tests
+lint_tests: MYPY_CACHE=.mypy_cache_test
+
+lint lint_diff lint_package lint_tests:
+	poetry run ruff .
+	poetry run ruff format $(PYTHON_FILES) --diff
+	poetry run ruff --select I $(PYTHON_FILES)
+	mkdir $(MYPY_CACHE); poetry run mypy $(PYTHON_FILES) --cache-dir $(MYPY_CACHE)
+
+format format_diff:
+	poetry run ruff format $(PYTHON_FILES)
+	poetry run ruff --select I --fix $(PYTHON_FILES)
+
+spell_check:
+	poetry run codespell --toml pyproject.toml
+
+spell_fix:
+	poetry run codespell --toml pyproject.toml -w
+
+check_imports: $(shell find langchain_cohere -name '*.py')
+	poetry run python ./scripts/check_imports.py $^
+
+######################
+# HELP
+######################
+
+help:
+	@echo '----'
+	@echo 'check_imports				- check imports'
+	@echo 'format                       - run code formatters'
+	@echo 'lint                         - run linters'
+	@echo 'test                         - run unit tests'
+	@echo 'tests                        - run unit tests'
+	@echo 'test TEST_FILE=<test_file>   - run all tests in file'

libs/partners/cohere/README.md

@@ -0,0 +1 @@
+# langchain-cohere

libs/partners/cohere/langchain_cohere/__init__.py

@@ -0,0 +1,12 @@
+from langchain_cohere.chat_models import ChatCohere
+from langchain_cohere.embeddings import CohereEmbeddings
+from langchain_cohere.rag_retrievers import CohereRagRetriever
+from langchain_cohere.rerank import CohereRerank
+
+__all__ = [
+    "ChatCohere",
+    "CohereVectorStore",
+    "CohereEmbeddings",
+    "CohereRagRetriever",
+    "CohereRerank",
+]