9i2`ddlmZddlZddlZddlmZmZddlmZddl m Z m Z m Z m Z ddlmZmZmZmZmZmZmZddlmZmZddlmZmZmZdd lmZdd lm Z d Z!e"eezZ#d Z$dd Z%e%dd ddZ&eddGddeZ'GddeZ(ddZ)dd ddZ*y)) 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__c d dfd }j|_ttttgtf|S)Ncj| | ||fi|S||d|rdndd}t|tfi|S)NzMMust specify non-null arguments for both 'left' and 'right'. Only received: 'leftrightz'.) ValueErrorr)r r!kwargsmsgfuncs a/var/www/html/backtest/airagagent/rag_env/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 Messages | Noner!r)r#rreturnz3Messages | Callable[[Messages, Messages], Messages])__doc__r rMessages)r%r's` r&_add_messages_wrapperr-)sU?C + +-< +OR + < +!LLM (H-x78- HHr()formatc 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)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"}, # }, # ], # ), # ] # } ``` Nz?Attempting to delete a message with an ID that doesn't exist ('z')zlangchain-openaizUnrecognized format=z+. Expected one of 'langchain-openai', None.) isinstancelistrrr r idstruuiduuid4 enumeraterrcopysetgetadddiscardr"lenappend_format_messages) r r!r.remove_all_idxmidxmergedi merged_by_id ids_to_remove existing_idxr$s r&rr<s~N 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 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")]} ``` ctjdtdt|t t ttfy)Nz}MessageGraph is deprecated in LangGraph v1.0.0, to be removed in v2.0.0. Please use StateGraph with a `messages` key instead.)rH stacklevel) warningswarnrsuper__init__rr2r r)self __class__s r&rPzMessageGraph.__init__*s9  L0 4 #3\#ABCr()r*None)__name__ __module__ __qualname__r+rP __classcell__)rRs@r&rrs ,\DDr(rceZdZUded<y)rz)Annotated[list[AnyMessage], add_messages]messagesN)rTrUrV__annotations__r(r&rr3s77r(rrYc 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.messagesr]r ImportErrorrMrNr2)rYr]r$s r&r?r?7sQ IF##=h#GHH    cH~s+AA) state_keyc> ddlm m}ddlm}ddlm |}tdt|gD}|j tdt|d|r|d}|j}n/t|dtrt fd |dDr|d}t fd Ddx}rJ|d }tt!t"|d j%t&|f} |j)| |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)StreamMessagesHandlerc3 K|]}|ywNr[).0xs r& zpush_message..Zs=1=s NzMessage ID is required callbacksc36K|]}t|ywrgr1)rhrirbs r&rjzpush_message..bs7/0 1)*7sc3<K|]}t|s|ywrgrm)rhrires r&rjzpush_message..hsEq 1.C DEsmetadatalanggraph_checkpoint_nsF)dedupe)langchain_core.callbacks.baserbrclanggraph.configrdlanggraph.pregel._messagesrenextrr3r"r1handlersr2alltupler r4splitr_emitrr) messager`rcrdconfigmanagerrvstream_handlerro message_metarbres @@r& push_messagerGs, ,@ \F=17)<==Gzz122&%':;%## F;' .374:;4G74+&EHEt~*% $sH%>?@FFvN O   \75A%t _% 7';&<= Nr()r%rr*z(Callable[[Messages, Messages], Messages])r r,r!r,r.z"Literal['langchain-openai'] | Noner*r,)rYzSequence[BaseMessage]r*zlist[BaseMessage])r{z,MessageLikeRepresentation | BaseMessageChunkr`z str | Noner*r )+ __future__rr5rMcollections.abcrr functoolsrtypingrrr r r^r r r rrrrtyping_extensionsrrlanggraph._internal._constantsrrrlanggraph.graph.staterlanggraph.warningsr__all__r2r,rr-rrrr?rr[r(r&rs " .4HH,:  ) *-F F&I& 26 w w w / w  wwt B 5D:5D 5Dp8I8 I&'- 9-- -r(