50+ tests for cohere

kyegomez · Nov 11, 2023 · bdc7337 · bdc7337
1 parent 4197920
commit bdc7337
Show file tree

Hide file tree

Showing 9 changed files with 931 additions and 338 deletions.
diff --git a/playground/models/cohere_example.py b/playground/models/cohere_example.py
@@ -0,0 +1,6 @@
+from swarms.models.cohere_chat import Cohere
+
+
+cohere = Cohere(model="command-light", cohere_api_key="")
+
+out = cohere("Hello, how are you?")
diff --git a/pyproject.toml b/pyproject.toml
@@ -48,6 +48,7 @@ attrs = "*"
 ggl = "*"
 ratelimit = "*"
 beautifulsoup4 = "*"
+cohere = "*"
 huggingface-hub = "*"
 pydantic = "*"
 tenacity = "*"

diff --git a/requirements.txt b/requirements.txt
@@ -64,6 +64,7 @@ webdataset
 yapf
 autopep8
 dalle3
+cohere
 torchvision
 rich
 

diff --git a/swarms/models/cohere.py b/swarms/models/cohere.py