o h@snddlmZddlZddlmZddlmZddlmZdd d Z ddddZ ddddZ ddddZ dS)) annotationsN)Agent)Handoff)ToolagentrreturnstrcCs6dg}|t||t||dd|S)z Generates the main graph structure in DOT format for the given agent. Args: agent (Agent): The agent for which the graph is to be generated. Returns: str: The DOT format string representing the graph. zq digraph G { graph [splines=true]; node [fontname="Arial"]; edge [penwidth=1.5]; })append get_all_nodes get_all_edgesjoin)rpartsr`/var/www/html/openai_agents/venv/lib/python3.10/site-packages/agents/extensions/visualization.pyget_main_graph s    rparent Agent | Nonevisitedset[str] | NonecCs$|durt}|j|vrdS||jg}|s+|d|d|jd|jd|jD]}|d|jd|jdq.|jD]}|d|jd|jdqB|jD]6}t|trk|d|j d|j d t|t r|j|vr|d|jd|jd |t |||qVd |S) z Recursively generates the nodes for the given agent and its handoffs in DOT format. Args: agent (Agent): The agent for which the nodes are to be generated. Returns: str: The DOT format string representing the nodes. Nr z"__start__" [label="__start__", shape=ellipse, style=filled, fillcolor=lightblue, width=0.5, height=0.3];"__end__" [label="__end__", shape=ellipse, style=filled, fillcolor=lightblue, width=0.5, height=0.3];"z " [label="zJ", shape=box, style=filled, fillcolor=lightyellow, width=1.5, height=0.8];zM", shape=ellipse, style=filled, fillcolor=lightgreen, width=0.5, height=0.3];zF", shape=box, style=filled, fillcolor=lightgrey, width=1, height=0.5];zY", shape=box, style=filled, style=rounded, fillcolor=lightyellow, width=1.5, height=0.8];) setnameaddr tools mcp_servershandoffs isinstancer agent_namerr rrrrrtool mcp_serverhandoffrrrr "sD          r c CsL|durt}|j|vrdS||jg}|s"|d|jd|jD]}|d|jd|jd|jd|jd q%|jD]}|d|jd|jd |jd|jd qA|jD]1}t|trr|d |jd|j dt|t r|d |jd|jd|t |||q]|jst|t s|d |jd d |S)aA Recursively generates the edges for the given agent and its handoffs in DOT format. Args: agent (Agent): The agent for which the edges are to be generated. parent (Agent, optional): The parent agent. Defaults to None. Returns: str: The DOT format string representing the edges. Nr z"__start__" -> "z";z "z" -> "z)" [style=dotted, penwidth=1.5]; "z" [style=dotted, penwidth=1.5];z)" [style=dashed, penwidth=1.5]; "z" [style=dashed, penwidth=1.5];z "rz" -> "__end__";)rrrr rrrrrrrr rrr rrrr cs\          r filename str | Nonegraphviz.SourcecCs*t|}t|}|r|j|ddd|S)aG Draws the graph for the given agent and optionally saves it as a PNG file. Args: agent (Agent): The agent for which the graph is to be drawn. filename (str): The name of the file to save the graph as a PNG. Returns: graphviz.Source: The graphviz Source object representing the graph. pngT)formatcleanup)rgraphvizSourcerender)rr$dot_codegraphrrr draw_graphs  r/)rrrr)NN)rrrrrrrr)N)rrr$r%rr&) __future__rr*agentsragents.handoffsr agents.toolrrr r r/rrrrs      B 1