diff --git a/bingchat.py b/playground/models/bingchat.py similarity index 100% rename from bingchat.py rename to playground/models/bingchat.py diff --git a/swarms/__init__.py b/swarms/__init__.py index ad2d5986..42ce223e 100644 --- a/swarms/__init__.py +++ b/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) \ No newline at end of file diff --git a/swarms/models/__init__.py b/swarms/models/__init__.py index 84b26f3e..67260c79 100644 --- a/swarms/models/__init__.py +++ b/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.zephyr import Zephyr \ No newline at end of file +from swarms.models.vilt import Vilt \ No newline at end of file diff --git a/swarms/tools/edge_gpt.py b/swarms/tools/edge_gpt.py index e69de29b..bef44cfb 100644 --- a/swarms/tools/edge_gpt.py +++ b/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) \ No newline at end of file