diff --git a/swarms/swarms.py b/swarms/swarms.py index f23302b1..c54f609a 100644 --- a/swarms/swarms.py +++ b/swarms/swarms.py @@ -45,7 +45,7 @@ tools = [ Tool(name='process_csv', func=process_csv, description='Processes a CSV file'), Tool(name='query_website_tool', func=WebpageQATool(qa_chain=load_qa_with_sources_chain(llm)), description='Queries a website'), - + # Tool(name='terminal', func=Terminal.execute, description='Operates a terminal'), # Tool(name='code_writer', func=CodeWriter(), description='Writes code'), # Tool(name='code_editor', func=CodeEditor(), description='Edits code'), @@ -58,8 +58,9 @@ index = faiss.IndexFlatL2(embedding_size) vectorstore = FAISS(embeddings_model.embed_query, index, InMemoryDocstore({}), {}) # ---------- Worker Node ---------- -@tool("WorkerAgent", return_direct=True) +@tool class WorkerNode: + """Useful for when you need to spawn an autonomous agent instance as a worker to accomplish complex tasks, it can search the internet or spawn child multi-modality models to process and generate images and text or audio and so on """ def __init__(self, llm, tools, vectorstore): self.llm = llm self.tools = tools @@ -143,10 +144,8 @@ tool_names = [tool.name for tool in tools] agent = ZeroShotAgent(llm_chain=llm_chain, allowed_tools=tool_names) agent_executor = AgentExecutor.from_agent_and_tools(agent=agent, tools=tools, verbose=True) - boss_node = BossNode(llm=llm, vectorstore=vectorstore, task_execution_chain=agent_executor, verbose=True, max_iterations=5) - class Swarms: def __init__(self, openai_api_key): self.openai_api_key = openai_api_key @@ -162,6 +161,7 @@ class Swarms: Tool(name='read_file_tool', func=ReadFileTool(root_dir=ROOT_DIR), description='Reads a file'), Tool(name='process_csv', func=process_csv, description='Processes a CSV file'), Tool(name='query_website_tool', func=WebpageQATool(qa_chain=load_qa_with_sources_chain(llm)), description='Queries a website'), + # Tool(name='terminal', func=Terminal.execute, description='Operates a terminal'), # Tool(name='code_writer', func=CodeWriter(), description='Writes code'), # Tool(name='code_editor', func=CodeEditor(), description='Edits code'), @@ -212,10 +212,6 @@ class Swarms: - - - - # class Swarms: # def __init__(self, num_nodes: int, llm: BaseLLM, self_scaling: bool): # self.nodes = [WorkerNode(llm) for _ in range(num_nodes)]