clean up

setup.py

@@ -4,7 +4,7 @@ from setuptools import setup, find_packages
 setup(
   name = 'swarms',
   packages = find_packages(exclude=[]),
-  version = '0.4.5',
+  version = '0.4.6',
   license='MIT',
   description = 'Swarms - Pytorch',
   author = 'Kye Gomez',

swarms/agents/workers/agents.py

@@ -224,8 +224,7 @@ from langchain.callbacks.base import BaseCallbackHandler
 from langchain.schema import AgentAction, AgentFinish, LLMResult
 from celery import Task
 
-from ansi import ANSI, Color, Style, dim_multiline
-from logger import logger
+from swarms.utils import ANSI, Color, Style, dim_multiline, logger
 
 
 class EVALCallbackHandler(BaseCallbackHandler):

swarms/tools/main.py

@@ -23,17 +23,7 @@ from bs4 import BeautifulSoup
 
 # from logger import logger
 
-
-import logging
-
-logger = logging.getLogger()
-formatter = logging.Formatter("%(message)s")
-ch = logging.StreamHandler()
-ch.setFormatter(formatter)
-logger.addHandler(ch)
-
-logger.setLevel(logging.DEBUG)
-
+from swarms.utils import logger
 
 class ToolScope(Enum):
     GLOBAL = "global"

swarms/utils/__init__.py

@@ -1 +1,2 @@
-from swarms.utils.ansi import Code, Color, Style, ANSI, dim_multiline
+from swarms.utils.ansi import Code, Color, Style, ANSI, dim_multiline
+from swarms.utils.logger import logger

swarms/utils/logger.py

@@ -0,0 +1,9 @@
+import logging
+
+logger = logging.getLogger()
+formatter = logging.Formatter("%(message)s")
+ch = logging.StreamHandler()
+ch.setFormatter(formatter)
+logger.addHandler(ch)
+
+logger.setLevel(logging.DEBUG)