clean up

Former-commit-id: 9d9638ee62b475883b8ed12b27e3a7c9e93606af

api/container.py

@@ -6,12 +6,12 @@ from typing import Dict, List
 from fastapi.templating import Jinja2Templates
 
 from swarms.agents.utils.manager import AgentManager
-from swarms.utils.utils import BaseHandler, FileHandler, FileType
+from swarms.utils.main import BaseHandler, FileHandler, FileType
 from swarms.tools.main import CsvToDataframe, ExitConversation, RequestsGet, CodeEditor, Terminal
 
 from swarms.tools.main import BaseToolSet
 
-from swarms.utils.utils import StaticUploader
+from swarms.utils.main import StaticUploader
 
 BASE_DIR = Path(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 os.chdir(BASE_DIR / os.environ["PLAYGROUND_DIR"])

swarms/agents/utils/callback.py

@@ -5,7 +5,7 @@ from langchain.schema import AgentAction, AgentFinish, LLMResult
 # from celery import Task
 
 # from ansi import ANSI, Color, Style, dim_multiline
-from swarms.utils.utils import ANSI, Color, Style, dim_multiline
+from swarms.utils.main import ANSI, Color, Style, dim_multiline
 from swarms.utils.logger import logger
 
 

swarms/agents/utils/llm.py

@@ -30,7 +30,7 @@ from tenacity import (
 
 
 # from ansi import ANSI, Color, Style
-from swarms.utils.utils import ANSI, Color, Style
+from swarms.utils.main import ANSI, Color, Style
 import os
 
 def _create_retry_decorator(llm: ChatOpenAI) -> Callable[[Any], Any]:

swarms/agents/workers/agents.py

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

swarms/agents/workers/worker_ultranode.py

@@ -3,10 +3,10 @@ from pathlib import Path
 from typing import Dict, List
 
 from swarms.agents.utils.manager import AgentManager
-from swarms.utils.utils import BaseHandler, FileHandler, FileType
+from swarms.utils.main import BaseHandler, FileHandler, FileType
 from swarms.tools.main import CsvToDataframe, ExitConversation, RequestsGet, CodeEditor, Terminal
 from swarms.tools.main import BaseToolSet
-from swarms.utils.utils import StaticUploader
+from swarms.utils.main import StaticUploader
 
 BASE_DIR = Path(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 os.chdir(BASE_DIR / os.environ["PLAYGROUND_DIR"])

swarms/tools/main.py

@@ -428,7 +428,7 @@ import time
 from datetime import datetime
 from typing import Dict, List
 
-from swarms.utils.utils import ANSI, Color, Style # test
+from swarms.utils.main import ANSI, Color, Style # test
 
 class Terminal(BaseToolSet):
     def __init__(self):
@@ -1100,7 +1100,7 @@ from transformers import (
 )
 
 
-from swarms.utils.utils import get_new_image_name
+from swarms.utils.main import get_new_image_name
 
 
 class MaskFormer(BaseToolSet):
@@ -1309,7 +1309,7 @@ from transformers import BlipForConditionalGeneration, BlipProcessor
 # from core.prompts.file import IMAGE_PROMPT
 from swarms.prompts.prompts import IMAGE_PROMPT
 
-from swarms.utils.utils import BaseHandler
+from swarms.utils.main import BaseHandler
 
 class ImageCaptioning(BaseHandler):
     def __init__(self, device):

swarms/utils/static.py

@@ -4,7 +4,7 @@ from pathlib import Path
 
 # from env import DotEnv
 
-from swarms.utils.utils import AbstractUploader
+from swarms.utils.main import AbstractUploader
 
 class StaticUploader(AbstractUploader):
     def __init__(self, server: str, path: Path, endpoint: str):