praisonaiagents 0.0.150__py3-none-any.whl → 0.0.152__py3-none-any.whl

praisonaiagents/__init__.py

@@ -32,6 +32,12 @@ from .session import Session
 from .memory.memory import Memory
 from .guardrails import GuardrailResult, LLMGuardrail
 from .agent.handoff import Handoff, handoff, handoff_filters, RECOMMENDED_PROMPT_PREFIX, prompt_with_handoff_instructions
+# Flow display
+try:
+    from .flow_display import FlowDisplay, track_workflow
+except ImportError:
+    FlowDisplay = None
+    track_workflow = None
 from .main import (
     TaskOutput,
     ReflectionOutput,
@@ -135,4 +141,9 @@ __all__ = [
 
 # Add MCP to __all__ if available
 if _mcp_available:
-    __all__.append('MCP')
+    __all__.append('MCP')
+    
+# Add flow display if available
+if FlowDisplay is not None:
+    __all__.extend(['FlowDisplay', 'track_workflow'])
+

praisonaiagents/flow_display.py

@@ -0,0 +1,194 @@
+"""
+Flow Display for PraisonAI Agents
+
+Visual display with agents in center and tools on sides.
+"""
+
+from typing import Dict, List, Set, Tuple
+from collections import defaultdict
+import threading
+from rich.console import Console
+from rich.panel import Panel
+from rich.text import Text
+from rich.align import Align
+from rich.columns import Columns
+from rich.table import Table
+from rich import box
+
+class FlowDisplay:
+    """Displays agent workflow with agents centered and tools on sides."""
+    
+    def __init__(self):
+        self.console = Console()
+        self.agents = []  # List of agents in order
+        self.agent_tools = defaultdict(list)  # agent -> [tools]
+        self.tracking = False
+        self.lock = threading.Lock()
+        
+    def start(self):
+        """Start tracking workflow."""
+        self.tracking = True
+        self.agents.clear()
+        self.agent_tools.clear()
+        
+        # Register callbacks
+        try:
+            from praisonaiagents.main import register_display_callback
+            
+            def on_interaction(**kwargs):
+                if self.tracking:
+                    agent = kwargs.get('agent_name', 'Unknown')
+                    self._add_agent(agent)
+                    
+            def on_tool_call(message, **kwargs):
+                if self.tracking and "called function" in message:
+                    parts = message.split("'")
+                    if len(parts) > 1:
+                        tool_name = parts[1]
+                        agent = kwargs.get('agent_name', 'Unknown')
+                        self._add_tool(agent, tool_name)
+                        
+            register_display_callback('interaction', on_interaction)
+            register_display_callback('tool_call', on_tool_call)
+            
+        except ImportError:
+            pass
+            
+    def stop(self):
+        """Stop tracking and display the flow."""
+        self.tracking = False
+        self.display()
+        
+    def _add_agent(self, name: str):
+        """Add an agent if not already present."""
+        with self.lock:
+            if name not in self.agents:
+                self.agents.append(name)
+                
+    def _add_tool(self, agent_name: str, tool_name: str):
+        """Add a tool to an agent."""
+        with self.lock:
+            if agent_name not in self.agents:
+                self.agents.append(agent_name)
+            if tool_name not in self.agent_tools[agent_name]:
+                self.agent_tools[agent_name].append(tool_name)
+                
+    def display(self):
+        """Display the flow chart with agents in center and tools on sides."""
+        if not self.agents:
+            return
+            
+        self.console.print("\n[bold cyan]🔄 Agent Workflow Flow[/bold cyan]\n")
+        
+        # Display start
+        self._display_centered_node("── start ──", "grey35")
+        self._display_arrow_down()
+        
+        # Display each agent with their tools
+        for i, agent in enumerate(self.agents):
+            self._display_agent_with_tools(agent)
+            
+            # Add arrow to next agent or end
+            if i < len(self.agents) - 1:
+                self._display_arrow_down()
+        
+        # Display end
+        self._display_arrow_down()
+        self._display_centered_node("── end ──", "grey35")
+        
+    def _display_agent_with_tools(self, agent: str):
+        """Display agent with tools on the sides."""
+        tools = self.agent_tools.get(agent, [])
+        
+        if not tools:
+            # No tools - just agent
+            self._display_centered_node(agent, "purple")
+        else:
+            # Split tools between left and right
+            left_tools = tools[::2]  # Even indices
+            right_tools = tools[1::2]  # Odd indices
+            
+            # Create the layout
+            table = Table(show_header=False, show_edge=False, box=None, padding=0)
+            table.add_column(justify="center", min_width=20)  # Left tools
+            table.add_column(justify="center", min_width=5)   # Space
+            table.add_column(justify="center", min_width=20)  # Agent
+            table.add_column(justify="center", min_width=5)   # Space
+            table.add_column(justify="center", min_width=20)  # Right tools
+            
+            # Create panels
+            left_panel = self._create_tools_panel(left_tools) if left_tools else ""
+            agent_panel = Panel(
+                Text(agent, style="white on purple", justify="center"),
+                style="white on purple",
+                box=box.ROUNDED,
+                padding=(0, 2)
+            )
+            right_panel = self._create_tools_panel(right_tools) if right_tools else ""
+            
+            # Add row
+            table.add_row(left_panel, "", agent_panel, "", right_panel)
+            
+            # Display centered
+            self.console.print(Align.center(table))
+            
+            # Show arrows
+            if left_tools or right_tools:
+                arrow_parts = []
+                if left_tools:
+                    arrow_parts.append("←→")
+                else:
+                    arrow_parts.append("  ")
+                    
+                arrow_parts.append("        ")  # Center space
+                
+                if right_tools:
+                    arrow_parts.append("←→")
+                else:
+                    arrow_parts.append("  ")
+                    
+                self.console.print(Align.center(Text("".join(arrow_parts))))
+                
+    def _create_tools_panel(self, tools: List[str]) -> Panel:
+        """Create a panel for tools."""
+        if not tools:
+            return ""
+            
+        if len(tools) == 1:
+            return Panel(
+                Text(tools[0], style="black on yellow", justify="center"),
+                style="black on yellow",
+                box=box.ROUNDED,
+                padding=(0, 1)
+            )
+        else:
+            # Multiple tools
+            content = "\n".join(tools)
+            return Panel(
+                Text(content, style="black on yellow", justify="center"),
+                style="black on yellow",
+                box=box.ROUNDED,
+                padding=(0, 1)
+            )
+            
+    def _display_centered_node(self, label: str, color: str):
+        """Display a centered node."""
+        panel = Panel(
+            Text(label, style=f"white on {color}", justify="center"),
+            style=f"white on {color}",
+            box=box.ROUNDED,
+            padding=(0, 2)
+        )
+        self.console.print(Align.center(panel))
+        
+    def _display_arrow_down(self):
+        """Display a downward arrow."""
+        self.console.print()
+        self.console.print(Align.center("↓"))
+        self.console.print()
+
+
+# Simple function to create and use
+def track_workflow():
+    """Create a flow display tracker."""
+    return FlowDisplay()

praisonaiagents/llm/llm.py

@@ -1456,6 +1456,8 @@ class LLM:
                         )
                         if should_break:
                             final_response_text = tool_summary_text
+                            # Reset interaction_displayed to ensure final summary is shown
+                            interaction_displayed = False
                             break
                         elif tool_summary_text is None and iteration_count > self.OLLAMA_SUMMARY_ITERATION_THRESHOLD:
                             # Continue iteration after adding final answer prompt
@@ -1485,6 +1487,8 @@ class LLM:
                             tool_summary = self._generate_ollama_tool_summary(accumulated_tool_results, response_text)
                             if tool_summary:
                                 final_response_text = tool_summary
+                                # Reset interaction_displayed to ensure final summary is shown
+                                interaction_displayed = False
                                 break
                         
                         # If we've executed tools in previous iterations, this response contains the final answer
@@ -2567,6 +2571,8 @@ Output MUST be JSON with 'reflection' and 'satisfactory'.
                     )
                     if should_break:
                         final_response_text = tool_summary_text
+                        # Reset interaction_displayed to ensure final summary is shown
+                        interaction_displayed = False
                         break
                     elif tool_summary_text is None and iteration_count > self.OLLAMA_SUMMARY_ITERATION_THRESHOLD:
                         # Continue iteration after adding final answer prompt
@@ -2594,6 +2600,8 @@ Output MUST be JSON with 'reflection' and 'satisfactory'.
                         tool_summary = self._generate_ollama_tool_summary(accumulated_tool_results, response_text)
                         if tool_summary:
                             final_response_text = tool_summary
+                            # Reset interaction_displayed to ensure final summary is shown
+                            interaction_displayed = False
                             break
                     
                     # If we've executed tools in previous iterations, this response contains the final answer

{praisonaiagents-0.0.150.dist-info → praisonaiagents-0.0.152.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: praisonaiagents
-Version: 0.0.150
+Version: 0.0.152
 Summary: Praison AI agents for completing complex tasks with Self Reflection Agents
 Author: Mervin Praison
 Requires-Python: >=3.10

{praisonaiagents-0.0.150.dist-info → praisonaiagents-0.0.152.dist-info}/RECORD

@@ -1,7 +1,8 @@
-praisonaiagents/__init__.py,sha256=pmoUafSn5f-ubjgY0rBBTw7N1wsRtg7X8-UGe477OH4,3619
+praisonaiagents/__init__.py,sha256=RJfNHR2k00o_U8W_YJb05uwemrvQ_7SmyNhjp__G29E,3887
 praisonaiagents/_logging.py,sha256=WfgUX6jo9hClpgHVKSGz8gqkna9DDNhPJBv-wjhcJoM,4648
 praisonaiagents/_warning_patch.py,sha256=FSLdw1SnA9b1PSxHWaRIcuG9IiIwO5JT6uo_m3CM0NI,2816
 praisonaiagents/approval.py,sha256=UJ4OhfihpFGR5CAaMphqpSvqdZCHi5w2MGw1MByZ1FQ,9813
+praisonaiagents/flow_display.py,sha256=E84J_H3h8L-AqL_F1JzEUInQYdjmIEuNL1LZr4__Hes,6935
 praisonaiagents/main.py,sha256=BqqskwUP-qrqCsHF9h_Wv99uS24ZeFTD8CaN6o6bnCE,17150
 praisonaiagents/session.py,sha256=FHWButPBaFGA4x1U_2gImroQChHnFy231_aAa_n5KOQ,20364
 praisonaiagents/agent/__init__.py,sha256=KBqW_augD-HcaV3FL88gUmhDCpwnSTavGENi7RqneTo,505
@@ -20,7 +21,7 @@ praisonaiagents/knowledge/__init__.py,sha256=xL1Eh-a3xsHyIcU4foOWF-JdWYIYBALJH9b
 praisonaiagents/knowledge/chunking.py,sha256=G6wyHa7_8V0_7VpnrrUXbEmUmptlT16ISJYaxmkSgmU,7678
 praisonaiagents/knowledge/knowledge.py,sha256=OzK81oA6sjk9nAUWphS7AkXxvalrv2AHB4FtHjzYgxI,30115
 praisonaiagents/llm/__init__.py,sha256=SqdU1pRqPrR6jZeWYyDeTvmZKCACywk0v4P0k5Fuowk,1107
-praisonaiagents/llm/llm.py,sha256=UQmnKyPQSnaxCIlUwJmHC4vjQfFvlZIg2w883qQTkTM,172825
+praisonaiagents/llm/llm.py,sha256=ChQQoBkfVoVaPg13qZ9S3rM0rhrQJjVMCuJv4irvTVY,173421
 praisonaiagents/llm/model_capabilities.py,sha256=cxOvZcjZ_PIEpUYKn3S2FMyypfOSfbGpx4vmV7Y5vhI,3967
 praisonaiagents/llm/model_router.py,sha256=Jy2pShlkLxqXF3quz-MRB3-6L9vaUSgUrf2YJs_Tsg0,13995
 praisonaiagents/llm/openai_client.py,sha256=3EVjIs3tnBNFDy_4ZxX9DJVq54kS0FMm38m5Gkpun7U,57234
@@ -64,7 +65,7 @@ praisonaiagents/tools/xml_tools.py,sha256=iYTMBEk5l3L3ryQ1fkUnNVYK-Nnua2Kx2S0dxN
 praisonaiagents/tools/yaml_tools.py,sha256=uogAZrhXV9O7xvspAtcTfpKSQYL2nlOTvCQXN94-G9A,14215
 praisonaiagents/tools/yfinance_tools.py,sha256=s2PBj_1v7oQnOobo2fDbQBACEHl61ftG4beG6Z979ZE,8529
 praisonaiagents/tools/train/data/generatecot.py,sha256=H6bNh-E2hqL5MW6kX3hqZ05g9ETKN2-kudSjiuU_SD8,19403
-praisonaiagents-0.0.150.dist-info/METADATA,sha256=5JM0pL2Ubiq3OFuq2JU268cFwP6ic553UU1LWVMUBho,2146
-praisonaiagents-0.0.150.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-praisonaiagents-0.0.150.dist-info/top_level.txt,sha256=_HsRddrJ23iDx5TTqVUVvXG2HeHBL5voshncAMDGjtA,16
-praisonaiagents-0.0.150.dist-info/RECORD,,
+praisonaiagents-0.0.152.dist-info/METADATA,sha256=a-HyisF8Pw70UqkSloPXYxFj9p1e4pe1BITREE9xlkU,2146
+praisonaiagents-0.0.152.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+praisonaiagents-0.0.152.dist-info/top_level.txt,sha256=_HsRddrJ23iDx5TTqVUVvXG2HeHBL5voshncAMDGjtA,16
+praisonaiagents-0.0.152.dist-info/RECORD,,