} i2nddlmZddlZddlZddlmZmZddlmZddl m Z m Z m Z m Z ddlmZmZmZmZmZmZmZddlmZmZddlmZmZmZdd lmZdd lm Z d Z!e"eezZ#d Z$d*dZ%e%ddd+dZ&eddGddeZ'GddeZ(d,d"Z)dd#d-d)Z*dS).) annotationsN)CallableSequence)partial) AnnotatedAnyLiteralcast) AnyMessage BaseMessageBaseMessageChunkMessageLikeRepresentation RemoveMessageconvert_to_messagesmessage_chunk_to_message) TypedDict deprecated)CONFCONFIG_KEY_SENDNS_SEP) StateGraph)LangGraphDeprecatedSinceV10) add_messages MessagesState MessageGraphREMOVE_ALL_MESSAGES__remove_all__funcrreturn(Callable[[Messages, Messages], Messages]c d d fd }j|_ttttgtf|S) NleftMessages | Nonerightkwargsrr3Messages | Callable[[Messages, Messages], Messages]ct| | ||fi|S||d|rdndd}t|tfi|S)NzMMust specify non-null arguments for both 'left' and 'right'. Only received: 'r"r$z'.) ValueErrorr)r"r$r%msgrs C:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\langgraph/graph/message.py _add_messagesz,_add_messages_wrapper.._add_messages*s~   14e..v.. .  !2>(,9ff'>>> S// !4**6** *)NN)r"r#r$r#r%rrr&)__doc__r rMessages)rr+s` r*_add_messages_wrapperr/)sT?C + + + + + + +!LM (H-x78- H HHr,)formatr"r.r$r0"Literal['langchain-openai'] | Nonec d}t|ts|g}t|ts|g}dt|D}dt|D}|D].}|j%t t j|_/t|D]X\}}|j%t t j|_t|tr|jtkr|}Y| ||dzdS| }dt|D}t |D]}| |jx}Pt|tr |jN |j|||<nt|trtd|jdt!|||j<||Ȉ fd|D}|d krt%|}n|rd |d } t|  |S) aMerges two lists of messages, updating existing messages by ID. By default, this ensures the state is "append-only", unless the new message has the same ID as an existing message. Args: left: The base list of `Messages`. right: The list of `Messages` (or single `Message`) to merge into the base list. format: The format to return messages in. If `None` then `Messages` will be returned as is. If `langchain-openai` then `Messages` will be returned as `BaseMessage` objects with their contents formatted to match OpenAI message format, meaning contents can be string, `'text'` blocks, or `'image_url'` blocks and tool responses are returned as their own `ToolMessage` objects. !!! important "Requirement" Must have `langchain-core>=0.3.11` installed to use this feature. Returns: A new list of messages with the messages from `right` merged into `left`. If a message in `right` has the same ID as a message in `left`, the message from `right` will replace the message from `left`. Example: Basic usage ```python from langchain_core.messages import AIMessage, HumanMessage msgs1 = [HumanMessage(content="Hello", id="1")] msgs2 = [AIMessage(content="Hi there!", id="2")] add_messages(msgs1, msgs2) # [HumanMessage(content='Hello', id='1'), AIMessage(content='Hi there!', id='2')] ``` Example: Overwrite existing message ```python msgs1 = [HumanMessage(content="Hello", id="1")] msgs2 = [HumanMessage(content="Hello again", id="1")] add_messages(msgs1, msgs2) # [HumanMessage(content='Hello again', id='1')] ``` Example: Use in a StateGraph ```python from typing import Annotated from typing_extensions import TypedDict from langgraph.graph import StateGraph class State(TypedDict): messages: Annotated[list, add_messages] builder = StateGraph(State) builder.add_node("chatbot", lambda state: {"messages": [("assistant", "Hello")]}) builder.set_entry_point("chatbot") builder.set_finish_point("chatbot") graph = builder.compile() graph.invoke({}) # {'messages': [AIMessage(content='Hello', id=...)]} ``` Example: Use OpenAI message format ```python from typing import Annotated from typing_extensions import TypedDict from langgraph.graph import StateGraph, add_messages class State(TypedDict): messages: Annotated[list, add_messages(format="langchain-openai")] def chatbot_node(state: State) -> list: return { "messages": [ { "role": "user", "content": [ { "type": "text", "text": "Here's an image:", "cache_control": {"type": "ephemeral"}, }, { "type": "image", "source": { "type": "base64", "media_type": "image/jpeg", "data": "1234", }, }, ], }, ] } builder = StateGraph(State) builder.add_node("chatbot", chatbot_node) builder.set_entry_point("chatbot") builder.set_finish_point("chatbot") graph = builder.compile() graph.invoke({"messages": []}) # { # 'messages': [ # HumanMessage( # content=[ # {"type": "text", "text": "Here's an image:"}, # { # "type": "image_url", # "image_url": {"url": "data:image/jpeg;base64,1234"}, # }, # ], # ), # ] # } ``` NcRg|]$}ttt|%Srr r .0ms r* z add_messages..s;     !&6!:!:;;   r,cRg|]$}ttt|%Sr4r5r6s r*r9z add_messages..s;     !&6!:!:;;   r,c$i|] \}}|j|Sr4id)r7ir8s r* z add_messages..s :::1AD!:::r,z?Attempting to delete a message with an ID that doesn't exist ('z')c&g|] }|jv |Sr4r=)r7r8 ids_to_removes r*r9z add_messages..s% = = =A14}#<#struuiduuid4 enumeraterrcopysetgetadddiscardr(lenappend_format_messages) r"r$r0remove_all_idxr8idxmerged merged_by_id existing_idxr)rBs @r*rr<s~N dD ! !v eT " "  $T**   D  $U++   E %% 4<tz||$$ADE""!!Q 4<tz||$$AD a ' ' !AD4G,G,G N!^a'))**YY[[F:: &(9(9:::LEEM (,,QT22 2L ?!]++ )!!!$''''%%ad+++'(|$$!]++  ^VWVZ^^^"%VL  MM!     = = = = = = =F ###!&))  RfRRRoo Mr,z{MessageGraph is deprecated in langgraph 1.0.0, to be removed in 2.0.0. Please use StateGraph with a `messages` key instead.)categoryc$eZdZdZdfd ZxZS)raA StateGraph where every node receives a list of messages as input and returns one or more messages as output. MessageGraph is a subclass of StateGraph whose entire state is a single, append-only* list of messages. Each node in a MessageGraph takes a list of messages as input and returns zero or more messages as output. The `add_messages` function is used to merge the output messages from each node into the existing list of messages in the graph's state. Examples: ```pycon >>> from langgraph.graph.message import MessageGraph ... >>> builder = MessageGraph() >>> builder.add_node("chatbot", lambda state: [("assistant", "Hello!")]) >>> builder.set_entry_point("chatbot") >>> builder.set_finish_point("chatbot") >>> builder.compile().invoke([("user", "Hi there.")]) [HumanMessage(content="Hi there.", id='...'), AIMessage(content="Hello!", id='...')] ``` ```pycon >>> from langchain_core.messages import AIMessage, HumanMessage, ToolMessage >>> from langgraph.graph.message import MessageGraph ... >>> builder = MessageGraph() >>> builder.add_node( ... "chatbot", ... lambda state: [ ... AIMessage( ... content="Hello!", ... tool_calls=[{"name": "search", "id": "123", "args": {"query": "X"}}], ... ) ... ], ... ) >>> builder.add_node( ... "search", lambda state: [ToolMessage(content="Searching...", tool_call_id="123")] ... ) >>> builder.set_entry_point("chatbot") >>> builder.add_edge("chatbot", "search") >>> builder.set_finish_point("search") >>> builder.compile().invoke([HumanMessage(content="Hi there. Can you search for X?")]) {'messages': [HumanMessage(content="Hi there. Can you search for X?", id='b8b7d8f4-7f4d-4f4d-9c1d-f8b8d8f4d9c1'), AIMessage(content="Hello!", id='f4d9c1d8-8d8f-4d9c-b8b7-d8f4f4d9c1d8'), ToolMessage(content="Searching...", id='d8f4f4d9-c1d8-4f4d-b8b7-d8f4f4d9c1d8', tool_call_id="123")]} ``` rNonectjdtdtt t ttfdS)Nz}MessageGraph is deprecated in LangGraph v1.0.0, to be removed in v2.0.0. Please use StateGraph with a `messages` key instead.)rV stacklevel) warningswarnrsuper__init__rrDr r)self __class__s r*r_zMessageGraph.__init__*sW  L0    4 #3\#ABCCCCCr,)rrX)__name__ __module__ __qualname__r-r_ __classcell__)ras@r*rrsR ,,\DDDDDDDDDDr,rceZdZUded<dS)rz)Annotated[list[AnyMessage], add_messages]messagesN)rbrcrd__annotations__r4r,r*rr3s777777r,rrgSequence[BaseMessage]list[BaseMessage]c ddlm}t||S#t$r(d}t j|t |cYSwxYw)Nr)convert_to_openai_messageszMust have langchain-core>=0.3.11 installed to use automatic message formatting (format='langchain-openai'). Please update your langchain-core version or remove the 'format' flag. Returning un-formatted messages.)langchain_core.messagesrlr ImportErrorr\r]rD)rgrlr)s r*rPrP7s IFFFFFF##=#=h#G#GHHH    cH~~s /AA) state_keymessage,MessageLikeRepresentation | BaseMessageChunkro str | Noner c ddlm m}ddlm}ddlm |}tdt|gD}|j tdt|d|r|d}|j }nDt|dtr)t fd |dDr|d}t fd |Ddx}rb|d }tt!t"|d t&|f} || |d |r$|t*t,||fg|S)zWrite a message manually to the `messages` / `messages-tuple` stream mode. Will automatically write to the channel specified in the `state_key` unless `state_key` is `None`. r)BaseCallbackHandlerBaseCallbackManager) get_config)StreamMessagesHandlerc3K|]}|VdSNr4)r7xs r* zpush_message..Zs"==1======r,NzMessage ID is required callbacksc38K|]}t|VdSryrC)r7rzrts r*r{zpush_message..bs?77/0 1)**777777r,c3<K|]}t||VdSryr~)r7rzrws r*r{zpush_message..hs3EEq 1.C D DEEEEEEEr,metadatalanggraph_checkpoint_nsF)dedupe)langchain_core.callbacks.basertrulanggraph.configrvlanggraph.pregel._messagesrwnextrr>r(rChandlersrDalltupler rEsplitr_emitrr) rprorurvconfigmanagerrstream_handlerr message_metartrws @@r* push_messagerGs ,+++++@@@@@@ Z\\F==17)<<=====Gz1222&%':;;'%# F;' . .'377774:;4G77744'+&EEEEHEEEt~B*% $sH%>?@@FFvNN O O   \75AAA>%t _% 7';&<=== Nr,)rrrr )r"r.r$r.r0r1rr.)rgrirrj)rprqrorrrr )+ __future__rrFr\collections.abcrr functoolsrtypingrrr r rmr r r rrrrtyping_extensionsrrlanggraph._internal._constantsrrrlanggraph.graph.staterlanggraph.warningsr__all__rDr.rr/rrrrPrr4r,r*rs"""""" ........43333333HHHHHHHHHH,,,,,,::::::  ) *-F F&IIII& 26 wwwwwwt B 5D5D5D5D5D:5D5D 5Dp88888I888 I I I I&'--------r,