From cc27ee5396a66c4ea8296ffae68d6f55f900ce03 Mon Sep 17 00:00:00 2001 From: Selcuk Ayguney Date: Thu, 5 Sep 2024 14:23:47 +1000 Subject: [PATCH] Fixed dependencies Added missing dependencies, and pinned `mistralai` as 1.x is incompatible with PyLLMs. --- requirements.txt | 4 +++- setup.py | 6 ++++-- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/requirements.txt b/requirements.txt index f33c17c..29dcf92 100644 --- a/requirements.txt +++ b/requirements.txt @@ -15,4 +15,6 @@ protobuf grpcio~=1.54.2 google-generativeai ollama -reka-api \ No newline at end of file +reka-api +mistralai<1.0 +together diff --git a/setup.py b/setup.py index e6cbae6..5d485cd 100644 --- a/setup.py +++ b/setup.py @@ -19,7 +19,7 @@ install_requires=[ "openai>=1", "tiktoken", - "anthropic>=0.18", + "anthropic>=0.3", "ai21", "cohere", "aleph-alpha-client", @@ -29,8 +29,10 @@ "protobuf>=3.20.3", "grpcio>=1.54.2", "google-generativeai", - "mistralai", + "mistralai<1.0", "ollama", + "reka-api", + "together", ], extras_require={ "local": ["einops", "accelerate"]