o Zh@sddlmZddlmZddlmZmZmZmZm Z m Z er*ddl m Z ddl mZGdddeZGdd d e ZGd d d Zd S) ) annotations)Enum) TYPE_CHECKINGAnyDictList NamedTupleOptional)DeltaGenerator) SupportsStrc@seZdZdZdZdZdS) ChildTypezEnumerator of the child type.MARKDOWN EXCEPTIONN)__name__ __module__ __qualname____doc__r rrro/var/www/html/lang_env/lib/python3.10/site-packages/langchain_community/callbacks/streamlit/mutable_expander.pyr sr c@s*eZdZUdZded<ded<ded<dS) ChildRecordzChild record as a NamedTuple.r typeDict[str, Any]kwargsr dgN)rrrr__annotations__rrrrrs  rc@seZdZdZd9dd Zed:d d Zed;d dZdddZ d?dddd@d&d'Z dd(dAd+d,Z dBd1d2Z dCd5d6ZdDd7d8ZdS)EMutableExpanderzJStreamlit expander that can be renamed and dynamically expanded/collapsed.parent_containerr labelstrexpandedboolcCs0||_||_||_|j|||_g|_dS)a;Create a new MutableExpander. Parameters ---------- parent_container The `st.container` that the expander will be created inside. The expander transparently deletes and recreates its underlying `st.expander` instance when its label changes, and it uses `parent_container` to ensure it recreates this underlying expander in the same location onscreen. label The expander's initial label. expanded The expander's initial `expanded` value. N)_label _expandedempty_parent_cursorexpander _container_child_records)selfrrrrrr__init__s   zMutableExpander.__init__returncC|jS)zExpander's label string.)r!r(rrrr4zMutableExpander.labelcCr+)z6True if the expander was created with `expanded=True`.)r"r,rrrr9r-zMutableExpander.expandedNonecCs|j|_|jdS)zeRemove the container and its contents entirely. A cleared container can't be reused. N)r$r#r&r'clearr,rrrr/>s zMutableExpander.clearothercCs(|j}|D] }||j|jqdS)z]Append a copy of another MutableExpander's children to this MutableExpander. N)r'copy _create_childrr)r(r0Z other_recordsrecordrrr append_copyEs zMutableExpander.append_copyN) new_label new_expandedr5 Optional[str]r6Optional[bool]cCsz|dur|j}|dur|j}|j|kr|j|krdS||_||_|j|||_|j}g|_|D] }||j|jq0dS)z.Change the expander's label and expanded stateN) r!r"r$r%r&r'r2rr)r(r5r6Z prev_recordsr3rrrupdateMszMutableExpander.updateF)helpindexbodyr unsafe_allow_htmlr:r; Optional[int]intcCs<|||d}||jdi|}ttj||}|||S)z=Add a Markdown element to the container and return its index.)r<r=r:Nr)_get_dgmarkdownrr r _add_record)r(r<r=r:r;rnew_dgr3rrrrAes  zMutableExpander.markdown)r; exception BaseExceptioncCs8d|i}||jdi|}ttj||}|||S)z?Add an Exception element to the container and return its index.rDNr)r@rDrr rrB)r(rDr;rrCr3rrrrDss zMutableExpander.exceptionrr rrcCsJ|tjkr|jdi|dS|tjkr|jdi|dStd|)z(Create a new child with the given paramszUnexpected child type Nr)r r rArrD RuntimeError)r(rrrrrr2|s  zMutableExpander._create_childr3rcCs0|dur ||j|<|S|j|t|jdS)zAdd a ChildRecord to self._children. If `index` is specified, replace the existing record at that index. Otherwise, append the record to the end of the list. Return the index of the added record. N)r'appendlen)r(r3r;rrrrBs   zMutableExpander._add_recordcCsD|durd|krt|jksnJd||j|jS|jS)Nrz Bad index: )rIr'rr&)r(r;rrrr@s* zMutableExpander._get_dg)rr rrrr )r*r)r*r )r*r.)r0rr*r.)r5r7r6r8r*r.)F) r<r r=r r:r7r;r>r*r?)rDrEr;r>r*r?)rr rrr*r.)r3rr;r>r*r?)r;r>r*r )rrrrr)propertyrrr/r4r9rArDr2rBr@rrrrrs*       rN) __future__renumrtypingrrrrrr Zstreamlit.delta_generatorr Zstreamlit.type_utilr r rrrrrrs