A'hKddlmZmZmZddlmZddlmZmZm Z m Z m Z ddl m Z ddlmZddlmZddlmZddlmZmZmZmZdd lmZmZdd lmZdd lmZ dd l!m"Z"dd l#m$Z$ddl%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-ddl.m/Z/ddl0m1Z1ddl2m3Z3m4Z4m5Z5mZddl6mZm7Z7m8Z8ddl9m:Z:e;e(e'e)e+fZZ?Gdde1Z@y)) AsyncIteratorIteratorSequence)asdict)AnyLiteralOptionalUnioncast)RunnableConfig)Edge)Graph)Node)LangGraphClientSyncLangGraphClient get_clientget_sync_client) Checkpoint ThreadState)Command) StreamMode)Self)CheckpointMetadata)CONFCONFIG_KEY_CHECKPOINT_IDCONFIG_KEY_CHECKPOINT_MAPCONFIG_KEY_CHECKPOINT_NSCONFIG_KEY_STREAMCONFIG_KEY_TASK_ID INTERRUPTNS_SEP)GraphInterrupt)PregelProtocol)All PregelTask StateSnapshotr)r InterruptStreamProtocol) merge_configsvreturncht|ttttfr|St|t r@i}|j D])\}}t|tst|}|%|||<+|St|ttfr*g}|D]!}t|}||j|#|Sy)zJRecursively sanitize a config value to ensure it contains only primitives.N) isinstancestrintfloatbooldictitemssanitize_config_valuelisttupleappend)r*sanitized_dictkvalsanitized_valuesanitized_listitemsanitized_items ^/home/kushmeetdev/Regenta/Chatbot/venv/lib/python3.12/site-packages/langgraph/pregel/remote.pyr4r4;s!c3t,- At ggi 8FAs!S!"7"<".(7N1%  8  Ae} % 6D248N)%%n5 6 ceZdZdZy)RemoteExceptionz:Exception raised when an error occurs in the remote graph.N)__name__ __module__ __qualname____doc__r@r?rBrBQsDr@rBceZdZUdZeed<eeed<dddddddddedeedeedeeeefd eed ee d ee deefd Z d efdZ d e fdZ deeefd efdZ dHd ee ded efdZdeeeeeeffd eeeffdZ dHddd ee deeefd efdZ dHddd ee deeefd efdZded efdZd ee d eefdZded e fdZ d e d e fd Z!dd!d e d"ed efd#Z"dd!d e d"ed efd$Z#dddd%d e d&eeeefd'ee d(eed e$ef d)Z%dddd%d e d&eeeefd'ee d(eed e&ef d*Z'd e d+ee(eeeefeefd e fd,Z)d e d+ee(eeeefeefd e fd-Z* dHd e d.eeeeefefd/eed e fd0Z+ dHd e d.eeeeefefd/eed e fd1Z, dId2eee-ee-fd ee d3e-d e(ee.ee.eee/ffd4Z0 dHddddd5d6eeeefefd ee d2eee-ee-fd7eee1e2efd8eee1e2efd"eded e$eeeefeffd9Z3 dHddddd5d6eeeefefd ee d2eee-ee-fd7eee1e2efd8eee1e2efd"eded e&eeeefeffd:Z4 dHddddddd;d6ed ee dee2e1d?ee2e1d@ee2e1dAee2e1dBee2e1dCee2e1ded e&eeeffdDZ6 dHdddEd6eeeefefd ee d7eee1e2efd8eee1e2efded eeeefeff dFZ7 dHdddEd6eeeefefd ee d7eee1e2efd8eee1e2efded eeeefeff dGZ8y)J RemoteGraphaiThe `RemoteGraph` class is a client implementation for calling remote APIs that implement the LangGraph Server API specification. For example, the `RemoteGraph` class can be used to call APIs from deployments on LangGraph Platform. `RemoteGraph` behaves the same way as a `Graph` and can be used directly as a node in another `Graph`. assistant_idnameN)urlapi_keyheadersclient sync_clientconfigrKrLrMrNrOrPrQc||_|||_n||_||_||t|||}||_||t |||}||_y)a Specify `url`, `api_key`, and/or `headers` to create default sync and async clients. If `client` or `sync_client` are provided, they will be used instead of the default clients. See `LangGraphClient` and `SyncLangGraphClient` for details on the default clients. At least one of `url`, `client`, or `sync_client` must be provided. Args: assistant_id: The assistant ID or graph name of the remote graph to use. url: The URL of the remote API. api_key: The API key to use for authentication. If not provided, it will be read from the environment (`LANGGRAPH_API_KEY`, `LANGSMITH_API_KEY`, or `LANGCHAIN_API_KEY`). headers: Additional headers to include in the requests. client: A `LangGraphClient` instance to use instead of creating a default client. sync_client: A `SyncLangGraphClient` instance to use instead of creating a default client. config: An optional `RunnableConfig` instance with additional configuration. name: Human-readable name to attach to the RemoteGraph instance. This is useful for adding `RemoteGraph` as a subgraph via `graph.add_node(remote_graph)`. If not provided, defaults to the assistant ID. N)rLrMrN)rJrKrQrrOrrP) selfrJrLrMrNrOrPrQrKs r?__init__zRemoteGraph.__init__esd>) <$DIDI >coC'JF  3?)c7GTK&r@r+cH|j td|jS)NzbAsync client is not initialized: please provide `url` or `client` when initializing `RemoteGraph`.)rO ValueErrorrSs r?_validate_clientzRemoteGraph._validate_clients( ;; t {{r@cH|j td|jS)NzfSync client is not initialized: please provide `url` or `sync_client` when initializing `RemoteGraph`.)rPrVrWs r?_validate_sync_clientz!RemoteGraph._validate_sync_clients,    #x r@updatecfi|j|}|j|jdfi|S)NrJ)__dict__ __class__pop)rSr[attrss r?copyzRemoteGraph.copys4+4==+F+t~~eii7A5AAr@kwargsc n|jdt|j|tt|iS)NrQ)rar)rQr r )rSrQrbs r? with_configzRemoteGraph.with_configs1yy }T[[&$~v:VW X  r@graphc i}|dD]{}t|d}|jdi}|jd}|%t|tr|jd|}n|}t ||||jd||<}|S)NnodesiddatarKmetadata)rhrKrirj)r.getr-r2 DrawableNode)rSrergnodenode_id node_data node_names r?_get_drawable_nodeszRemoteGraph._get_drawable_nodess'N D$t*oG,I(I i. ) fg >I 'I)*- E'N $ r@F)xrayrrc |j}|jj|j|}t |j ||dDcgc] }t di|c}Scc}w)Get graph by graph name. This method calls `GET /assistants/{assistant_id}/graph`. Args: config: This parameter is not used. xray: Include graph representation of subgraphs. If an integer value is provided, only subgraphs with a depth less than or equal to the value will be included. Returns: The graph information for the assistant in JSON format. rJrredgesrgrvrG)rZ assistants get_graphrJ DrawableGraphrq DrawableEdge)rSrQrrrPreedges r?ryzRemoteGraph.get_graphss&002 &&00**1 **5149'NCD<'$'C  CsA. c K|j}|jj|j|d{}t |j ||dDcgc] }t di|c}S7:cc}ww)rtruNrvrwrG)rXrxryrJrzrqr{)rSrQrrrOrer|s r? aget_graphzRemoteGraph.aget_graphs&&&(''11**2  **5149'NCD<'$'C   Ds!;A?A8 A?A: 0 A?:A?statec0g}|dD]}g}|dD]}|jtdi||jt|d|dt|drt |dndt||dr|j |dn|drt td|dind|jd  t|d |d rt|d n td|dd |dd|dd|djdiditdi|d|d|dr.d|dd |dd|dd|djdidindt|t|Dcgc]}|jD]}|c}}Scc}}w)Ntasks interruptsrhrKerrorr checkpoint configurableresult)rhrKpathrrrrvaluesnext thread_id checkpoint_ns checkpoint_idcheckpoint_maprrrrrj created_atparent_checkpoint)rrrQrjr parent_configrrrG) r7r'r%r6 Exception_create_state_snapshotr r rkr&rr)rSrrtaskr interruptis r?rz"RemoteGraph._create_state_snapshots"$'N DJ!,/ : !!)"8i"89 : LLDzf6:7m)DM2$Z0 =33DMB $L1!.$|BT1UV!% 88H-   2?).vuV}%EG!&|!4[!A%*<%8%I%*<%8%I&+L&9&=&=>NPR&S !(<% *;<\*,-#%*+>%? %L)./B)CO)T)./B)CO)T*/0C*D*H*H,b+ % ,EKD4??KaaKaKL9  8Ls+Fc|yi}d|dvr |dd|d<d|dvr |dd|d<d|dvr |dd|d<d|dvr |dd|d<|r|SdS)NrrrrrrG)rSrQrs r?_get_checkpointzRemoteGraph._get_checkpoint8s > &0 0&,^&<[&IJ{ # f^4 4*0*@*QJ ' f^4 4*0*@*QJ ' vn5 5+1.+ABR+SJ' ('z1T1r@rcFd|d|d|d|jdidiS)Nrrrrrr)rk)rSrs r? _get_configzRemoteGraph._get_configIs; ' 4!+O!Ij)!,  ? V #(*In %~.446 C1q#&.,A!,DDQ3BIn-a0  C+!Xs CC) subgraphsrc|j}t|j|}|jj |dd|j ||}|j |S)Get the state of a thread. This method calls `POST /threads/{thread_id}/state/checkpoint` if a checkpoint is specified in the config or `GET /threads/{thread_id}/state` if no checkpoint is specified. Args: config: A `RunnableConfig` that includes `thread_id` in the `configurable` field. subgraphs: Include subgraphs in the state. Returns: The latest state of the thread. rrrrr)rZr)rQthreads get_staterr)rSrQrrP merged_configrs r?rzRemoteGraph.get_statepsn"002 %dkk6: ##--#N3K@++M:.  **511r@cK|j}t|j|}|jj |dd|j ||d{}|j |S7w)rrrrN)rXr)rQrrrr)rSrQrrOrrs r? aget_statezRemoteGraph.aget_statesz"&&(%dkk6: nn..#N3K@++M:/   **511  sAA7A5 A7)filterbeforelimitrrrc #K|j}t|j|}|jj |dd|r|nd|j |||j |}|D]}|j |ywaGet the state history of a thread. This method calls `POST /threads/{thread_id}/history`. Args: config: A `RunnableConfig` that includes `thread_id` in the `configurable` field. filter: Metadata to filter on. before: A `RunnableConfig` that includes checkpoint metadata. limit: Max number of states to return. Returns: States of the thread. rr )rrrrjrN)rZr)rQr get_historyrr) rSrQrrrrPrstatesrs r?get_state_historyzRemoteGraph.get_state_historys,002 %dkk6: $$00#N3K@ %b''/++M: 1  5E--e4 4 5sB B c 2K|j}t|j|}|jj |dd|r|nd|j |||j |d{}|D]}|j |y7 wr)rXr)rQrrrr) rSrQrrrrOrrrs r?aget_state_historyzRemoteGraph.aget_state_historys,&&(%dkk6: ~~11#N3K@ %b''/++M: 2   5E--e4 4 5 sA2B4B5!BupdatesctNNotImplementedErrorrSrQrs r?bulk_update_statezRemoteGraph.bulk_update_states "!r@cKtwrrrs r?abulk_update_statezRemoteGraph.abulk_update_states "! ras_nodec|j}t|j|}|jj |dd|||j |}|j |dS)Update the state of a thread. This method calls `POST /threads/{thread_id}/state`. Args: config: A `RunnableConfig` that includes `thread_id` in the `configurable` field. values: Values to update to the state. as_node: Update the state as if this node had just executed. Returns: `RunnableConfig` for the updated thread. rrrrrrr)rZr)rQr update_staterr)rSrQrrrPrresponses r?rzRemoteGraph.update_statesv&002 %dkk6: $,,99#N3K@++M: :   677r@cK|j}t|j|}|jj |dd|||j |d{}|j |dS7w)rrrrNr)rXr)rQrrrr)rSrQrrrOrrs r? aupdate_statezRemoteGraph.aupdate_states&&&(%dkk6: %~~::#N3K@++M: ;    677  sAA; A9!A; stream_modedefaultcHg}d}|r6t|tr|j|n%d}|j|n|j||j }|xsij t ij t}|r|j|jd|vr"|jd|jdd|vr"|jd|jdd|vr|jdd|vr|jd||||fS)aHReturn a tuple of the final list of stream modes sent to the remote graph and a boolean flag indicating if stream mode 'updates' was present in the original list of stream modes. 'updates' mode is added to the list of stream modes so that interrupts can be detected in the remote graph. TFmessageszmessages-tuplerevents) r-r.r7extendrarkrrmodesremove)rSrrQrupdated_stream_modes req_singlerequested_stream_modesstreams r?_get_stream_modeszRemoteGraph._get_stream_modes6s(57 +s+$++K8" $++K8 ' ' 0!5!:!:!<\r  tR ( , ,-> ?   ' ' 5 - - ' ' 3 ' '(8 9 5 5 " ) )*: ; " ) )* 5 0 0 ' ' 2 + + ' ' 1$&U[[( % 1 1&! LsHML< D'Mc K|j}t|j|} |j| } |j ||\} } } }t |t rttt|}d}nd}|jjd| djd|j||| | |||xs|dudd |23d{}t|jvrB|jj!td\}}t#|j!t}n|jd}}|xsijt$ijt&x}r#t#|j!t}||z}|#||j(vr||||j*f|jj-drat |j*t.rwt0|j*vre|rct3|j*t0Dcgc] }t5di|c}|jj-d rt7|j*|| vr|jj-d r%|j9t#|j* }|rt|jvrB|jj!td\}}t#|j!t}n|jd}}| r||j*fM|||j*fa| r|j*t|{7wcc}w6ywr)rXr)rQrrr-rr rrrrrkrJr!rrr6rrrrirr2r r"r'rBr)rSrrQrrrrrbrOrrrrrrrrrrrrrs r?astreamzRemoteGraph.astreams:&&(%dkk6: 00?6:6L6L 7 3 iV eW %,0VE],KGEG-6;;--  &~6::;G**#$-+&<&*<"    8 8 %$!KK--fa8 c399V,- ;;b#\r..tR8<<=UVVyV!)//&"9: ^!dfll&:D%**-.{{%%i0ejj$/I4K ,5:ZZ 5JKY^^K''0%ejj119${{%%j1E%**,=>U[[( % 1 1&! L?  s8CM M  M M D=M  M D$M M M ) include_names include_types include_tags exclude_names exclude_types exclude_tagsversion)v1v2rrrrrrcKtwrr) rSrrQrrrrrrrrbs r?astream_eventszRemoteGraph.astream_events1s"!r)rrc `|j|f|||dd|D]} S#t$rYywxYw)aCreate a run, wait until it finishes and return the final state. Args: input: Input to the graph. config: A `RunnableConfig` for graph invocation. interrupt_before: Interrupt the graph before these nodes. interrupt_after: Interrupt the graph after these nodes. **kwargs: Additional params to pass to RemoteGraph.stream. Returns: The output of the graph. rrQrrrN)rUnboundLocalErrorrSrrQrrrbrs r?invokezRemoteGraph.invokeAs]*!T[[  -+     E   L   s ! --cxK|j|f|||dd|23d{} 76 S#t$rYywxYww)aCreate a run, wait until it finishes and return the final state. Args: input: Input to the graph. config: A `RunnableConfig` for graph invocation. interrupt_before: Interrupt the graph before these nodes. interrupt_after: Interrupt the graph after these nodes. **kwargs: Additional params to pass to RemoteGraph.astream. Returns: The output of the graph. rrN)rrrs r?ainvokezRemoteGraph.ainvokedsk*(4<<  -+      %    L   s4:'%':':+: 7:7:r)r)9rCrDrErFr.__annotations__r r2rrr rTrXrZrrrardr5rlrqr r/r1rzryr~rr&rrrrrrrrrrrr6rrrrr StreamModeSDKr(rr$rrrrrrrrGr@r?rIrIWs| 3-"!%,0,059+/",',' c] ,' # ,'$sCx.),'),'12,'(,'sm,'\/ ': B4S>BdB 26 ~. AD  #tDcN334 c< 4,0 "'  ( CI    >,0 "'  ( CI    :8 K8 M8 t2h~&>28JCW2" j ^ ~.<,0+/# !5!5c3h( !5 ( !5 } !5 - !5N,0+/# !5!5c3h( !5 ( !5 } !5 } %!5F""eHT#s(^4hsmCDE"  """eHT#s(^4hsmCDE"  ""& 88tCH~s2348# 8  8D"& 88tCH~s2348# 8  8D( 3ReJZ0@$@AB3R(3R 3R  ]T-0$8PP  3Rp,0a FJ@D?CaT#s(^S()a(a eJZ0@$@AB a #5hsm);#<= a"%Xc](:";<aaa %S#X+, -aL,0a FJ@D?CaT#s(^S()a(a eJZ0@$@AB a #5hsm);#<= a"%Xc](:";<aaa uT#s(^S01 2aL,0" 261504151504""(" $ "   . "  ."x}-"  ."  ."x}-"" tCH~ &"&,0! AE?C !T#s(^S()!(! #5hsm);#<= ! "%Xc](:";< !! tCH~s" #!L,0! AE?C !T#s(^S()!(! #5hsm);#<= ! "%Xc](:";< !! tCH~s" #!r@rIN)Acollections.abcrrr dataclassesrtypingrrr r r langchain_core.runnablesr langchain_core.runnables.graphr r{rrzrrllanggraph_sdk.clientrrrrlanggraph_sdk.schemarrrrrrtyping_extensionsrlanggraph.checkpoint.baserlanggraph.constantsrrrrrrr r!langgraph.errorsr"langgraph.pregel.protocolr#langgraph.pregel.typesr$r%r&langgraph.typesr'r(langgraph.utils.configr) frozensetrr4rrBrIrGr@r?rs==4 96<"8   ,4MM>>0!    SS, i n .n r@