diff --git a/Containerfile b/Containerfile index e01bd7a2..38b60b6a 100644 --- a/Containerfile +++ b/Containerfile @@ -13,7 +13,7 @@ RUN dnf install -y python3 python3-pip python3-wheel python3-aiohttp \ RUN pip3 install torch --index-url https://download.pytorch.org/whl/cpu -RUN pip3 install anthropic google-cloud-aiplatform langchain langchain-core \ +RUN pip3 install anthropic cohere google-cloud-aiplatform langchain langchain-core \ langchain-huggingface langchain-text-splitters langchain-community \ pymilvus sentence-transformers transformers huggingface-hub \ pulsar-client && \ diff --git a/setup.py b/setup.py index 6e9c8b9e..1fec809b 100644 --- a/setup.py +++ b/setup.py @@ -45,6 +45,7 @@ setuptools.setup( "pyyaml", "prometheus-client", "pyarrow", + "cohere", ], scripts=[ "scripts/chunker-recursive", @@ -71,6 +72,7 @@ setuptools.setup( "scripts/text-completion-claude", "scripts/text-completion-ollama", "scripts/text-completion-vertexai", + "scripts/text-completion-cohere", "scripts/triples-dump-parquet", "scripts/triples-write-cassandra", ]