It is calling Cohere

This commit is contained in:
JackColquitt 2024-08-01 12:11:30 -07:00
parent db006a4b7b
commit 7f08d1e3a2
2 changed files with 3 additions and 1 deletions

View file

@ -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 && \

View file

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