ޠj&D%SSKJr SSKJrJrJrJr SSKJrJ r SSK J r J r J r SSKJrJr SSKJr SSKJrJr SSKJr SS KJrJrJr SS KJr SS KJr SS K J!r! SS K"J#r#J$r$ SSK%J&r& SSK'J(r( SSK)J*r* SSK)J-r- \ "S5r.\/\/\0S4\1\0\ 44r2S!Sjr3"SS\\*5r4"SS\4\-5r5\6"1Sk5r7S\8S'\6"1Sk5r9S\8S'S"Sjr:S#Sjr;g !\+a \,r*Npf=f!\+a \,r-Nwf=f)$) annotations) AsyncIteratorCallableIteratorSequence)fields is_dataclass)AnyTypeVarcast)UUIDuuid4)BaseCallbackHandler) BaseMessage ToolMessage)convert_to_messages)ChatGenerationChatGenerationChunk LLMResult) BaseModel)filter_to_user_tags)NS_SEP) TAG_HIDDEN TAG_NOSTREAM) StreamChunk)Command)_StreamingCallbackHandler)_V2StreamingCallbackHandlerT.c[U[5(a[UR55$[U[5(a.[ U5R Vs/sHn[X5PM sn$[U5(aC[U[ 5(d.[U5Vs/sHn[XR5PM sn$gs snfs snf)zSExtract top-level field values from a state object (dict, BaseModel, or dataclass).) isinstancedictlistvaluesrtype model_fieldsgetattrr rname)objkfs T/var/www/html/gaurav/venv/lib/python3.13/site-packages/langgraph/pregel/_messages.py _state_valuesr.&s#tCJJL!! C # #)-c)?)?@)?A)?@@ c  :c4#8#8.4Sk:kVV$k:: A:s C$C c\rSrSrSrSrSS.SSjjrSS.SS jjrSS jrSS jr SS jr SSSS .SSjjr SSSS.SSjjr SS.SSjjr SS.S SjjrSSSS .S!SjjrSS.S"SjjrSS.S SjjrSrg)#StreamMessagesHandler1zA callback handler that implements stream_mode=messages. Collects messages from: (1) chat model stream events; and (2) node outputs. TN parent_nscTXlX l0Ul[5UlX0lg)aCConfigure the handler to stream messages from LLMs and nodes. Args: stream: A callable that takes a StreamChunk and emits it. subgraphs: Whether to emit messages from subgraphs. parent_ns: The namespace where the handler was created. We keep track of this namespace to allow calls to subgraphs that were explicitly requested as a stream with `messages` mode configured. Example: parent_ns is used to handle scenarios where the subgraph is explicitly streamed with `stream_mode="messages"`. ```python def parent_graph_node(): # This node is in the parent graph. async for event in some_subgraph(..., stream_mode="messages"): do something with event # <-- these events will be emitted return ... parent_graph.invoke(subgraphs=False) ``` N)stream subgraphsmetadatasetseenr3)selfr5r6r3s r-__init__StreamMessagesHandler.__init__<s#> "*, $'E "FdedupecU(aURUR;agURc[[55UlURR UR5 UR USSX!S445 g)Nrmessages)idr9strraddr5)r:metamessager?s r-_emitStreamMessagesHandler._emitas` gjjDII- zz! \ IIMM'** % KKa*wQ.@A Br=c[U[5(aURXSS9 g[U[5(a1UH*n[U[5(dMURXSS9 M, g[ U5Hqn[U[5(aURXSS9 M*[U[5(dMAUH*n[U[5(dMURXSS9 M, Ms g)NTr>)r"rrHrr.r:rFresponsevalueitems r-_find_and_emit_messages-StreamMessagesHandler._find_and_emit_messagesjs h , , JJtdJ 3 ( + +!e[11JJt4J8"'x0e[11JJt4J8x00 %%dK88 JJt$J?!& 1r=cU$Nr!r:run_idoutputs r-tap_output_aiter&StreamMessagesHandler.tap_output_aiterzs  r=cU$rRr!rSs r-tap_output_iter%StreamMessagesHandler.tap_output_iters r=) parent_run_idtagsr7c @U(aU(a [U;a[[[US5R [ 55SSnUR (d[U5S:aXR:wag[U5=n bXS'X4URU'ggg)Nlanggraph_checkpoint_nsrr\) rtupler rDsplitrr6lenr3rr7) r: serializedrArTr[r\r7kwargsns filtered_tagss r-on_chat_model_start)StreamMessagesHandler.on_chat_model_starts Tl$&>tC*C!DEKKFSTB>>c"gkbNN6J!4T!:: G#0 %'NDMM& !'?8r=chunkr[r\c [U[5(dgURRU5=n(aUR XrR 5 ggrR)r"rr7getrHrG)r:tokenrjrTr[r\rdrFs r-on_llm_new_token&StreamMessagesHandler.on_llm_new_tokensD%!455 ==$$V, ,4 , JJt]] + -r=r[c LURRU5=n(afUR(aUURS(aAURSSn[U[5(aUR XVR SS9 URRUS5 gNrTr>)r7rl generationsr"rrHrGpopr:rLrTr[rdrFgens r- on_llm_end StreamMessagesHandler.on_llm_ends~==$$V, ,4 ,##(<(22JJt[[J> &$'r=c <URRUS5 grRr7rtr:errorrTr[rds r- on_llm_error"StreamMessagesHandler.on_llm_error &$'r=c  U(Ga|URS5URS5:XGaVU(a [U;GaC[[[US5R [ 55SSnUR(d[U5S:agX4URU'[U5Hn [U [5(a6U Rb'URRU R5 MLMN[U [ 5(dMe[U [5(aM|U HNn [U [5(dMU RcM)URRU R5 MP M gggg)Nr)langgraph_noder^r_r)rlrr`r rDrarr6rbr7r.r"rrCr9rEr) r:rcinputsrTr[r\r7rdrerMrNs r-on_chain_start$StreamMessagesHandler.on_chain_starts  6"hll3C&DDZt3tC*C!DEKKFSTB>>c"gk%'NDMM& !&v.e[11xx+ ehh/,x00E39O9O %%dK88#ww2 $ dgg 6!& /4E r=c URRUS5=n(a[U[5(aUR XQR 5 g[U[ 5(ad[SU55(aMUHFn[U[5(aUR XVR 5 M5UR XV5 MH gUR XQ5 gg)Nc3@# UHn[U[5v M g7frR)r"r).0rMs r- 5StreamMessagesHandler.on_chain_end..s88@u 5'**s)r7rtr"rrOupdaterany)r:rLrTr[rdrFrMs r- on_chain_end"StreamMessagesHandler.on_chain_ends==$$VT2 24 2(G,,,,T??CHh//C88@855&E!%1144T<<H44TA &,,T< 3r=c <URRUS5 grRrzr{s r-on_chain_error$StreamMessagesHandler.on_chain_errorrr=)r7r3r9r5r6r5zCallable[[StreamChunk], None]r6boolr3ztuple[str, ...] | NonereturnNone)rFMetarGrr?rrrrFrrLr rr)rTr rUAsyncIterator[T]rr)rTr rU Iterator[T]rr)rcdict[str, Any]rAzlist[list[BaseMessage]]rTr r[ UUID | Noner\list[str] | Noner7dict[str, Any] | Nonerdr rr rmrDrjzChatGenerationChunk | NonerTr r[rr\rrdr rr rLrrTr r[rrdr rr r| BaseExceptionrTr r[rrdr rr )rcrrrrTr r[rr\rr7rrdr rr ) rLr rTr r[rrdr rr )__name__ __module__ __qualname____firstlineno____doc__ run_inliner;rHrOrVrYrgrnrwr}rrr__static_attributes__r!r=r-r0r01suJX-1 ##-#### * ##  ##JINC@ $4  &*!%*.3"3*3  3 # 33(33 32-1%)!% , ,* ,  , # , , ,  ,(&* ( ( ( # (  (  ((&* (( ( # (  ( ( &*!%*.7"77  7 # 77(77 7J&* == = # =  = =<&* (( ( # (  ( ((r=r0c^\rSrSrSrSSSS.SSjjrSS.SU4SjjjrSSjrSS .SS jjrSS .SU4S jjjr SSS .SS jjr Sr U=r $)StreamMessagesHandlerV2iaNv2 variant of `StreamMessagesHandler`. Declaring `_V2StreamingCallbackHandler` as a base flips `BaseChatModel.invoke` to route through `_stream_chat_model_events` (firing `on_stream_event`) instead of `_stream` (firing `on_llm_new_token`). Inherits `on_stream_event` from the parent, which forwards protocol events onto the messages stream channel. Pregel attaches this class instead of the v1 handler only when `StreamingHandler` opts in via the internal `CONFIG_KEY_STREAM_MESSAGES_V2` config key; direct `graph.stream(stream_mode="messages")` callers keep the v1 AIMessageChunk shape. Nric g)uIntentional no-op — v1 chunks are not used on v2-flagged runs. The v2 marker already steers `invoke` to the event generator, so `on_llm_new_token` should not fire under normal routing. This override stays a pass-through (no call to `super()`) to make the intent explicit and to guard against any caller (e.g. a node that calls `model.stream()` directly, which still fires the v1 callback) leaking AIMessageChunks onto a v2-flagged messages stream. Nr!)r:rmrjrTr[r\rds r-rn(StreamMessagesHandlerV2.on_llm_new_tokensr=r2c@>[TU]XUS9 [5Ulg)Nr2)superr;r8_streamed_run_ids)r:r5r6r3 __class__s r-r; StreamMessagesHandlerV2.__init__)s! i@,/Er=c[U[5(a&[U[5(dURXSS9 g[U[5(aHUHAn[U[5(dM[U[5(aM1URXSS9 MC g[ U5Hn[U[5(a'[U[5(dURXSS9 M?[U[5(dMVUHAn[U[5(dM[U[5(aM1URXSS9 MC M g)a;Like the v1 handler, but skip ToolMessage from node outputs. Tool results belong on the tools channel / state in v3; v2-flagged streams must not replay finalized ToolMessages as chat tokens (see MessagesTransformer). Legacy v1-only `stream_mode="messages"` still emits ToolMessages (see subgraph streaming tests). Tr>N)r"rrrHrr.rKs r-rO/StreamMessagesHandlerV2._find_and_emit_messages3s h , ,Z+5V5V JJtdJ 3 ( + +!e[11*;;;JJt4J8 " 'x0e[11*;;;JJt4J8x00 %%dK88 +BB!JJt$J? !& 1r=rpc rURRU5=n(aUR(aURS(aURSSn[U[5(aX R ;aiUR Rc"[[55UR lURRUR R5 OURXVR SS9 UR RU5 URRUS5 grr)r7rlrsr"rrrGrCrDrr9rErHdiscardrtrus r-rw"StreamMessagesHandlerV2.on_llm_endPs==$$V, ,4 ,##(<(22!7!77;;>>1-0\CKKN ckknn5 4T B &&v. &$'r=c d>URRU5 [TU] "U4UUS.UD6 g)N)rTr[)rrrr})r:r|rTr[rdrs r-r}$StreamMessagesHandlerV2.on_llm_errores> &&v.   '   r=)r[r\c hURRU5=n(aURS5S:XaNURRU5 URS5nU(aURRU5 0USES[ U50EnUR USSX445 gg) uForward a protocol event from `stream_events(version="v3")` as a messages stream part. Fires once per `MessagesData` event (`message-start`, per-block `content-block-*`, `message-finish`). The transformer layer correlates events back to a single `ChatModelStream` via `metadata["run_id"]` — attached here so the v1 `stream_mode="messages"` output (which emits `(AIMessageChunk, metadata)` via `on_llm_new_token`) keeps its original metadata shape. Lives on the v2 handler rather than the v1 base: content-block events are a v2-only concept, and forwarding them only when the v2 handler is attached keeps the message channel's shape predictable for v1 callers. eventz message-start message_idrBrTrrAN)r7rlrrEr9rDr5) r:rrTr[r\rdrFmsg_idv2_metas r-on_stream_event'StreamMessagesHandlerV2.on_stream_eventus0==$$V, ,4 , yy!_4&&**62<0IIMM&)8a8(CK8G KKa*u.>? @ -r=)rrrrrr) rrrTr r[rr\rrdr rr ) rrrrrrnr;rOrwr}rr __classcell__)rs@r-rrse &-1%)!%  *    #      6-1 2-22 * 2  22@D&* (( ( # (  ( (4&*     #       *&*!% #A#A #A # #A  #A#A #A#Ar=r>aitooluserhumansystemfunction assistantzfrozenset[str]_MESSAGE_ROLES>rrrremoverr_MESSAGE_TYPEScpURS5[;=(d URS5[;$)Nroler&)rlrr)rNs r-_is_message_dictrs) 88F ~ - S&1A^1SSr=c[U[5(a'URc[[ 55Ulgg[U[ 5(a>[ U5(a.URS5(d[[ 55US'gg[U[5(a[U5Hup[U[5(a)URc[[ 55UlMAMC[U[ 5(dMZ[ U5(dMl[U/5SnURc[[ 55UlX0U'M gg)aCoerce message-like write values to typed BaseMessages with stable IDs. Called in put_writes() before DeltaChannel writes are submitted to the checkpointer. Without this the checkpoint may store raw dicts or id=None BaseMessages; every get_state() replay then produces a different UUID and the same message appears with a different ID in each LangSmith trace. Handles three input shapes: - BaseMessage objects: assign a UUID if id is None. - Dicts with a known "role" (OpenAI-style) or "type" (LangChain format) at the root level: stamp "id" into the dict in-place. The reducer's convert_to_messages call will forward the id to the resulting BaseMessage. - Lists of the above: apply the same logic to each element, replacing dict items with coerced BaseMessages so the shared list reference seen by checkpoint_pending_writes and the background thread both get typed messages. Mutating synchronously here (before the background thread is submitted) is safe: the serialised bytes always reflect the post-coercion state. NrCr) r"rrCrDrr#rrlr$ enumerater)rMirNmsgs r-ensure_message_idsrs(%%% 88 57|EH  E4 %5e%<%<yyeg,E$K E4  'GA$ ,,77?!%'lDG#D$'', \CFa( !r=N)r*r rz Sequence[Any])rNr#rr)rMr rr)< __future__rcollections.abcrrrr dataclassesrr typingr r r uuidr rlangchain_core.callbacksrlangchain_core.messagesrrlangchain_core.messages.utilsrlangchain_core.outputsrrrpydanticrlanggraph._internal._configrlanggraph._internal._constantsrlanggraph.constantsrrlanggraph.pregel.protocolrlanggraph.typesr!langchain_core.tracers._streamingr ImportErrorobjectrrr`rDr#rr.r0r frozensetr__annotations__rrrr!r=r-rs "GG, 8<=QQ;181#'K)M CL U38_d38n ,-O(/1JO(dUA35PUAv"+F""+;" T#a ' &' )"()s$3C:C-C*)C*-C87C8