[TESTS][CLEANUP]]

kyegomez · Apr 25, 2024 · 2f88e92 · 2f88e92
1 parent f93bc98
commit 2f88e92
Show file tree

Hide file tree

Showing 29 changed files with 304 additions and 2,940 deletions.
diff --git a/.gitignore b/.gitignore
@@ -15,6 +15,7 @@ Unit Testing Agent_state.json
 Devin_state.json
 swarms/__pycache__
 artifacts
+transcript_generator.json
 venv
 .DS_Store
 Cargo.lock

diff --git a/example.py b/example.py
@@ -20,9 +20,11 @@
 )
 
 # Run the Agent on a task
-# out = agent("Generate a transcript for a youtube video on what swarms are!")
+out = agent.run("Generate a transcript for a youtube video on what swarms are!")
+print(out)
+
+# Save the state
 check = agent.save_state(
     "transcript_generator.json",
     "Generate a transcript for a youtube video on what swarms are!",
 )
-print(check)
diff --git a/swarms/models/__init__.py b/swarms/models/__init__.py
@@ -13,6 +13,7 @@
 from swarms.models.mixtral import Mixtral  # noqa: E402
 from swarms.models.mpt import MPT7B  # noqa: E402
 from swarms.models.nougat import Nougat  # noqa: E402
+from swarms.models.palm import GooglePalm as Palm  # noqa: E402
 from swarms.models.openai_tts import OpenAITTS  # noqa: E402
 from swarms.models.popular_llms import (
     AnthropicChat as Anthropic,
@@ -42,7 +43,7 @@
     VideoModality,
 )
 from swarms.models.vilt import Vilt  # noqa: E402
-
+from swarms.models.openai_embeddings import OpenAIEmbeddings
 
 __all__ = [
     "AbstractLLM",
@@ -62,20 +63,22 @@
     "Mistral",
     "Mixtral",
     "MPT7B",
-    "MultimodalData",
     "Nougat",
     "OpenAI",
     "OpenAIChat",
+    "OpenAIEmbeddings",
     "OpenAITTS",
+    "OctoAIChat",
+    "Palm",
     "QwenVLMultiModal",
     "Replicate",
     "SamplingParams",
     "SamplingType",
     "TextModality",
-    "TogetherLLM",
-    "Vilt",
-    "AudioModality",
+    "MultimodalData",
     "ImageModality",
+    "AudioModality",
     "VideoModality",
-    "OctoAIChat",
-]
+    "TogetherLLM",
+    "Vilt",
+]
diff --git a/swarms/models/gpt4_sam.py b/swarms/models/gpt4_sam.py
diff --git a/swarms/models/huggingface_pipeline.py b/swarms/models/huggingface_pipeline.py
@@ -4,12 +4,7 @@
 from termcolor import colored
 
 from swarms.models.base_llm import AbstractLLM
-
-if torch.cuda.is_available():
-    try:
-        from optimum.nvidia.pipelines import pipeline
-    except ImportError:
-        from transformers.pipelines import pipeline
+from transformers.pipelines import pipeline
 
 
 class HuggingfacePipeline(AbstractLLM):