swarms 7.7.3__py3-none-any.whl → 7.7.5__py3-none-any.whl

swarms/structs/hiearchical_swarm.py

@@ -251,7 +251,7 @@ class HierarchicalSwarm(BaseSwarm):
         director: Optional[Union[Agent, Any]] = None,
         agents: List[Union[Agent, Any]] = None,
         max_loops: int = 1,
-        output_type: OutputType = "dict",
+        output_type: OutputType = "dict-all-except-first",
         director_model_name: str = "gpt-4o",
         teams: Optional[List[TeamUnit]] = None,
         inter_agent_loops: int = 1,

swarms/structs/hybrid_hiearchical_peer_swarm.py

@@ -1,5 +1,5 @@
 import os
-from typing import List
+from typing import List, Literal
 from swarms.structs.agent import Agent
 from swarms.structs.conversation import Conversation
 from swarms.structs.multi_agent_exec import get_swarms_info
@@ -10,6 +10,23 @@ from swarms.utils.history_output_formatter import (
 from concurrent.futures import ThreadPoolExecutor, as_completed
 from typing import Union, Callable
 
+
+HistoryOutputType = Literal[
+    "list",
+    "dict",
+    "dictionary",
+    "string",
+    "str",
+    "final",
+    "last",
+    "json",
+    "all",
+    "yaml",
+    # "dict-final",
+    "dict-all-except-first",
+    "str-all-except-first",
+]
+
 tools = [
     {
         "type": "function",
@@ -105,7 +122,7 @@ class HybridHierarchicalClusterSwarm:
         description: str = "A swarm that uses a hybrid hierarchical-peer model to solve complex tasks.",
         swarms: List[Union[SwarmRouter, Callable]] = [],
         max_loops: int = 1,
-        output_type: str = "list",
+        output_type: HistoryOutputType = "list",
         router_agent_model_name: str = "gpt-4o-mini",
         *args,
         **kwargs,
@@ -121,7 +138,7 @@ class HybridHierarchicalClusterSwarm:
         self.router_agent = Agent(
             agent_name="Router Agent",
             agent_description="A router agent that routes tasks to the appropriate swarms.",
-            system_prompt=f"{router_system_prompt}\n\n{get_swarms_info()}",
+            system_prompt=f"{router_system_prompt}\n\n{get_swarms_info(swarms=self.swarms)}",
             tools_list_dictionary=tools,
             model_name=router_agent_model_name,
             max_loops=1,