9iddlmZddlmZmZmZddlmZddlm Z ddl m Z ddl m Z mZddlmZdZGd d e ZGd d e ZeezZd d d ddd ddZddddZ ddZy )) annotations)AnyLiteralcast)uuid4) AnyMessage) TypedDict) get_configget_stream_writer)CONF) UIMessageRemoveUIMessage AnyUIMessagepush_ui_messagedelete_ui_messageui_message_reducercDeZdZUdZded<ded<ded<ded<ded <y ) r aA message type for UI updates in LangGraph. This TypedDict represents a UI message that can be sent to update the UI state. It contains information about the UI component to render and its properties. Attributes: type: Literal type indicating this is a UI message. id: Unique identifier for the UI message. name: Name of the UI component to render. props: Properties to pass to the UI component. metadata: Additional metadata about the UI message. z Literal['ui']typestridnamedict[str, Any]propsmetadataN__name__ __module__ __qualname____doc____annotations__\/var/www/html/backtest/airagagent/rag_env/lib/python3.12/site-packages/langgraph/graph/ui.pyr r s%   G I r"r c&eZdZUdZded<ded<y)ra:A message type for removing UI components in LangGraph. This TypedDict represents a message that can be sent to remove a UI component from the current state. Attributes: type: Literal type indicating this is a remove-ui message. id: Unique identifier of the UI message to remove. zLiteral['remove-ui']rrrNrr!r"r#rr+s  Gr"rNuiF)rrmessage state_keymergec ddlm}t}t} d} |r>t |t rd|vr|j d} nt|dr |j} d|xstt|||| j dd| j dd| j ddd |xsi| rd | inid } || |r| t||| fg| S) aPush a new UI message to update the UI state. This function creates and sends a UI message that will be rendered in the UI. It also updates the graph state with the new UI message. Args: name: Name of the UI component to render. props: Properties to pass to the UI component. id: Optional unique identifier for the UI message. If not provided, a random UUID will be generated. metadata: Optional additional metadata about the UI message. message: Optional message object to associate with the UI message. state_key: Key in the graph state where the UI messages are stored. merge: Whether to merge props with existing UI message (True) or replace them (False). Returns: The created UI message. Example: ```python push_ui_message( name="component-name", props={"content": "Hello world"}, ) ``` rCONFIG_KEY_SENDNrr%run_idtagsrun_name)r(r,r-r message_id)rrrrr) langgraph._internal._constantsr+r r isinstancedictgethasattrrrrr ) rrrrr&r'r(r+writerconfigr/evts r#rr=sL?  F \FJ gt $ T*J Wd # J CLjj40JJvt,JJz40  ~2  .8 j)R  C 3K%t _% 3'7&89 Jr")r'c~ddlm}t}t}d|d}|||t|||fg|S)aDelete a UI message by ID from the UI state. This function creates and sends a message to remove a UI component from the current state. It also updates the graph state to remove the UI message. Args: id: Unique identifier of the UI component to remove. state_key: Key in the graph state where the UI messages are stored. Defaults to "ui". Returns: The remove UI message. Example: ```python delete_ui_message("message-123") ``` rr* remove-ui)rr)r0r+r r r )rr'r+r5r6r7s r#rrsG&?  F \F$/r:C 3K!F4L!Is#3"45 Jr"c8t|ts|g}t|ts|g}|j}t|Dcic]\}}|j d|}}}t }|D]}|j d}|j |x} |j ddk(r|j |M|j|tt|j dij ddr$|| } |j}i| d|d|d<||| <|j ddk(rtd|d t|||<|j||Dcgc]}|j d|vs|}}|Scc}}wcc}w) aMerge two lists of UI messages, supporting removing UI messages. This function combines two lists of UI messages, handling both regular UI messages and `remove-ui` messages. When a `remove-ui` message is encountered, it removes any UI message with the matching ID from the current state. Args: left: First list of UI messages or single UI message. right: Second list of UI messages or single UI message. Returns: Combined list of UI messages with removals applied. Example: ```python messages = ui_message_reducer( [{"type": "ui", "id": "1", "name": "Chat", "props": {}}], {"type": "remove-ui", "id": "1"}, ) ``` rrr9rr(FrzCAttempting to delete an UI message with an ID that doesn't exist ('z')) r1listcopy enumerater3setadddiscardrr ValueErrorlenappend) leftrightmergedim merged_by_id ids_to_removemsgmsg_id existing_idxprev_msgs r#rrs4 dD !v eT "YY[F/8/@Atq!AEE$KNALAEM(,,V4 4L Awwv+-!!&)%%f- 3'++J;??O%l3H((*C#Hhw&7#H3w<#HCL'*|$wwv+- YZ`Yaacd$'v;L MM# -0 DA155;m#Ca DF D M9B6EsF0F F)rrrrr str | Nonerzdict[str, Any] | Noner&zAnyMessage | Noner'rOr(boolreturnr )rrr'rrQr)rD!list[AnyUIMessage] | AnyUIMessagerErRrQzlist[AnyUIMessage]) __future__rtypingrrruuidrlangchain_core.messagesrtyping_extensionsr langgraph.configr r langgraph.constantsr __all__r rrrrrr!r"r#r[s"%%.':$  *  i  ?* &*!% E E E  E $ E  EE EEP48@> +> ,>>r"