From 8fe317b909edc09b1187080483c0ca71adcb0c46 Mon Sep 17 00:00:00 2001
From: Kye <kye@apacmediasolutions.com>
Date: Fri, 15 Mar 2024 10:00:55 -0700
Subject: [PATCH] [CLEANUP]

---
 .gitignore                                    |  1 +
 .../demos/ai_acceleerated_learning/Vocal.py   | 46 -------------------
 pyproject.toml                                | 13 ++----
 requirements.txt                              |  1 +
 swarms/models/__init__.py                     | 33 +++++++++----
 5 files changed, 31 insertions(+), 63 deletions(-)
 delete mode 100644 playground/demos/ai_acceleerated_learning/Vocal.py

diff --git a/.gitignore b/.gitignore
index a583d768..ba330622 100644
--- a/.gitignore
+++ b/.gitignore
@@ -21,6 +21,7 @@ Cargo.lock
 Cargo.lock
 swarms/agents/.DS_Store
 
+logs
 _build
 conversation.txt
 stderr_log.txt
diff --git a/playground/demos/ai_acceleerated_learning/Vocal.py b/playground/demos/ai_acceleerated_learning/Vocal.py
deleted file mode 100644
index 85470156..00000000
--- a/playground/demos/ai_acceleerated_learning/Vocal.py
+++ /dev/null
@@ -1,46 +0,0 @@
-import pytest
-
-def test_create_youtube_account():
-    # Arrange
-    # Act
-    # Assert
-
-def test_install_video_editing_software():
-    # Arrange
-    # Act
-    # Assert
-
-def test_write_script():
-    # Arrange
-    # Act
-    # Assert
-
-def test_gather_footage():
-    # Arrange
-    # Act
-    # Assert
-
-def test_edit_video():
-    # Arrange
-    # Act
-    # Assert
-
-def test_export_video():
-    # Arrange
-    # Act
-    # Assert
-
-def test_upload_video_to_youtube():
-    # Arrange
-    # Act
-    # Assert
-
-def test_optimize_video_for_search():
-    # Arrange
-    # Act
-    # Assert
-
-def test_share_video():
-    # Arrange
-    # Act
-    # Assert
\ No newline at end of file
diff --git a/pyproject.toml b/pyproject.toml
index 82554af5..5d9c7fd1 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -24,30 +24,29 @@ classifiers = [
 
 
 [tool.poetry.dependencies]
-python = "^3.9"
+python = ">=3.9,<4.0"
 torch = "2.1.1"
-transformers = "4.37.1"
-openai = "0.28.0"
-langchain = "0.0.333"
+transformers = "*"
 asyncio = "3.4.3"
 einops = "0.7.0"
 google-generativeai = "0.3.1"
 langchain-experimental = "0.0.10"
-opencv-python-headless = "4.8.1.78"
+langchain-community = "*"
 faiss-cpu = "1.7.4"
 backoff = "2.2.1"
 datasets = "*"
 optimum = "1.15.0"
 diffusers = "*"
+langchain = "*"
 toml = "*"
 pypdf = "4.0.1"
 accelerate = "*"
-anthropic = "*"
 sentencepiece = "0.1.98"
 httpx = "0.24.1"
 tiktoken = "0.4.0"
 ratelimit = "2.2.1"
 loguru = "0.7.2"
+pydantic-settings = "*"
 huggingface-hub = "*"
 pydantic = "*"
 tenacity = "8.2.2"
@@ -78,8 +77,6 @@ types-chardet = "^5.0.4.6"
 mypy-protobuf = "^3.0.0"
 
 
-
-
 [tool.ruff]
 line-length = 70
 # Enable Pyflakes (`F`) and a subset of the pycodestyle (`E`)  codes by default.
diff --git a/requirements.txt b/requirements.txt
index 6cf9831d..3e7a86a1 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -31,6 +31,7 @@ einops==0.7.0
 opencv-python-headless==4.8.1.78
 numpy
 openai==0.28.0
+langchain-community
 opencv-python==4.9.0.80
 timm
 cohere==4.53
diff --git a/swarms/models/__init__.py b/swarms/models/__init__.py
index 4c2e9ac5..259ada12 100644
--- a/swarms/models/__init__.py
+++ b/swarms/models/__init__.py
@@ -1,4 +1,12 @@
-from swarms.models.anthropic import Anthropic  # noqa: E402
+from langchain_community.llms import (
+    Anthropic,
+    AzureOpenAI,
+    Cohere,
+    MosaicML,
+    OpenAI,
+    OpenAIChat,
+    Replicate,
+)
 from swarms.models.base_embedding_model import BaseEmbeddingModel
 from swarms.models.base_llm import AbstractLLM  # noqa: E402
 from swarms.models.base_multimodal_model import BaseMultiModalModel
@@ -40,7 +48,6 @@ from swarms.models.zephyr import Zephyr  # noqa: E402
 from swarms.models.zeroscope import ZeroscopeTTV  # noqa: E402
 
 __all__ = [
-    "Anthropic",
     "AbstractLLM",
     "BaseEmbeddingModel",
     "BaseMultiModalModel",
@@ -48,18 +55,18 @@ __all__ = [
     "CLIPQ",
     "FireFunctionCaller",
     "Fuyu",
+    "Gemini",
     "Gigabind",
     "GPT4VisionAPI",
     "HuggingfaceLLM",
     "Idefics",
     "Kosmos",
-    "LavaMultiModal",
     "LayoutLMDocumentQA",
+    "LavaMultiModal",
     "Mistral",
     "Mixtral",
     "MPT7B",
     "Nougat",
-    "Gemini",
     "OpenAITTS",
     "Petals",
     "QwenVLMultiModal",
@@ -69,14 +76,22 @@ __all__ = [
     "SamplingType",
     "TimmModel",
     "TogetherLLM",
-    "UltralyticsModel",
-    "Vilt",
-    "WizardLLMStoryTeller",
-    "Zephyr",
-    "ZeroscopeTTV",
     "AudioModality",
     "ImageModality",
     "MultimodalData",
     "TextModality",
     "VideoModality",
+    "UltralyticsModel",
+    "Vilt",
+    "WizardLLMStoryTeller",
+    "Zephyr",
+    "ZeroscopeTTV",
+    "AzureChatOpenAI",
+    "OpenAIChat",
+    "Anthropic",
+    "AzureOpenAI",
+    "Cohere",
+    "MosaicML",
+    "OpenAI",
+    "Replicate",
 ]