diff --git a/api/container.py b/api/container.py index 6d32c6d4..1ffeea25 100644 --- a/api/container.py +++ b/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"]) diff --git a/swarms/agents/utils/callback.py b/swarms/agents/utils/callback.py index d4720fd2..cca65037 100644 --- a/swarms/agents/utils/callback.py +++ b/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 diff --git a/swarms/agents/utils/llm.py b/swarms/agents/utils/llm.py index d4e7ee18..b09cc817 100644 --- a/swarms/agents/utils/llm.py +++ b/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]: diff --git a/swarms/agents/workers/agents.py b/swarms/agents/workers/agents.py index 8419dd95..2e70d4e5 100644 --- a/swarms/agents/workers/agents.py +++ b/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): diff --git a/swarms/agents/workers/worker_ultranode.py b/swarms/agents/workers/worker_ultranode.py index 08230671..04108a84 100644 --- a/swarms/agents/workers/worker_ultranode.py +++ b/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"]) diff --git a/swarms/tools/main.py b/swarms/tools/main.py index dedfafc1..67a32b9b 100644 --- a/swarms/tools/main.py +++ b/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): diff --git a/swarms/utils/utils.py b/swarms/utils/main.py similarity index 100% rename from swarms/utils/utils.py rename to swarms/utils/main.py diff --git a/swarms/utils/static.py b/swarms/utils/static.py index 7b642857..72acbcef 100644 --- a/swarms/utils/static.py +++ b/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):