swarms 7.9.3__py3-none-any.whl → 7.9.4__py3-none-any.whl

swarms/structs/ma_utils.py

@@ -1,8 +1,11 @@
-from typing import List, Any, Optional, Union, Callable
+from typing import Dict, List, Any, Optional, Union, Callable
 import random
 from swarms.prompts.collaborative_prompts import (
     get_multi_agent_collaboration_prompt_one,
 )
+from functools import lru_cache
+
+from loguru import logger
 
 
 def list_all_agents(
@@ -116,3 +119,65 @@ def set_random_models_for_agents(
     else:
         setattr(agents, "model_name", random.choice(model_names))
         return agents
+
+
+@lru_cache(maxsize=128)
+def _create_agent_map_cached(
+    agent_tuple: tuple,
+) -> Dict[str, Union[Callable, Any]]:
+    """Internal cached version of create_agent_map that takes a tuple for hashability."""
+    try:
+        return {
+            (
+                agent.agent_name
+                if isinstance(agent, Callable)
+                else agent.__name__
+            ): agent
+            for agent in agent_tuple
+        }
+    except (AttributeError, TypeError) as e:
+        logger.error(f"Error creating agent map: {e}")
+        return {}
+
+
+def create_agent_map(
+    agents: List[Union[Callable, Any]],
+) -> Dict[str, Union[Callable, Any]]:
+    """Creates a map of agent names to agents for fast lookup.
+
+    This function is optimized with LRU caching to avoid recreating maps for identical agent lists.
+    The cache stores up to 128 different agent map configurations.
+
+    Args:
+        agents (List[Union[Callable, Any]]): List of agents to create a map of. Each agent should either be:
+            - A callable with a __name__ attribute
+            - An object with an agent_name attribute
+
+    Returns:
+        Dict[str, Union[Callable, Any]]: Map of agent names to agents
+
+    Examples:
+        >>> def agent1(): pass
+        >>> def agent2(): pass
+        >>> agents = [agent1, agent2]
+        >>> agent_map = create_agent_map(agents)
+        >>> print(agent_map.keys())
+        dict_keys(['agent1', 'agent2'])
+
+        >>> class Agent:
+        ...     def __init__(self, name):
+        ...         self.agent_name = name
+        >>> agents = [Agent("bot1"), Agent("bot2")]
+        >>> agent_map = create_agent_map(agents)
+        >>> print(agent_map.keys())
+        dict_keys(['bot1', 'bot2'])
+
+    Raises:
+        ValueError: If agents list is empty
+        TypeError: If any agent lacks required name attributes
+    """
+    if not agents:
+        raise ValueError("Agents list cannot be empty")
+
+    # Convert list to tuple for hashability
+    return _create_agent_map_cached(tuple(agents))

swarms/utils/formatter.py

@@ -5,9 +5,27 @@ from typing import Any, Callable, Dict, List, Optional
 from rich.console import Console
 from rich.live import Live
 from rich.panel import Panel
-from rich.progress import Progress, SpinnerColumn, TextColumn
+from rich.progress import (
+    Progress,
+    SpinnerColumn,
+    TextColumn,
+)
 from rich.table import Table
 from rich.text import Text
+from rich.spinner import Spinner
+
+# Global lock to ensure only a single Rich Live context is active at any moment.
+# Rich's Live render is **not** thread-safe; concurrent Live contexts on the same
+# console raise runtime errors. Using a module-level lock serialises access and
+# prevents crashes when multiple agents stream simultaneously in different
+# threads (e.g., in ConcurrentWorkflow).
+live_render_lock = threading.Lock()
+
+# Global Live display for the dashboard
+dashboard_live = None
+
+# Create a spinner for loading animation
+spinner = Spinner("dots", style="yellow")
 
 
 def choose_random_color():
@@ -37,6 +55,53 @@ class Formatter:
         Initializes the Formatter with a Rich Console instance.
         """
         self.console = Console()
+        self._dashboard_live = None
+        self._spinner_frames = [
+            "⠋",
+            "⠙",
+            "⠹",
+            "⠸",
+            "⠼",
+            "⠴",
+            "⠦",
+            "⠧",
+            "⠇",
+            "⠏",
+        ]
+        self._spinner_idx = 0
+
+    def _get_status_with_loading(self, status: str) -> Text:
+        """
+        Creates a status text with loading animation for running status.
+        """
+        if status.lower() == "running":
+            # Create loading bar effect
+            self._spinner_idx = (self._spinner_idx + 1) % len(
+                self._spinner_frames
+            )
+            spinner_char = self._spinner_frames[self._spinner_idx]
+            progress_bar = "█" * (self._spinner_idx % 5) + "░" * (
+                4 - (self._spinner_idx % 5)
+            )
+            return Text(
+                f"{spinner_char} {status} {progress_bar}",
+                style="bold yellow",
+            )
+
+        # Style other statuses
+        status_style = {
+            "completed": "bold green",
+            "pending": "bold red",
+            "error": "bold red",
+        }.get(status.lower(), "white")
+
+        status_symbol = {
+            "completed": "✓",
+            "pending": "○",
+            "error": "✗",
+        }.get(status.lower(), "•")
+
+        return Text(f"{status_symbol} {status}", style=status_style)
 
     def _print_panel(
         self, content: str, title: str = "", style: str = "bold blue"
@@ -209,58 +274,155 @@ class Formatter:
         complete_response = ""
         chunks_collected = []
 
-        # TRUE streaming with Rich's automatic text wrapping
-        with Live(
-            create_streaming_panel(streaming_text),
-            console=self.console,
-            refresh_per_second=20,
-        ) as live:
-            try:
-                for part in streaming_response:
-                    if (
-                        hasattr(part, "choices")
-                        and part.choices
-                        and part.choices[0].delta.content
-                    ):
-                        # Add ONLY the new chunk to the Text object with random color style
-                        chunk = part.choices[0].delta.content
-                        streaming_text.append(chunk, style=text_style)
-                        complete_response += chunk
-
-                        # Collect chunks if requested
-                        if collect_chunks:
-                            chunks_collected.append(chunk)
-
-                        # Call chunk callback if provided
-                        if on_chunk_callback:
-                            on_chunk_callback(chunk)
-
-                        # Update display with new text - Rich handles all wrapping automatically
-                        live.update(
-                            create_streaming_panel(
-                                streaming_text, is_complete=False
+        # Acquire the lock so that only one Live panel is active at a time.
+        # Other threads will wait here until the current streaming completes,
+        # avoiding Rich.Live concurrency errors.
+        with live_render_lock:
+            # TRUE streaming with Rich's automatic text wrapping
+            with Live(
+                create_streaming_panel(streaming_text),
+                console=self.console,
+                refresh_per_second=20,
+            ) as live:
+                try:
+                    for part in streaming_response:
+                        if (
+                            hasattr(part, "choices")
+                            and part.choices
+                            and part.choices[0].delta.content
+                        ):
+                            # Add ONLY the new chunk to the Text object with random color style
+                            chunk = part.choices[0].delta.content
+                            streaming_text.append(
+                                chunk, style=text_style
+                            )
+                            complete_response += chunk
+
+                            # Collect chunks if requested
+                            if collect_chunks:
+                                chunks_collected.append(chunk)
+
+                            # Call chunk callback if provided
+                            if on_chunk_callback:
+                                on_chunk_callback(chunk)
+
+                            # Update display with new text - Rich handles all wrapping automatically
+                            live.update(
+                                create_streaming_panel(
+                                    streaming_text, is_complete=False
+                                )
                             )
+
+                    # Final update to show completion
+                    live.update(
+                        create_streaming_panel(
+                            streaming_text, is_complete=True
                         )
+                    )
 
-                # Final update to show completion
-                live.update(
-                    create_streaming_panel(
-                        streaming_text, is_complete=True
+                except Exception as e:
+                    # Handle any streaming errors gracefully
+                    streaming_text.append(
+                        f"\n[Error: {str(e)}]", style="bold red"
                     )
-                )
+                    live.update(
+                        create_streaming_panel(
+                            streaming_text, is_complete=True
+                        )
+                    )
+
+        return complete_response
+
+    def _create_dashboard_table(
+        self, agents_data: List[Dict[str, Any]], title: str
+    ) -> Panel:
+        """
+        Creates the dashboard table with the current agent statuses.
+        """
+        # Create main table
+        table = Table(
+            show_header=True,
+            header_style="bold magenta",
+            expand=True,
+            title=title,
+            title_style="bold cyan",
+            border_style="bright_blue",
+            show_lines=True,  # Add lines between rows
+        )
+
+        # Add columns with adjusted widths
+        table.add_column(
+            "Agent Name", style="cyan", width=30, no_wrap=True
+        )
+        table.add_column(
+            "Status", style="green", width=20, no_wrap=True
+        )  # Increased width for loading animation
+        table.add_column(
+            "Output", style="white", width=100, overflow="fold"
+        )  # Allow text to wrap
+
+        # Add rows for each agent
+        for agent in agents_data:
+            name = Text(agent["name"], style="bold cyan")
+            status = self._get_status_with_loading(agent["status"])
+            output = Text(str(agent["output"]))
+            table.add_row(name, status, output)
+
+        # Create a panel to wrap the table
+        dashboard_panel = Panel(
+            table,
+            border_style="bright_blue",
+            padding=(1, 2),
+            title=f"[bold cyan]{title}[/bold cyan] - Total Agents: [bold green]{len(agents_data)}[/bold green]",
+            expand=True,  # Make panel expand to full width
+        )
+
+        return dashboard_panel
 
-            except Exception as e:
-                # Handle any streaming errors gracefully
-                streaming_text.append(
-                    f"\n[Error: {str(e)}]", style="bold red"
+    def print_agent_dashboard(
+        self,
+        agents_data: List[Dict[str, Any]],
+        title: str = "🤖 Agent Dashboard",
+        is_final: bool = False,
+    ) -> None:
+        """
+        Displays a beautiful dashboard showing agent information in a panel-like spreadsheet format.
+        Updates in place instead of printing multiple times.
+
+        Args:
+            agents_data (List[Dict[str, Any]]): List of dictionaries containing agent information.
+                Each dict should have: name, status, output
+            title (str): The title of the dashboard.
+            is_final (bool): Whether this is the final update of the dashboard.
+        """
+        with live_render_lock:
+            if self._dashboard_live is None:
+                # Create new Live display if none exists
+                self._dashboard_live = Live(
+                    self._create_dashboard_table(agents_data, title),
+                    console=self.console,
+                    refresh_per_second=10,  # Increased refresh rate
+                    transient=False,  # Make display persistent
                 )
-                live.update(
-                    create_streaming_panel(
-                        streaming_text, is_complete=True
-                    )
+                self._dashboard_live.start()
+            else:
+                # Update existing Live display
+                self._dashboard_live.update(
+                    self._create_dashboard_table(agents_data, title)
                 )
 
-        return complete_response
+            # If this is the final update, add a newline to separate from future output
+            if is_final:
+                self.console.print()  # Add blank line after final display
+
+    def stop_dashboard(self):
+        """
+        Stops and cleans up the dashboard display.
+        """
+        if self._dashboard_live is not None:
+            self._dashboard_live.stop()
+            self.console.print()  # Add blank line after stopping
+            self._dashboard_live = None
 
 
 formatter = Formatter()

{swarms-7.9.3.dist-info → swarms-7.9.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: swarms
-Version: 7.9.3
+Version: 7.9.4
 Summary: Swarms - TGSC
 License: MIT
 Keywords: artificial intelligence,deep learning,optimizers,Prompt Engineering,swarms,agents,llms,transformers,multi-agent,swarms of agents,Enterprise-Grade Agents,Production-Grade Agents,Agents,Multi-Grade-Agents,Swarms,Transformers,LLMs,Prompt Engineering,Agents,Generative Agents,Generative AI,Agent Marketplace,Agent Store,quant,finance,algorithmic trading,portfolio optimization,risk management,financial modeling,machine learning for finance,natural language processing for finance

{swarms-7.9.3.dist-info → swarms-7.9.4.dist-info}/RECORD

@@ -12,7 +12,7 @@ swarms/agents/gkp_agent.py,sha256=5Jms3zHQ2qwJ6-PHDh9X-cFtAlH4dSUoDgRqN-xZzog,21
 swarms/agents/i_agent.py,sha256=_kqGt3a4SGB21_GP-KcV8A5p_9wneShs6eJUJYFdluw,12274
 swarms/agents/openai_assistant.py,sha256=mTSEtj26J0mc5pCeWrmMY0EXzTRYQfyfw_BtOqtcCHc,11044
 swarms/agents/react_agent.py,sha256=yM8lQoRsqJZicqtmgBrC7RHv0aKEb5oZHh4q5aAA_xs,5804
-swarms/agents/reasoning_agents.py,sha256=Z_YETF-hzG7qL-h6WB1qNR6QU4g784M9NzO5C0_kWtA,9639
+swarms/agents/reasoning_agents.py,sha256=0pdvNtRBtBJU-fGYeGfplxQkyxZGTqkZisqpAq1zGkU,8631
 swarms/agents/reasoning_duo.py,sha256=4qw9RtwmWkWnDWjEtJqYFVLh--EV9YQd24-fQa0lMfM,3873
 swarms/agents/self_agent_builder.py,sha256=bX7xSwak6HiyK901VdeE8OlT4yqE0n7jyHcWJrkMeew,1104
 swarms/agents/tool_agent.py,sha256=G7rhBACsHsGUMT4H9eF5aY7e3Gx-5jOmJkhCF1jm9mU,5087
@@ -106,7 +106,7 @@ swarms/schemas/mcp_schemas.py,sha256=XZJ4HyiY_cv8Gvj-53ddjzXuqT9hBU2f0cHbhIKs_jY
 swarms/schemas/swarms_api_schemas.py,sha256=uKqleW_7hNpqHi06yoba9jS2i9yzZp-SBV944MnkN68,6233
 swarms/schemas/tool_schema_base_model.py,sha256=0biTGIoibsPPP3fOrkC6WvNU5vXaalyccVKC1fpO_eg,1409
 swarms/structs/__init__.py,sha256=8_CG2mct9cS-TQlpKlC5-240kSjmoooy9bfYXRuB9oQ,4552
-swarms/structs/agent.py,sha256=3SL7DNr6W2WuHsHlkSb7xoM0cjqim0JYD0pVgFspksM,119647
+swarms/structs/agent.py,sha256=hkFhZYZVjNrcHQv4nm5qkaRNHmjyf2LNsp5UZy1CXJk,127157
 swarms/structs/agent_builder.py,sha256=tYNpfO4_8cgfMHfgA5DAOWffHnt70p6CLt59esqfVCY,12133
 swarms/structs/agent_rag_handler.py,sha256=g17YRrNmf16TLvyFCCcsitVk3d-QNZmck_XYmjSN_YM,21372
 swarms/structs/agent_registry.py,sha256=il507cO1NF-d4ChyANVLuWrN8bXsEAi8_7bLJ_sTU6A,12112
@@ -119,7 +119,7 @@ swarms/structs/base_swarm.py,sha256=bO5olk16je7aJ_XmJxiD6y4nVlbvOySZpc5JX6Rdpvg,
 swarms/structs/base_workflow.py,sha256=DTfFwX3AdFYxACDYwUDqhsbcDZnITlg5TeEYyxmJBCc,11414
 swarms/structs/batch_agent_execution.py,sha256=d85DzeCq4uTbbPqLhAXFqFx_cxXUS5yRnJ1-gJkwU5w,1871
 swarms/structs/concat.py,sha256=utezSxNyh1mIwXgdf8-dJ803NDPyEy79WE8zJHuooGk,732
-swarms/structs/concurrent_workflow.py,sha256=SCyK-weYYpRb5EG3wZW4z7cF9isoIdpN7CvIVQaB7jg,8343
+swarms/structs/concurrent_workflow.py,sha256=0Yo_eMKxdbLfqkVmEJpyL0_cjMUH5NrBu-r-EWltMwU,14477
 swarms/structs/conversation.py,sha256=mmsvR15zWxsJ4dgyxwwunIifvfcVO4gfJr5zl0RxaSA,52258
 swarms/structs/council_judge.py,sha256=siYDKiHMvFmShUTXxdo4R6vXiQhKt7bEBI205oC3kU4,19639
 swarms/structs/csv_to_agent.py,sha256=Zv41sjeWA50msq-paGHESzlxZyMU78DYDLNNKZtNfoI,11125
@@ -131,10 +131,10 @@ swarms/structs/groupchat.py,sha256=jjH0BqU9Nrd_3jl9QzrcvbSce527SFpUaepawaRiw2o,1
 swarms/structs/hiearchical_swarm.py,sha256=6JwYWlsmwsVtOr2xlzl2q0IRsGvr-RFzDG9ClIJWTQo,33400
 swarms/structs/hybrid_hiearchical_peer_swarm.py,sha256=0BrmzSVit-I_04DFfrs7onLblLA6PSPa0JE3-4j05FA,9316
 swarms/structs/image_batch_processor.py,sha256=31Z8vTVL4dw18QxGwb0Jg1nvp0YzX8lwVgGj_-KrkhY,8207
-swarms/structs/interactive_groupchat.py,sha256=tlI4BiDRKQK2YtHvbWvT-Z0rhfpfsywavj1CHfZc7F8,41169
+swarms/structs/interactive_groupchat.py,sha256=zgumadD7YCFAqzNrh9CKMSWKe66aBkohIxPgQs3IU2k,40370
 swarms/structs/long_agent.py,sha256=KFjE2uUI8ONTkeJO43Sh3y5-Ec0kga28BECGklic-S4,15049
 swarms/structs/ma_blocks.py,sha256=04dF3DOSHXxNwrcl9QUloKgNDkI9yLRv6UbyyiKsIb0,4551
-swarms/structs/ma_utils.py,sha256=XJYxqWxFKBmdDekdX_BVGmmBxLbnz7hURMS2ZpVIcK8,3854
+swarms/structs/ma_utils.py,sha256=-7wTLVq9PHbXxt3QP9ngFA4mHV8H4dIOmhrXml4Iwcc,5933
 swarms/structs/majority_voting.py,sha256=7RGMTNX-ulH5McV0LPkpGEOkwYgsBdzTzHk5G_g_YUY,10120
 swarms/structs/malt.py,sha256=uLofKBWHkP3uNhyCkkgEyE4Z7qnOHTtcg-OTiR19x_Y,19572
 swarms/structs/matrix_swarm.py,sha256=qHuhOYrTyOv6ujHMe8PrQT-h-WmaCPCfX4ghv5L8UFI,9765
@@ -192,7 +192,7 @@ swarms/utils/check_all_model_max_tokens.py,sha256=ZHIKlrU-L-OM2IJAbYkCoVyBKe2d0J
 swarms/utils/data_to_text.py,sha256=1PUoWokylp7MOrGNk1cmO3cJlfskdAIiImGk9ECwsKU,3427
 swarms/utils/disable_logging.py,sha256=KKPKQVfQqLPFgj03uveOoyeHOTlfEJt-yfLc3SA53Rk,2470
 swarms/utils/file_processing.py,sha256=QjQCIPTcwicQlfy656BXBYpIzMR0s2343E7ftnok5Uo,4865
-swarms/utils/formatter.py,sha256=PDtaj93jwySIaGy4kO7nEoHSAEaNs5CRiM6sKOBZEME,8813
+swarms/utils/formatter.py,sha256=IpYrGlGfcBX1jGrsxqIDhO4szrMbk0jQqtKrTGZbMQY,14724
 swarms/utils/function_caller_model.py,sha256=ZfgCMzOizNnuZipYLclTziECNHszH9p8RQcUq7VNr4Q,4156
 swarms/utils/generate_keys.py,sha256=o5zp_8rwu5sgQnItWS1xAuIIRIkahwm02qy1vsV6DSQ,997
 swarms/utils/history_output_formatter.py,sha256=whjfk4N5SjMo3mtrafNny_alNiAhQcGza3l1dCyg7Nw,1388
@@ -208,8 +208,8 @@ swarms/utils/str_to_dict.py,sha256=T3Jsdjz87WIlkSo7jAW6BB80sv0Ns49WT1qXlOrdEoE,8
 swarms/utils/try_except_wrapper.py,sha256=uvDZDZJcH986EF0Ej6zZBLcqHJ58NHizPsAH5olrE7Q,3919
 swarms/utils/vllm_wrapper.py,sha256=sNkm4EbeMrqqmHidnvq5zTnofQAaARy3HIrNBu11lKs,5072
 swarms/utils/xml_utils.py,sha256=D4nEdo1nkHqSoTKrWylXBXjcHFhGaOYvvfGNQQoYV5o,2514
-swarms-7.9.3.dist-info/LICENSE,sha256=jwRtEmTWjLrEsvFB6QFdYs2cEeZPRMdj-UMOFkPF8_0,11363
-swarms-7.9.3.dist-info/METADATA,sha256=mWniYYlqAwYjIBrUriqn4yGxMezGA7h8zPUm-8VnntY,32138
-swarms-7.9.3.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-swarms-7.9.3.dist-info/entry_points.txt,sha256=2K0rTtfO1X1WaO-waJlXIKw5Voa_EpAL_yU0HXE2Jgc,47
-swarms-7.9.3.dist-info/RECORD,,
+swarms-7.9.4.dist-info/LICENSE,sha256=jwRtEmTWjLrEsvFB6QFdYs2cEeZPRMdj-UMOFkPF8_0,11363
+swarms-7.9.4.dist-info/METADATA,sha256=8jzaBd4MWGhsDSfrmyblph_Qa5-Pz5_F3EG1QCmnNAc,32138
+swarms-7.9.4.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+swarms-7.9.4.dist-info/entry_points.txt,sha256=2K0rTtfO1X1WaO-waJlXIKw5Voa_EpAL_yU0HXE2Jgc,47
+swarms-7.9.4.dist-info/RECORD,,