diff --git a/swarms/agents/worker_agent.py b/swarms/agents/worker_agent.py index d11ac925..12760318 100644 --- a/swarms/agents/worker_agent.py +++ b/swarms/agents/worker_agent.py @@ -4,6 +4,7 @@ from typing import List # import faiss from langchain.docstore import InMemoryDocstore from langchain.embeddings import OpenAIEmbeddings + # from langchain.vectorstores import FAISS from langchain_experimental.autonomous_agents import AutoGPT diff --git a/swarms/structs/base_workflow.py b/swarms/structs/base_workflow.py index c06d9339..89ac99bf 100644 --- a/swarms/structs/base_workflow.py +++ b/swarms/structs/base_workflow.py @@ -28,6 +28,9 @@ class BaseWorkflow(BaseStructure): **kwargs, ): super().__init__(*args, **kwargs) + self.agents = agents + self.task_pool = task_pool + self.models = models self.task_pool = [] self.agent_pool = [] diff --git a/swarms/utils/loguru_logger.py b/swarms/utils/loguru_logger.py index 1b0a42f4..818baa8d 100644 --- a/swarms/utils/loguru_logger.py +++ b/swarms/utils/loguru_logger.py @@ -9,6 +9,7 @@ logger.add( diagnose=True, ) + def loguru_logger(file_path: str = "swarms.log"): return logger.add( file_path, @@ -18,4 +19,3 @@ def loguru_logger(file_path: str = "swarms.log"): backtrace=True, diagnose=True, ) - diff --git a/swarms/utils/main.py b/swarms/utils/main.py index 2ca8f262..ffb496d1 100644 --- a/swarms/utils/main.py +++ b/swarms/utils/main.py @@ -97,10 +97,6 @@ def get_new_dataframe_name(org_img_name, func_name="update"): return os.path.join(head, new_file_name) - - - - STATIC_DIR = "static" @@ -114,8 +110,6 @@ class AbstractUploader(ABC): pass - - class FileType(Enum): IMAGE = "image" AUDIO = "audio"