A'h) >ddlZddlZddlmZddlmZddlmZmZm Z m Z m Z m Z m Z ddlmZmZmZmZmZmZmZddlmZddlmZe eeefZdZd e d e eegeffd Zedd d edede e dd efdZGddeZGddeZ deed eefdZ!y)N)Sequence)partial) AnnotatedAnyCallableLiteralOptionalUnioncast) AnyMessage BaseMessageBaseMessageChunkMessageLikeRepresentation RemoveMessageconvert_to_messagesmessage_chunk_to_message) TypedDict) StateGraph__remove_all__funcreturnc  ddttdttdtdtttttgtffffd }j |_t tttgtf|S)Nleftrightkwargsrcj| | ||fi|S||d|rdndd}t|tfi|S)NzMMust specify non-null arguments for both 'left' and 'right'. Only received: 'rrz'.) ValueErrorr)rrrmsgrs ^/home/kushmeetdev/Regenta/Chatbot/venv/lib/python3.12/site-packages/langgraph/graph/message.py _add_messagesz,_add_messages_wrapper.._add_messages"sd   1e.v. .  !2(,f':"> S/ !4*6* *)NN)r Messagesrr r__doc__r )rr s` r_add_messages_wrapperr$!sxEI +x  +080B +UX + x8X"6"@AA B +!LLM (H-x78- HHr!)formatrrr%langchain-openaic d}t|ts|g}t|ts|g}t|Dcgc]}tt t |}}t|Dcgc]}tt t |}}|D]1}|j ttj|_3t|D]Z\}}|j "ttj|_t|tsE|j tk(sY|}\|||dzdS|j}t|Dcic]\}}|j |}}}t} |D]}|j|j x} Mt|tr| j!|j L| j#|j ||| <mt|trt%d|j dt'|||j <|j)||Dcgc]}|j | vs|}}|dk(r t+|}|S|rd|d} t%|  |Scc}wcc}wcc}}wcc}w)a>Merges 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 ToolMessages. !!! 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: ```python title="Basic usage" 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')] ``` ```python title="Overwrite existing message" msgs1 = [HumanMessage(content="Hello", id="1")] msgs2 = [HumanMessage(content="Hello again", id="1")] add_messages(msgs1, msgs2) # [HumanMessage(content='Hello again', id='1')] ``` ```python title="Use in a StateGraph" 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=...)]} ``` ```python title="Use OpenAI message format" 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"}, # }, # ], # ), # ] # } ``` Nz?Attempting to delete a message with an ID that doesn't exist ('z')r&zUnrecognized format=z+. Expected one of 'langchain-openai', None.) isinstancelistrrr ridstruuiduuid4 enumeraterREMOVE_ALL_MESSAGEScopysetgetadddiscardrlenappend_format_messages) rrr%remove_all_idxmidxmergedi merged_by_id ids_to_remove existing_idxrs r add_messagesrA4shN dD !v eT "%T*   !&6!:; D  %U+   !&6!:; E  % 44<tzz|$AD%E"!Q 44<tzz|$AD a 'ADD4G,G N ! !^a')**YY[F(1&(9:1ADD!G:L:EM (,,QTT2 2L ?!]+!!!$$'%%add+'(|$!]+ UVWVZVZU[[]^"%VL  MM!  =A144}#s I:% I? J8J  J c$eZdZdZdfd ZxZS) MessageGraphaA 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")]} ``` cTt|tttt fy)N)super__init__rr*r rA)self __class__s rrFzMessageGraph.__init__s 4 #3\#ABCr!)rN)__name__ __module__ __qualname__r#rF __classcell__)rHs@rrCrCs,\DDr!rCc(eZdZUeeeefed<y) MessagesStatemessagesN)rIrJrKrr*r rA__annotations__r!rrNrNsZ(,677r!rNrOc 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.messagesrSr ImportErrorwarningswarnr*)rOrSrs rr8r8sQ IF##=h#GHH    cH~s+AA)"r-rVcollections.abcr functoolsrtypingrrrrr r r rTr r rrrrrtyping_extensionsrlanggraph.graph.staterr*r"r0r$rArCrNr8rQr!rr]s $(, /02KK L&IIXx6JH6T-UI& 59 l l l W/0 1 l  ll^0D:0Df8I8 Ix 4 Ik9J Ir!