[Bug] SequentialWorkflow.run accepts imgs and then drops it (#1822)

* [Bug] SequentialWorkflow.run accepts imgs and then drops it

run() takes an imgs parameter and documents it as "Optional list of
images for the agents", but only task and img ever make it into the
kwargs handed to AgentRearrange:

    run_kwargs = {"task": task}
    if img is not None:
        run_kwargs["img"] = img

So a multi-image run silently degrades to a text-only run. No error,
no warning -- the agents just answer a question about images they were
never shown.

AgentRearrange forwards **kwargs down to Agent.run, which does accept
imgs, so passing it through is all that's needed.

* Annotate run_kwargs as Dict[str, Any] so Pyre accepts the imgs list

Pyre inferred Dict[str, str] from the task entry alone, so assigning the
imgs list tripped 'Incompatible parameter type [6]' in code scanning.

Author: ayaangazali

swarms/structs/sequential_workflow.py

@@ -2,7 +2,7 @@
 import json
 import os
 from concurrent.futures import as_completed
-from typing import Callable, List, Optional, Union
+from typing import Any, Callable, Dict, List, Optional, Union
 
 from loguru import logger as loguru_logger
 from swarms.prompts.multi_agent_collab_prompt import (
@@ -326,9 +326,13 @@ def run(
         """
         try:
             # prompt = f"{MULTI_AGENT_COLLAB_PROMPT}\n\n{task}"
-            run_kwargs = {"task": task}
+            # Annotated because imgs is a list: without it the dict is
+            # inferred as Dict[str, str] from the task entry alone.
+            run_kwargs: Dict[str, Any] = {"task": task}
             if img is not None:
                 run_kwargs["img"] = img
+            if imgs is not None:
+                run_kwargs["imgs"] = imgs
             result = self.agent_rearrange.run(**run_kwargs)
 
             # Run drift detection if configured

tests/structs/test_sequential_workflow.py

@@ -612,3 +612,30 @@ def test_workflow_drift_max_retries_zero_never_reruns():
 def test_negative_drift_max_retries_is_rejected():
     with pytest.raises(ValueError, match="drift_max_retries"):
         _make_workflow(drift_detection=True, drift_max_retries=-1)
+
+
+def test_run_forwards_imgs_to_the_pipeline():
+    """run(imgs=[...]) must reach the agents, not be dropped.
+
+    imgs is an accepted, documented parameter, but it was never put
+    into the kwargs handed to AgentRearrange, so multi-image runs
+    silently became text-only runs.
+    """
+    wf = _make_workflow()
+    with patch.object(
+        wf.agent_rearrange, "run", return_value="out"
+    ) as pipeline:
+        wf.run("describe these", imgs=["a.png", "b.png"])
+
+    assert pipeline.call_args.kwargs["imgs"] == ["a.png", "b.png"]
+
+
+def test_run_omits_imgs_when_not_supplied():
+    """No imgs key when the caller didn't pass one."""
+    wf = _make_workflow()
+    with patch.object(
+        wf.agent_rearrange, "run", return_value="out"
+    ) as pipeline:
+        wf.run("plain task")
+
+    assert "imgs" not in pipeline.call_args.kwargs