[CLEANUP]

playground/agents/multion_agent.py

@@ -1,12 +1,10 @@
 from swarms.agents.multion_agent import MultiOnAgent
-from swarms.structs.agent import Agent
+import timeit
-from swarms.structs.concurrent_workflow import ConcurrentWorkflow
+from swarms import Agent, ConcurrentWorkflow, Task
-from swarms.structs.task import Task
+from swarms.utils.loguru_logger import logger
 
 # model
-model = MultiOnAgent(
+model = MultiOnAgent(multion_api_key="")
-    multion_api_key=""
-)
 
 # out = model.run("search for a recipe")
 agent = Agent(
@@ -17,6 +15,7 @@ agent = Agent(
     system_prompt=None,
 )
 
+logger.info("[Agent][ID][MultiOnAgent][Initialized][Successfully")
 
 # Task
 task = Task(
@@ -27,16 +26,25 @@ task = Task(
     ),
 )
 
-
 # Swarm
+logger.info(
+    f"Running concurrent workflow with task: {task.description}"
+)
+
+# Measure execution time
+start_time = timeit.default_timer()
+
 workflow = ConcurrentWorkflow(
-    max_workers=21,
+    max_workers=1,
     autosave=True,
     print_results=True,
     return_results=True,
 )
 
-
 # Add task to workflow
 workflow.add(task)
 workflow.run()
+
+# Calculate execution time
+execution_time = timeit.default_timer() - start_time
+logger.info(f"Execution time: {execution_time} seconds")

playground/structs/kyle_hackathon.py

@@ -37,7 +37,9 @@ def multion_tool(
 
 # Execute the interpreter tool
 @tool
-def execute_interpreter_tool(code: str,):
+def execute_interpreter_tool(
+    code: str,
+):
     """
     Executes a single command using the interpreter.
 

pyproject.toml

@@ -6,7 +6,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "swarms"
-version = "4.1.8"
+version = "4.1.9"
 description = "Swarms - Pytorch"
 license = "MIT"
 authors = ["Kye Gomez <kye@apac.ai>"]

runtime/multi_threading.rs

@@ -29,8 +29,8 @@ fn my_module(py: Python, m: &PyModule) -> PyResult<()> {
     Ok(())
 }
 
-#[pyfunction]
+
-fn process_python_modules(modules: Vec<PythonModule>, num_threads: usize) -> Result<(), PythonError> {
+
 /// The function returns `Ok(())` if all modules are processed successfully.
 /// Note: This code assumes that the necessary dependencies (`pyo3`, `rayon`, `log`) are already imported and initialized.
 ///
@@ -71,6 +71,9 @@ fn process_python_modules(modules: Vec<PythonModule>, num_threads: usize) -> Res
 ///
 /// This function is safe to call, but it assumes that the necessary dependencies (`pyo3`, `rayon`, `log`) are already imported and initialized.
 // Initialize Python interpreter
+#[pyfunction]
+fn process_python_modules(modules: Vec<PythonModule>, num_threads: usize) -> Result<(), PythonError> {
+
     let gil = Python::acquire_gil();
     let py = gil.python();
 

tests/agents/test_multion.py

@@ -0,0 +1,57 @@
+import pytest
+from unittest.mock import patch, MagicMock
+from swarms.agents.multion_agent import MultiOnAgent
+
+
+@patch("swarms.agents.multion_agent.multion")
+def test_multion_agent_run(mock_multion):
+    mock_response = MagicMock()
+    mock_response.result = "result"
+    mock_response.status = "status"
+    mock_response.lastUrl = "lastUrl"
+    mock_multion.browse.return_value = mock_response
+
+    agent = MultiOnAgent(
+        multion_api_key="test_key",
+        max_steps=5,
+        starting_url="https://www.example.com",
+    )
+    result, status, last_url = agent.run("task")
+
+    assert result == "result"
+    assert status == "status"
+    assert last_url == "lastUrl"
+    mock_multion.browse.assert_called_once_with(
+        {
+            "cmd": "task",
+            "url": "https://www.example.com",
+            "maxSteps": 5,
+        }
+    )
+
+
+# Additional tests for different tasks
+@pytest.mark.parametrize(
+    "task", ["task1", "task2", "task3", "task4", "task5"]
+)
+@patch("swarms.agents.multion_agent.multion")
+def test_multion_agent_run_different_tasks(mock_multion, task):
+    mock_response = MagicMock()
+    mock_response.result = "result"
+    mock_response.status = "status"
+    mock_response.lastUrl = "lastUrl"
+    mock_multion.browse.return_value = mock_response
+
+    agent = MultiOnAgent(
+        multion_api_key="test_key",
+        max_steps=5,
+        starting_url="https://www.example.com",
+    )
+    result, status, last_url = agent.run(task)
+
+    assert result == "result"
+    assert status == "status"
+    assert last_url == "lastUrl"
+    mock_multion.browse.assert_called_once_with(
+        {"cmd": task, "url": "https://www.example.com", "maxSteps": 5}
+    )