codekingpro/portable-devtools
115k
1�
2 ��j� � �� � S SK Jr S SKJrJrJr S SKJr S SKJ r S SK3J r S SK J
r
Jr S SKJr Sr " S S4\ 5 r " S S \ 5 r\\- rS
S
S
SSS. SS jjrSS.SS jjr SS jrg
)� )� annotations)�Any�Literal�cast)�uuid4)�5AnyMessage)� TypedDict)�6get_config�get_stream_writer)�CONF)� UIMessage�RemoveUIMessage� AnyUIMessage�push_ui_message�delete_ui_message�ui_message_reducerc �L � \ rS rSr% SrS\S' S\S' S\S' S\S ' S\S7' S rg )
r
� a� A message type for UI updates in LangGraph.8 9This TypedDict represents a UI message that can be sent to update the UI state.10It contains information about the UI component to render and its properties.11 12Attributes:13 type: Literal type indicating this is a UI message.14 id: Unique identifier for the UI message.15 name: Name of the UI component to render.16 props: Properties to pass to the UI component.17 metadata: Additional metadata about the UI message.18z
Literal['ui']�type�str�id�name�dict[str, Any]�props�metadata� N��__name__�19__module__� __qualname__�__firstlineno__�__doc__�__annotations__�__static_attributes__r � �XD:\code\apps\devtools\python\user_packages\Python313\site-packages\langgraph/graph/ui.pyr
r
s% � � � �� �G�20
�I� �� � r% r
c �. � \ rS rSr% SrS\S' S\S' Srg) r �+ a" A message type for removing UI components in LangGraph.21 22This TypedDict represents a message that can be sent to remove a UI component23from the current state.24 25Attributes:26 type: Literal type indicating this is a remove-ui message.27 id: Unique identifier of the UI message to remove.28zLiteral['remove-ui']r r r r Nr r r% r& r r + s � �� � � �Gr% r N�uiF)r r �message� state_key�mergec � � SSK Jn [ 5 n[ 5 n Sn29U( aJ [ U[30 5 ( a SU; a UR
S5 n31O [ US5 ( a UR n32SU=( d [ [ 5 5 U UUU R
SS5 U R
SS5 U R
SS5 S .U=( d 0 EU33( a S34U350O0 ES .n U" U 5 U( a U [ U " X[4/5 U $ ) am Push a new UI message to update the UI state.36 37This function creates and sends a UI message that will be rendered in the UI.38It also updates the graph state with the new UI message.39 40Args:41 name: Name of the UI component to render.42 props: Properties to pass to the UI component.43 id: Optional unique identifier for the UI message.44 If not provided, a random UUID will be generated.45 metadata: Optional additional metadata about the UI message.46 message: Optional message object to associate with the UI message.47 state_key: Key in the graph state where the UI messages are stored.48 merge: Whether to merge props with existing UI message (True) or replace49 them (False).50 51Returns:52 The created UI message.53 54Example:55 ```python56 push_ui_message(57 name="component-name",58 props={"content": "Hello world"},59 )60 ```61 62r ��CONFIG_KEY_SENDNr r) �run_id�tags�run_name)r, r0 r1 r �63message_id)r r r r r ) � langgraph._internal._constantsr/ r r64 �65isinstance�dict�get�hasattrr r r r ) r r r r r* r+ r, r/ �writer�configr3 �evts r& r r = s� � �L ?�
�
�F�
�\�F��J�� �g�t� $� $��� � ���T�*�J�
�W�d�
#�
#� ���J� �� �C���L���� �j�j��4�0��J�J�v�t�,��J�J�z�4�0� 66�67 �~�2� 68� .8� �j�)�R�
69�
�C� �3�K���t� �_�%� �'7�&8�9� �Jr% )r+ c �| � SSK Jn [ 5 n[ 5 nSU S.nU" U5 U[ U " X4/5 U$ )a� Delete a UI message by ID from the UI state.70 71This function creates and sends a message to remove a UI component from the current state.72It also updates the graph state to remove the UI message.73 74Args:75 id: Unique identifier of the UI component to remove.76 state_key: Key in the graph state where the UI messages are stored. Defaults to "ui".77 78Returns:79 The remove UI message.80 81Example:82 ```python83 delete_ui_message("message-123")84 ```85 86r r. � remove-ui)r r )r4 r/ r r87 r )r r+ r/ r9 r: r; s r& r r � sC � �& ?�
�
�F�
�\�F�$/�r�:�C�88�3�K�89�4�L� �!�I�#3�"4�5� �Jr% c �p � [ U [ 5 ( d U /n [ U[ 5 ( d U/nU R 5 n[ U5 VVs0 s H u p4UR S5 U_M nnn[ 5 nU GH nUR S5 nUR U5 =n b� UR S5 S:X a UR
U5 MQ UR U5 [ [ U5 R S0 5 R SS5 ( a# X) n90UR 5 n0 U91S EUS EUS' XrU ' M� UR S5 S:X a [ SU S 35 e[ U5 XX' UR U5 GM U Vs/ s H oDR S5 U; d M UPM nnU$ s snnf s snf )92a� Merge two lists of UI messages, supporting removing UI messages.93 94This function combines two lists of UI messages, handling both regular UI messages95and `remove-ui` messages. When a `remove-ui` message is encountered, it removes any96UI message with the matching ID from the current state.97 98Args:99 left: First list of UI messages or single UI message.100 right: Second list of UI messages or single UI message.101 102Returns:103 Combined list of UI messages with removals applied.104 105Example:106 ```python107 messages = ui_message_reducer(108 [{"type": "ui", "id": "1", "name": "Chat", "props": {}}],109 {"type": "remove-ui", "id": "1"},110 )111 ```112 113r r r= r r, Fr zCAttempting to delete an UI message with an ID that doesn't exist ('z'))
r5 �list�copy� enumerater7 �set�add�discardr r
�114ValueError�len�append) �left�right�merged�i�m� merged_by_id�
ids_to_remove�msg�msg_id� existing_idx�prev_msgs r& r r � s� � �4 �d�D� !� !��v�� �e�T� "� "���� �Y�Y�[�F�/8��/@�A�/@�t�q�A�E�E�$�K��N�/@�L�A��E�M������� �� (� ,� ,�V� 4� 4�L� A��w�w�v��+�-� �!�!�&�)� �%�%�f�-�� �3�'�+�+�J��;�?�?���O�O� %� 3�H� �(�(�*�C�#H�h�w�&7�#H�3�w�<�#H�C��L�'*�|�$��w�w�v��+�-� �Y�Z`�Ya�ac�d�� � $'�v�;�L� � �M�M�#� �- �0 �
D��A�5�5��;�m�#C�a��F�
D� �M��9 B��6 Es � F-�F3�#F3)r r r r r �115str | Noner zdict[str, Any] | Noner* zAnyMessage | Noner+ rS r, �bool�returnr
)r r r+ r rU r )rH �!list[AnyUIMessage] | AnyUIMessagerI rV rU zlist[AnyUIMessage])�116__future__r �typingr r r �uuidr �langchain_core.messagesr �typing_extensionsr �langgraph.configr117 r �langgraph.constantsr �__all__r
r r r r r r r% r& �<module>r_ s� �� "� %� %� � .� '� :� $� �� � � �* �i� � �?�*� � �&*�!%� ��E�118
�E� �E� � E�119 $� E� �
E� �E� �E� �E�P 48� �@>�120+�>� ,�>� �>r%