pull/430/head
Kye 10 months ago
parent 10829b03e2
commit 9b225e955b

@ -4,6 +4,7 @@ from typing import List
# import faiss # import faiss
from langchain.docstore import InMemoryDocstore from langchain.docstore import InMemoryDocstore
from langchain.embeddings import OpenAIEmbeddings from langchain.embeddings import OpenAIEmbeddings
# from langchain.vectorstores import FAISS # from langchain.vectorstores import FAISS
from langchain_experimental.autonomous_agents import AutoGPT from langchain_experimental.autonomous_agents import AutoGPT

@ -28,6 +28,9 @@ class BaseWorkflow(BaseStructure):
**kwargs, **kwargs,
): ):
super().__init__(*args, **kwargs) super().__init__(*args, **kwargs)
self.agents = agents
self.task_pool = task_pool
self.models = models
self.task_pool = [] self.task_pool = []
self.agent_pool = [] self.agent_pool = []

@ -9,6 +9,7 @@ logger.add(
diagnose=True, diagnose=True,
) )
def loguru_logger(file_path: str = "swarms.log"): def loguru_logger(file_path: str = "swarms.log"):
return logger.add( return logger.add(
file_path, file_path,
@ -18,4 +19,3 @@ def loguru_logger(file_path: str = "swarms.log"):
backtrace=True, backtrace=True,
diagnose=True, diagnose=True,
) )

@ -97,10 +97,6 @@ def get_new_dataframe_name(org_img_name, func_name="update"):
return os.path.join(head, new_file_name) return os.path.join(head, new_file_name)
STATIC_DIR = "static" STATIC_DIR = "static"
@ -114,8 +110,6 @@ class AbstractUploader(ABC):
pass pass
class FileType(Enum): class FileType(Enum):
IMAGE = "image" IMAGE = "image"
AUDIO = "audio" AUDIO = "audio"

Loading…
Cancel
Save