From e6127ac6e395fbff58ce53e5b225ae2a759c9d27 Mon Sep 17 00:00:00 2001 From: Kye Date: Mon, 25 Sep 2023 20:59:32 -0400 Subject: [PATCH] clean up --- swarms/__init__.py | 2 -- swarms/agents/__init__.py | 2 -- swarms/agents/omni_modal_agent.py | 5 ++++- swarms/logo.py | 14 +++++++------- 4 files changed, 11 insertions(+), 12 deletions(-) diff --git a/swarms/__init__.py b/swarms/__init__.py index f016d8f8..5ac610e1 100644 --- a/swarms/__init__.py +++ b/swarms/__init__.py @@ -18,8 +18,6 @@ from swarms.models.petals import Petals from swarms.workers.worker import Worker #from swarms.models.openai import OpenAIChat - - #structs from swarms.structs.workflow import Workflow diff --git a/swarms/agents/__init__.py b/swarms/agents/__init__.py index 946c60cf..6a1cfea3 100644 --- a/swarms/agents/__init__.py +++ b/swarms/agents/__init__.py @@ -14,8 +14,6 @@ # from swarms.tools.developer import Terminal, CodeEditor - - #agents from swarms.agents.profitpilot import ProfitPilot from swarms.agents.aot import AoTAgent diff --git a/swarms/agents/omni_modal_agent.py b/swarms/agents/omni_modal_agent.py index b5bc218d..a0d183a2 100644 --- a/swarms/agents/omni_modal_agent.py +++ b/swarms/agents/omni_modal_agent.py @@ -12,7 +12,10 @@ class OmniModalAgent: self.api_endpoint = api_endpoint self.api_type = api_type - def chat(self, data): + def chat( + self, + data + ): """Chat with omni-modality model that uses huggingface to query for a specific model at run time. Translate text to speech, create images and more""" messages = data.get("messages") api_key = data.get("api_key", self.api_key) diff --git a/swarms/logo.py b/swarms/logo.py index ceab8406..99d5ce08 100644 --- a/swarms/logo.py +++ b/swarms/logo.py @@ -1,10 +1,10 @@ -logo = """ - ________ _ _______ _______ _____ ______ - / ___/\ \/ \/ /\__ \\_ __ \/ \ / ___/ - \___ \ \ / / __ \| | \/ Y Y \\___ \ -/____ > \/\_/ (____ /__| |__|_| /____ > - \/ \/ \/ \/ -""" +# logo = """ +# ________ _ _______ _______ _____ ______ +# / ___/\ \/ \/ /\__ \\_ __ \/ \ / ___/ +# \___ \ \ / / __ \| | \/ Y Y \\___ \ +# /____ > \/\_/ (____ /__| |__|_| /____ > +# \/ \/ \/ \/ +# """ logo2 = """