jaf-py 2.5.9__py3-none-any.whl → 2.5.11__py3-none-any.whl

jaf/visualization/functional_core.py

@@ -2,7 +2,7 @@
 JAF Visualization - Functional Core
 
 Pure functional operations for graph creation that separate data transformation
-from the imperative Graphviz operations. This follows the functional core, 
+from the imperative Graphviz operations. This follows the functional core,
 imperative shell pattern.
 """
 
@@ -14,9 +14,11 @@ from .types import GraphOptions
 
 # ========== Pure Data Structures ==========
 
+
 @dataclass(frozen=True)
 class NodeSpec:
     """Immutable specification for a graph node."""
+
     id: str
     label: str
     attributes: Dict[str, str]
@@ -25,6 +27,7 @@ class NodeSpec:
 @dataclass(frozen=True)
 class EdgeSpec:
     """Immutable specification for a graph edge."""
+
     from_node: str
     to_node: str
     attributes: Dict[str, str]
@@ -33,6 +36,7 @@ class EdgeSpec:
 @dataclass(frozen=True)
 class GraphSpec:
     """Immutable specification for a complete graph."""
+
     title: str
     graph_attributes: Dict[str, str]
     nodes: Tuple[NodeSpec, ...]
@@ -41,22 +45,23 @@ class GraphSpec:
 
 # ========== Pure Functions ==========
 
+
 def create_graph_attributes(options: GraphOptions) -> Dict[str, str]:
     """Pure function to create graph attributes from options."""
     return {
-        'rankdir': options.rankdir,
-        'label': options.title or '',
-        'labelloc': 't',
-        'fontsize': '16',
-        'fontname': 'Arial Bold',
-        'bgcolor': 'white',
-        'pad': '0.5'
+        "rankdir": options.rankdir,
+        "label": options.title or "",
+        "labelloc": "t",
+        "fontsize": "16",
+        "fontname": "Arial Bold",
+        "bgcolor": "white",
+        "pad": "0.5",
     }
 
 
 def create_agent_label(agent: Agent, show_tool_details: bool) -> str:
     """Pure function to create agent label."""
-    model_name = getattr(agent.model_config, 'name', 'default') if agent.model_config else 'default'
+    model_name = getattr(agent.model_config, "name", "default") if agent.model_config else "default"
     label = f"{agent.name}\\n({model_name})"
 
     if show_tool_details and agent.tools:
@@ -78,98 +83,70 @@ def create_tool_label(tool: Tool) -> str:
 
 
 def create_agent_node_spec(
-    agent: Agent,
-    styles: Dict[str, Any],
-    show_tool_details: bool
+    agent: Agent, styles: Dict[str, Any], show_tool_details: bool
 ) -> NodeSpec:
     """Pure function to create agent node specification."""
-    agent_style = styles['agent']
+    agent_style = styles["agent"]
     label = create_agent_label(agent, show_tool_details)
 
     attributes = {
-        'label': label,
-        'shape': agent_style['shape'],
-        'fillcolor': agent_style['fillcolor'],
-        'fontcolor': agent_style['fontcolor'],
-        'style': agent_style['style']
+        "label": label,
+        "shape": agent_style["shape"],
+        "fillcolor": agent_style["fillcolor"],
+        "fontcolor": agent_style["fontcolor"],
+        "style": agent_style["style"],
     }
 
-    if 'fontname' in agent_style:
-        attributes['fontname'] = agent_style['fontname']
+    if "fontname" in agent_style:
+        attributes["fontname"] = agent_style["fontname"]
 
-    return NodeSpec(
-        id=agent.name,
-        label=label,
-        attributes=attributes
-    )
+    return NodeSpec(id=agent.name, label=label, attributes=attributes)
 
 
 def create_tool_node_spec(tool: Tool, styles: Dict[str, Any]) -> NodeSpec:
     """Pure function to create tool node specification."""
-    tool_style = styles['tool']
+    tool_style = styles["tool"]
     label = create_tool_label(tool)
 
     attributes = {
-        'label': label,
-        'shape': tool_style['shape'],
-        'fillcolor': tool_style['fillcolor'],
-        'fontcolor': tool_style['fontcolor'],
-        'style': tool_style['style']
+        "label": label,
+        "shape": tool_style["shape"],
+        "fillcolor": tool_style["fillcolor"],
+        "fontcolor": tool_style["fontcolor"],
+        "style": tool_style["style"],
     }
 
-    if 'fontname' in tool_style:
-        attributes['fontname'] = tool_style['fontname']
+    if "fontname" in tool_style:
+        attributes["fontname"] = tool_style["fontname"]
 
-    return NodeSpec(
-        id=tool.schema.name,
-        label=label,
-        attributes=attributes
-    )
+    return NodeSpec(id=tool.schema.name, label=label, attributes=attributes)
 
 
-def create_tool_edge_spec(
-    agent_name: str,
-    tool_name: str,
-    styles: Dict[str, Any]
-) -> EdgeSpec:
+def create_tool_edge_spec(agent_name: str, tool_name: str, styles: Dict[str, Any]) -> EdgeSpec:
     """Pure function to create tool edge specification."""
     edge_attrs = {
-        'color': styles['tool_edge']['color'],
-        'style': styles['tool_edge']['style'],
-        'penwidth': styles['tool_edge']['penwidth']
+        "color": styles["tool_edge"]["color"],
+        "style": styles["tool_edge"]["style"],
+        "penwidth": styles["tool_edge"]["penwidth"],
     }
 
-    if 'arrowhead' in styles['tool_edge']:
-        edge_attrs['arrowhead'] = styles['tool_edge']['arrowhead']
+    if "arrowhead" in styles["tool_edge"]:
+        edge_attrs["arrowhead"] = styles["tool_edge"]["arrowhead"]
 
-    return EdgeSpec(
-        from_node=agent_name,
-        to_node=tool_name,
-        attributes=edge_attrs
-    )
+    return EdgeSpec(from_node=agent_name, to_node=tool_name, attributes=edge_attrs)
 
 
-def create_handoff_edge_spec(
-    from_agent: str,
-    to_agent: str,
-    styles: Dict[str, Any]
-) -> EdgeSpec:
+def create_handoff_edge_spec(from_agent: str, to_agent: str, styles: Dict[str, Any]) -> EdgeSpec:
     """Pure function to create handoff edge specification."""
     return EdgeSpec(
         from_node=from_agent,
         to_node=to_agent,
-        attributes={
-            'color': styles['edge']['color'],
-            'style': 'dashed',
-            'label': 'handoff'
-        }
+        attributes={"color": styles["edge"]["color"], "style": "dashed", "label": "handoff"},
     )
 
 
 def create_agent_graph_spec(
-    agents: List[Agent],
-    options: GraphOptions,
-    styles: Dict[str, Any]
+    agents: List[Agent], options: GraphOptions, styles: Dict[str, Any]
 ) -> GraphSpec:
     """Pure function to create complete agent graph specification."""
     nodes = []
@@ -200,14 +177,12 @@ def create_agent_graph_spec(
         title=options.title or "JAF Graph",
         graph_attributes=create_graph_attributes(options),
         nodes=tuple(nodes),
-        edges=tuple(edges)
+        edges=tuple(edges),
     )
 
 
 def create_tool_graph_spec(
-    tools: List[Tool],
-    options: GraphOptions,
-    styles: Dict[str, Any]
+    tools: List[Tool], options: GraphOptions, styles: Dict[str, Any]
 ) -> GraphSpec:
     """Pure function to create tool graph specification."""
     nodes = []
@@ -220,5 +195,5 @@ def create_tool_graph_spec(
         title=options.title or "JAF Tool Graph",
         graph_attributes=create_graph_attributes(options),
         nodes=tuple(nodes),
-        edges=tuple()  # No edges in pure tool graph
+        edges=tuple(),  # No edges in pure tool graph
     )