A'h]NddlmZmZmZmZmZddlmZddlm Z ddl m Z ddl m Z mZddlmZmZGdde ZGd d e ZeeefZd d d d d ddedeeefdeedeeeefdee dededefdZd ddededefdZdeeeefdeeeefdeefdZy ))AnyLiteralOptionalUnioncast)uuid4) AnyMessage) TypedDict)CONFCONFIG_KEY_SEND) get_configget_stream_writerc^eZdZUdZeded<eed<eed<eeefed<eeefed<y) UIMessageaA 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. uitypeidnamepropsmetadataN) __name__ __module__ __qualname____doc__r__annotations__strdictrY/home/kushmeetdev/Regenta/Chatbot/venv/lib/python3.12/site-packages/langgraph/graph/ui.pyrr s9  $- G I S>38nrrc,eZdZUdZeded<eed<y)RemoveUIMessagea: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. remove-uirrN)rrrrrrrrrr r"r" s +  Grr"NrF)rrmessage state_keymergerrrrr$r%r&returnc t}t}d} |r>t|trd|vr|j d} nt |dr |j } d|xstt||||j dd|j dd|j ddd|xsi| rd| inid } || |tt|| fg| S) awPush 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. Defaults to "ui". Returns: The created UI message. Example: .. code-block:: python push_ui_message( name="component-name", props={"content": "Hello world"}, ) Nrrrun_idtagsrun_name)r&r)r*r message_id)rrrrr) rr isinstancergethasattrrrrr r ) rrrrr$r%r&writerconfigr,evts r push_ui_messager32sL F \FJ gt $ T*J Wd # J CLjj40JJvt,JJz40  ~2  .8 j)R  C 3K!F4L!Is#3"45 Jr)r%czt}t}d|d}|||tt||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: .. code-block:: python delete_ui_message("message-123") r#)rr)rr r r )rr%r0r1r2s r delete_ui_messager5wsD( F \F$/r:C 3K!F4L!Is#3"45 Jrleftrightc8t|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: .. code-block:: python messages = ui_message_reducer( [{"type": "ui", "id": "1", "name": "Chat", "props": {}}], {"type": "remove-ui", "id": "1"} ) rrr#rr&FrzCAttempting to delete an UI message with an ID that doesn't exist ('z')) r-listcopy enumerater.setadddiscardrr ValueErrorlenappend) r6r7mergedim merged_by_id ids_to_removemsgmsg_id existing_idxprev_msgs r ui_message_reducerrKs6 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)typingrrrrruuidrlangchain_core.messagesr typing_extensionsr langgraph.constantsr r langgraph.utils.configr rrr" AnyUIMessagerrboolr3r5r9rKrrr rTs.66.'5@ *  i  Y/0 )-$(B B S>B  B tCH~& B j ! BB BBJ48#SO>? \"L0 1? l#\1 2? ,?r