g8'VUddlmZddlZddlZddlmZmZddlm Z ddl m Z ddl m Z mZmZmZddlZddlmZmZddlmZmZmZd d lmZe rd d lmZd Zd Z edZ ddedddd ddZ ddZ! ddZ" ddZ#GddedZ$dZ%de&d<y) ) annotationsN)IterableSequence)Path)indent) TYPE_CHECKING AnnotatedAnyLiteral)GeLe) TypeAlias TypedDictUnpack)BaseNode)Graph) NodeIdentDEFAULT_HIGHLIGHT_CSS generate_code MermaidConfig request_image save_imageStateDiagramDirectionz fill:#fdff32)TBLRRLBTT) start_nodehighlighted_nodes highlight_csstitle edge_labelsnotesctt|xsd}|D]} | |jvstd| dg} |rdd|dg} | j d|| j d||jj D]e\} } | |vr| j d| | j r)|jD]} | j d | d | n[| jj D]>\} } d | d | }|r| jr|d | jz }| j |@| jx}r7d | d }|r|jr|d |jz }| j ||s| js| j d | tjdd| j}| j t|d| j dh|rg| j d| j d|t|D]4} | |jvrtd| d| j d| d6dj| S)adGenerate [Mermaid state diagram](https://mermaid.js.org/syntax/stateDiagram.html) code for a graph. Args: graph: The graph to generate the image for. start_node: Identifiers of nodes that start the graph. highlighted_nodes: Identifiers of nodes to highlight. highlight_css: CSS to use for highlighting nodes. title: The title of the diagram. edge_labels: Whether to include edge labels in the diagram. notes: Whether to include notes in the diagram. direction: The direction of flow. Returns: The Mermaid code for the graph. z Start node "z" is not in the graph.z---ztitle: zstateDiagram-v2z direction z [*] --> z z --> z: z --> [*]z note right of z {2,} z z end notezclassDef highlighted zHighlighted node "zclass z highlighted)set _node_ids node_defs LookupErrorappenditemsreturns_base_nodenext_node_edgeslabelend_edgenoteresubrjoin)graphrr r!r"r#r$ directionstart_node_idsnode_idlinesnode_def next_node_idedgeliner2 clean_docss K/var/www/openai/venv/lib/python3.12/site-packages/pydantic_graph/mermaid.pyrr)se8:#345N! %// ) WI5KLM M"E '%)51 LL"# |I;/0"__224 n $ LL:gY/ 0  % % %  r'% ~>?!0'/&>&>&D&D&F" dG9E,84::b --D T" 'G  (( (8 (y)Dx~~"X^^,-- LL  X]] LL+G95 6$ >J LL F3 4 LL &154 R ,]O<= !23Geoo-!$6wi?U"VWW LL6',7 84 99U c#Kt|tr|f}nt|tr|}n|f}|D])}t|tr||j+yw)z5Get the node IDs from a sequence of node identifiers.N) isinstancestrrget_id) node_idents node_iternodes rAr*r*usR+s# N K *  N  dC J++-  sAAc t||jd|jd|jdt|jd|jdd|jdd|jd }tj|j j }i}|jd d k(rJd |}|jd rd|d<|jdrd|d<|jdx}r=||d<n7|jd dk(rd|}nd|}|jd x}r||d<|jdx}r||d<|jdx} r| |d<|jdx} r| |d<|jdx} r| |d<|jdx} r| |d<|jdxstj} | j||}|js;tj|jd |j|j|!|jS)"a Generate an image of a Mermaid diagram using [mermaid.ink](https://mermaid.ink). Args: graph: The graph to generate the image for. **kwargs: Additional parameters to configure mermaid chart generation. Returns: The image data. rr r!r"r#Tr$r8)rr r!r"r#r$r8 image_typepdfzhttps://mermaid.ink/pdf/pdf_fitr(fit pdf_landscape landscape pdf_paperpapersvgzhttps://mermaid.ink/svg/zhttps://mermaid.ink/img/typebackground_colorbgColorthemewidthheightscale httpx_client)paramsz error generating image: )requestresponse)rgetrbase64 b64encodeencodedecodehttpxClient is_successHTTPStatusError status_codetextr]content)r7kwargscode code_base64r\urlrQrKrUrWrXrYrZr[r^s rArrsQ  ::l+ **%89jj2GHjj!JJ}d3jj$'**[) D""4;;=188:K%'F zz,5(( 6 ::i F5M ::o &"$F;  ;/ /9 /'F7O L !U *( 6( 6L1 1: 1'F6N!::&8999,y 7##u#w 7##u#wH%%v%!x 7##u#w::n-?LF3H   ####$$>x}}o N$$    rBc t|tr t|}d|vr&|jj dd}|dvr||d<t |fi|}|j |y)a,Generate an image of a Mermaid diagram using [mermaid.ink](https://mermaid.ink) and save it to a local file. Args: path: The path to save the image to. graph: The graph to generate the image for. **kwargs: Additional parameters to configure mermaid chart generation. rKrN)pngwebprSrL)rDrErsuffixlowerr write_bytes)pathr7rkext image_datas rArrsh$Dz6!kk!!"% / /#&F< u//JZ rBceZdZUdZded< ded< ded< ded< d ed < d ed < d ed < d ed< d ed< ded< ded< ded< ded< ded< ded< ded< ded<y)rz1Parameters to configure mermaid chart generation.Sequence[NodeIdent] | NodeIdentrr rEr! str | Noner"boolr#r$z,Literal['jpeg', 'png', 'webp', 'svg', 'pdf']rKrMrOzYLiteral['letter', 'legal', 'tabloid', 'ledger', 'a0', 'a1', 'a2', 'a3', 'a4', 'a5', 'a6']rQrUz/Literal['default', 'neutral', 'dark', 'forest']rWintrXrYzAnnotated[float, Ge(1), Le(3)]rZz httpx.Clientr[rr8N)__name__ __module__ __qualname____doc____annotations__r&rBrArrs;//466,, #8 KM<<W MRihA ;:: J# K$ ))K$$-rBrF)totalz=type[BaseNode[Any, Any, Any]] | BaseNode[Any, Any, Any] | strrr)r7Graph[Any, Any, Any]r&Sequence[NodeIdent] | NodeIdent | Noner rr!rEr"rzr#r{r$r{r8zStateDiagramDirection | NonereturnrE)rGryrz Iterable[str])r7rrkUnpack[MermaidConfig]rbytes)ruz Path | strr7rrkrrNone)' __future__r _annotationsr`r4collections.abcrrpathlibrtextwraprtypingrr r r rdannotated_typesr r typing_extensionsrrrnodesrr7r__all__rrrr*rrrrrr&rBrArsB2 .99 ":: '4 67:>@D.I I7 I > I  I II I,I IX  > >$> >B! ! !$ !  !4/.IU/.dW 9VrB