edgegpt clean up

Former-commit-id: b79c748e34

swarms/__init__.py

@@ -1,12 +1,3 @@
-from swarms import workers
-from swarms.workers.worker import Worker
-from swarms.chunkers import chunkers
-from swarms import models
-from swarms import structs
-from swarms import swarms
-from swarms.swarms.orchestrate import Orchestrator
-from swarms import agents
-from swarms.logo import logo
 import os
 import warnings
 
@@ -17,4 +8,15 @@ warnings.filterwarnings("ignore", category=UserWarning)
 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "2"
 
 
-print(logo)
+
+
+from swarms import workers
+from swarms.workers.worker import Worker
+from swarms.chunkers import chunkers
+from swarms import models
+from swarms import structs
+from swarms import swarms
+from swarms.swarms.orchestrate import Orchestrator
+from swarms import agents
+from swarms.logo import logo
+print(logo)

swarms/models/__init__.py

@@ -1,11 +1,12 @@
+# LLMs
 from swarms.models.anthropic import Anthropic
 from swarms.models.petals import Petals
 from swarms.models.mistral import Mistral
 from swarms.models.openai_models import OpenAI, AzureOpenAI, OpenAIChat
+from swarms.models.zephyr import Zephyr
 
 
 # MultiModal Models
 from swarms.models.idefics import Idefics
 from swarms.models.kosmos_two import Kosmos
-from swarms.models.vilt import Vilt
+from swarms.models.vilt import Vilt
-from swarms.models.zephyr import Zephyr

swarms/tools/edge_gpt.py

@@ -0,0 +1,9 @@
+from swarms.tools.tool import BaseTool
+
+class EdgeGPTTool(BaseTool):
+    def __init__(self, model, name="EdgeGPTTool", description="Tool that uses EdgeGPTModel to generate responses"):
+        super().__init__(name=name, description=description)
+        self.model = model
+
+    def _run(self, prompt):
+        return self.model.__call__(prompt)