clean up run swarms

swarms/swarms.py

@@ -3,11 +3,7 @@ import logging
 from swarms.tools.agent_tools import *
 from swarms.agents.workers.worker import WorkerNode, worker_tool
 from swarms.agents.boss.boss_agent import BossNode
-
 from swarms.tools.main import RequestsGet
-from swarms.agents.workers.multi_modal_workers.multi_modal_agent import MultiModalVisualAgent
-
-
 
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
 
@@ -19,7 +15,7 @@ class Swarms:
         # Initialize language model
         return llm_class(openai_api_key=self.openai_api_key, temperature=temperature)
 
-    def initialize_tools(self, llm_class, worker_node):
+    def initialize_tools(self, llm_class):
         llm = self.initialize_llm(llm_class)
         # Initialize tools
         web_search = DuckDuckGoSearchRun()
@@ -33,7 +29,6 @@ class Swarms:
 
             # RequestsGet()
             Tool(name="RequestsGet", func=RequestsGet.get, description="A portal to the internet, Use this when you need to get specific content from a website. Input should be a  url (i.e. https://www.google.com). The output will be the text response of the GET request."),
-            Tool(name="WorkerNode AI Agent", func=worker_node.run, description="Input: an objective with a todo list for that objective. Output: your task completed: Please be very clear what the objective and task instructions are. The Swarm worker agent is Useful for when you need to spawn an autonomous agent instance as a worker to accomplish any complex tasks, it can search the internet or write code or spawn child multi-modality models to process and generate images and text or audio and so on")
 
             
             # CodeEditor,
@@ -82,28 +77,14 @@ class Swarms:
         return BossNode(llm, vectorstore, agent_executor, max_iterations=5)
 
 
-    def run_swarms(self, objective, run_as="worker"):
-        # try:
-        #     # Run the swarm with the given objective
-        #     worker_tools = self.initialize_tools(OpenAI)
-        #     assert worker_tools is not None, "worker_tools is not initialized"
-
-        #     vectorstore = self.initialize_vectorstore()
-        #     worker_node = self.initialize_worker_node(worker_tools, vectorstore)
-
-        #     boss_node = self.initialize_boss_node(vectorstore, worker_node)
-
-        #     task = boss_node.create_task(objective)
-        #     return boss_node.execute_task(task)
-        # except Exception as e:
-        #     logging.error(f"An error occurred in run_swarms: {e}")
-        #     raise
-        #===============> optional approach
+    def run_swarms(self, objective, run_as='worker'):
         try:
+            # Run the swarm with the given objective
+            worker_tools = self.initialize_tools(OpenAI)
+            assert worker_tools is not None, "worker_tools is not initialized"
+
             vectorstore = self.initialize_vectorstore()
-            worker_node = self.initialize_worker_node([], vectorstore)  
-            worker_tools = self.initialize_tools(OpenAI, worker_node)  
-            worker_node.tools = worker_tools  # Set the tools attribute of the worker_node to worker_tools
+            worker_node = self.initialize_worker_node(worker_tools, vectorstore)
 
             if run_as.lower() == 'worker':
                 tool_input = {'prompt': objective}
@@ -120,17 +101,6 @@ class Swarms:
 
 
 
-# usage
-def swarm(api_key, objective):
-    swarms = Swarms(api_key)
-    return swarms.run_swarms(objective)
-
-# # Use the function
-# api_key = "APIKEY"
-# objective = "What is the capital of the UK?"
-# result = swarm(api_key, objective)
-# print(result)  # Prints: "The capital of the UK is London."
-
 
 
 
@@ -247,6 +217,20 @@ def swarm(api_key, objective):
 
 
 
+# usage
+def swarm(api_key, objective):
+    swarms = Swarms(api_key)
+    return swarms.run_swarms(objective)
+
+# # Use the function
+# api_key = "APIKEY"
+# objective = "What is the capital of the UK?"
+# result = swarm(api_key, objective)
+# print(result)  # Prints: "The capital of the UK is London."
+
+
+
+