tj5DdZddlmZddlmZmZmZddlmZm Z m Z dZ e hdZ d.d Zd/dZd0dZd1d2dZe dZd3dZd4dZd5d!ZGd"d#e ZGd$d%ZGd&d'ZGd(d)ZGd*d+ZGd,d-ZdS)6u;Per-channel event → items state machines. Used both by the projection iterators (`_ValuesProjection`, `_MessagesProjection`, `_ToolCallsProjection`, `_SubgraphsProjection`) on `AsyncThreadStream` / `SyncThreadStream`, and by `interleave_projections`, which drives multiple decoders from one shared subscription. ) annotations)CallableIterableMapping)AnyLiteralProtocol)valuesmessages tool_calls subgraphsupdates checkpointstasks>inputtools lifecyclechannels list[str]returnNonec |D]E}|tvr:|dkrdnd}t|d|ddtdFdS) aMReject reserved protocol channel names before they hit the fallback. Genuine extension names pass through untouched; only names that ``infer_channel`` treats as built-in methods without an interleave decoder are rejected, so a typo'd or unsupported protocol channel surfaces an error instead of an empty stream. rz (use "tool_calls")z. is not a valid interleave_projections channelz. Supported channels: z, z, or an extension name.N)RESERVED_INTERLEAVE_CHANNELS ValueErrorjoinSUPPORTED_INTERLEAVE_CHANNELS)rchhints [/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/langgraph_sdk/stream/decoders.pyvalidate_interleave_channelsr!"s - - -,.'MM((rD((t(('+yy1N'O'O(((  . params_fieldrct|tsgS|dpg}t|trt|ngS)N namespace) isinstancedictgetlist)r#r%s r _event_namespacer*4sP lD ) )   --3I(D99 A4 ???rAr"datadict[str, Any] str | Nonec||dp|d}|t|ndS)Nid message_idr(str)r+r0s r _message_event_idr3;s7$9488L#9#9J(43z???$>r"Nfallbackr2c@t|}|d|S|d|SdS)aReturn the routing key for a message-channel event in `active`. Keys on `message_id` when available so concurrent messages that share the same `run_id` (two AI turns in one agent step) route to independent streams rather than colliding on a shared `run:` slot. Nzmessage: __single__)r3)r+r4r0s r _message_route_keyr7@s>#4((J&*&&&$($$$ >SZZ!^ + Wi #e** 6M0N0NRW0WWr"ceZdZddZdS)DecodereventMapping[str, Any]r Iterable[Any]cdSN)selfrTs r feedz Decoder.feedgsr"NrTrUrrV)__name__ __module__ __qualname__r[rYr"r rSrSfsBBBBBBr"rSc$eZdZdZd ddZdd ZdS) DataDecoderuWYields `params.data` from events of a single `method`. Covers the channels whose projection is just "emit the payload": `values`, `updates`, `checkpoints`, `tasks` — the SDK analog of local's `Values`/`Updates`/`Checkpoints`/`TasksTransformer`, all of which push `params["data"]` unchanged. The REST-state seeding for `values` stays at the projection layer; it is a one-shot pre-stream fetch, not part of the event state machine. Args: method: The protocol `method` this decoder consumes. namespace: When not `None`, events whose namespace differs are ignored (scope filter, mirroring the local transformers' `namespace != scope` check). `None` consumes every namespace — the historical `values` projection behavior, where subscription scoping is handled upstream. Nmethodr2r%list[str] | NonecD||_|t|nd|_dSrX)_methodr) _namespace)rZrbr%s r __init__zDataDecoder.__init__|s# -6-B$y///r"rTrUrrVc#K|d|jkrdS|dpi}|jt||jkrdS|d}||VdSdS)Nrbparamsr+)r(rerfr*rZrTrir+s r r[zDataDecoder.feeds 99X  $, . . F8$$* ? &+;F+C+Ct+V+V Fzz&!!  JJJJJ  r"rX)rbr2r%rcr\r]r^r___doc__rgr[rYr"r rarajsP"MMMMMr"rac"eZdZdZd dZdd Zd S)MessagesDecoderaWYields one chat-model stream per `message-start` event. Subsequent events route to the matching stream via `stream.dispatch(data)`. Mirrors the per-event body of `_MessagesProjection._messages_iter` (`_async/stream.py:404-458`). The subscription open/close and the `_root_messages_inbox` drain branch stay at the projection layer. Args: namespace: Events whose namespace differs are ignored (scope filter). stream_factory: Keyword-only `(namespace, node, message_id) -> stream`. Sync binds `ChatModelStream`; async binds `AsyncChatModelStream`. r%rstream_factoryCallable[..., Any]cJt||_||_i|_dSrX)r)rf_stream_factory_active)rZr%ros r rgzMessagesDecoder.__init__s$ y//-') r"rTrUrrVc#K|ddkrdS|dpi}t||jkrdS|d}t|tsdS|ddkrt |}t ||}t|dtr|dni}|t|j|r|d nd| }||j |<| ||VdSt |}|j |}|Q|d krKt|j d kr3tt|j }|dS| ||dd vr8t|j D]\}} | |ur|j |=dSdS)Nrbr rir+rTz message-start)r4metadatalanggraph_node)r%noder0r6rN)zmessage-finishrH)r(r*rfr&r'r3r7rrr)rsdispatchrOnextiterr items) rZrTrir+r0keyrustream route_key candidates r r[zMessagesDecoder.feedsG 99X  * , , F8$$* F # #t 6 6 Fzz&!!$%%  F 88G   / /*400J$TJ???C(2488J3G3G(N(NV$$$TV ))t//7?IX\\"2333T%*F !'DL  OOD ! ! !LLLLL$T**C\%%c**F~#"5"5#dl:K:Kq:P:Pd4<#6#6#8#899::~ OOD ! ! !xx  $???,01C1C1E1E,F,F44(Iy F** L3@?44r"N)r%rrorpr\rkrYr"r rnrnsF  ****"4"4"4"4"4"4r"rnc"eZdZdZd dZdd Zd S)ToolCallsDecoderaYields one tool-call handle per `tool-started` event. Mirrors the per-event body of `_ToolCallsProjection._tool_calls_iter` (`_async/stream.py:1168-1217`). The thread register/unregister and the terminal-error-on-close finally stay at the projection / wrapper layer. Args: namespace: Events whose namespace differs are ignored. handle_factory: Keyword-only `(tool_call_id, name, input, namespace) -> handle`. r%rhandle_factoryrpcJt||_||_i|_dSrX)r)rf_handle_factoryrs)rZr%rs r rgzToolCallsDecoder.__init__s"y//-') r"rTrUrrVc#K|ddkrdS|dpi}t||jkrdS|d}t|tsdS|d}t|t sdS|d}|dkrz|d}||t|t r|nd |d t|j }||j|<|VdS|d kra|j|}|d }|,t|t r| |dSdSdS|dkrI|j |d}|*| |ddSdS|dkrg|j |d}|L|d} | t| rt | nddSdSdS)Nrbrrir+ tool_call_idrTz tool-started tool_namerr)rrArr%ztool-output-deltadeltaz tool-finishedoutputz tool-errormessagezTool call errored)r(r*rfr&r'r2rr)rs _push_deltapop_finish_fail RuntimeError) rZrTrir+r event_typerAhandlerrs r r[zToolCallsDecoder.feeds~ 99X  ' ) ) F8$$* F # #t 6 6 Fzz&!!$%%  Fxx// ,,,  FXXg&&  ' '88K((D)))'c22:TThhw''t// *F *0DL &LLLLL . . .\%%l33FHHW%%E!j&<&<!""5)))))"!!! ? * *\%%lD99F!txx1122222"! < ' '\%%lD99F!((9--  !QW>QRR ( '!!r"N)r%rrrpr\rkrYr"r rrsF  **** &&&&&&r"rc2eZdZdZddZdd ZddZddZdS)SubgraphsDecoderaDiscovers child subgraph handles and fans out events to active ones. Mirrors the per-event body of `_SubgraphsProjection._subgraphs_iter` (`_async/stream.py:963-1041`) plus `_apply_tasks_result`. Root-inbox forwarding and terminal-status-on-close stay at the projection / wrapper layer. Args: scope: Tuple-form namespace of this decoder's parent. `()` for root. handle_factory: Keyword-only `(path, graph_name, trigger_call_id) -> handle`. rJrKrrpcV||_||_i|_t|_dSrX)_scoperrsset_seen)rZrJrs r rgzSubgraphsDecoder.__init__s' -35 +.55 r"rTrUrrVc#K|dpi}t|}|d}t|tsdS|d}t |}|jD]L\}}t|} t|| kr%|d| |kr||nM|dkrPd|vr| ||dSt||j r| |Ed{VdSdS|dkrK|ddkr4t||j r!| |Ed{VdSdSdSdS) Nrir+rbrresultrrTr8) r(r*r&r'rPrsr{rO _push_event_apply_tasks_resultrQr _discover) rZrTrir%r+rbns_tuple child_path child_handle child_lens r r[zSubgraphsDecoder.feeds8$$*$V,, zz&!!$%%  F8$$##(, (:(:(<(<   $J JI8}} ))hz z.Bj.P.P((/// W  4((D99999!)T[99 5>>)44444444444 5 5 k ! !!!Y.. DK88/~~i00 0 0 0 0 0 0 0 0 0 " !....r"r%rc#Kt|}||jvrdS|j|t|d\}}|||pd|}||j|<|VdS)N)path graph_nametrigger_call_id)rPraddrDrrs)rZr%rrrrs r rzSubgraphsDecoder._discover1sY 4:   F t&>tBx&H&H# O%%!)T+&   $ T r"r+r,rcD|d}|sdSt|}t|jD]P\}}|dd|kr|j|kr t |\}}||||j|=QdS)Nr/r)r(rPr)rsr{rrIr) rZr%r+ result_id parent_pathrrstatusrHs r rz$SubgraphsDecoder._apply_tasks_result?sHHTNN   FI&& "&t|'9'9';';"<"< ) ) J#2#+--%227==MFE NN65 ) ) ) Z(( ) )r"N)rJrKrrpr\)r%rrrV)r%rr+r,rr)r]r^r_rlrgr[rrrYr"r rrsn  1111 1111:     ) ) ) ) ) )r"rc"eZdZdZd dZd d Zd S) ExtensionsDecoderaIYields `params.data` from one named custom channel. Mirrors `_ExtensionProjection._iter` (`_async/stream.py:1278-1299`), with an added name filter so it can share one subscription in interleave. Args: name: The extension name. Only `custom` events whose `data["name"]` matches are consumed. rAr2c6|std||_dS)Nz!extension name must be non-empty.)r_name)rZrAs r rgzExtensionsDecoder.__init__Ys% B@AA A r"rTrUrrVc#K|ddkrdS|dpi}|d}t|tsdS|d|jkrdS|VdS)Nrbcustomrir+rA)r(r&r'rrjs r r[zExtensionsDecoder.feed^s 99X  ( * * F8$$*zz&!!$%%  F 88F  tz ) ) F r"N)rAr2r\rkrYr"r rrNsF      r"r)rrrr)r#rrr)r+r,rr-rX)r+r,r4r-rr2)r<r2rr=)r+r,rrE)r%rrJrKrrL)rl __future__rcollections.abcrrrtypingrrr r frozensetrr!r*r3r7SubgraphStatusrDrIrQrSrarnrrrrYr"r rs6#"""""7777777777))))))))))! )y)H)H)HII$BBBB????     HI**** XXXXCCCCChCCCB9494949494949494x77777777tJ)J)J)J)J)J)J)J)Zr"