} iddlmZddlmZmZmZddlmZddlm Z ddl m Z ddl m Z mZddlmZdZGd d e ZGd d e ZeezZd d d dddd'dZdd d(d!Zd)d&Zd S)*) annotations)AnyLiteralcast)uuid4) AnyMessage) TypedDict) get_configget_stream_writer)CONF) UIMessageRemoveUIMessage AnyUIMessagepush_ui_messagedelete_ui_messageui_message_reducercFeZdZUdZded<ded<ded<ded<ded <d S) 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__C:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\langgraph/graph/ui.pyr r sT   GGG IIIr"r c(eZdZUdZded<ded<dS)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+s1 GGGGGr"rNuiF)rrmessage state_keymergerrrrr str | Nonerdict[str, Any] | Noner&AnyMessage | Noner'r(boolreturnc ddlm}t}t} d} |rFt |t rd|vr|d} nt|dr|j} d|ptt|||| dd| dd| ddd |pi| 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(r1r2r message_id)rrrrr) langgraph._internal._constantsr0r r isinstancedictgethasattrrrrr ) rrrrr&r'r(r0writerconfigr4evts r#rr=sKL?>>>>>  F \\FJ$ gt $ $ $ T**JJ Wd # # $ J CLLjj400JJvt,,JJz400   ~2  .8? j))R   C F3KKK:%t _% 3'7&8999 Jr")r'cddlm}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)r5r0r r r )rr'r0r:r;r<s r#rrsj&?>>>>>  F \\F$/r::C F3KKK!F4L!Is#3"4555 Jr"left!list[AnyUIMessage] | AnyUIMessagerightlist[AnyUIMessage]cvt|ts|g}t|ts|g}|}dt|D}t |D]:}|d}||x}|ddkr|^|tt|diddr1||}|}i|d |d |d <|||<|ddkrtd |d t|||<| |<fd |D}|S) 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"}, ) ``` c@i|]\}}|d|Srr8).0ims r# z&ui_message_reducer..s(AAAtq!AEE$KKAAAr"rNrr>rr(FrzCAttempting to delete an UI message with an ID that doesn't exist ('z')cBg|]}|dv|SrErF)rGrI ids_to_removes r# z&ui_message_reducer..s- D D DA155;;m#C#Ca#C#C#Cr") r6listcopy enumeratesetr8adddiscardrr ValueErrorlenappend) r?rAmerged merged_by_idmsgmsg_id existing_idxprev_msgrLs @r#rrs4 dD ! !v eT " "YY[[FAAy/@/@AAALEEM(,,V44 4L Awwv+--!!&))))%%f--- 3''++J;;??OOI%l3H((**C#Hhw&7#H3w<#HCL'*|$$wwv+-- dZ`ddd$'v;;L MM#     D D D D D D DF Mr")rrrrrr)rr*r&r+r'r)r(r,r-r )rrr'rr-r)r?r@rAr@r-rB) __future__rtypingrrruuidrlangchain_core.messagesrtyping_extensionsr langgraph.configr r langgraph.constantsr __all__r rrrrrr!r"r#rest""""""%%%%%%%%%%......''''''::::::::$$$$$$  *      i    ?* &*!% EEEEEEP48@>>>>>>r"