
    c
io                       U d Z ddlmZ ddlmZmZ ddlmZ ddlmZ ddl	m
Z
mZmZmZmZmZmZ ddlmZmZ eee
f         dz  Z	 ed	         Z	 ed
         Z	 ed         Z	 ed         Z	 ed         Z	 ed         Z	 ed         Z	 ed         Z	 ed         Z	 ed         Z 	 ed         Z!	 ed         Z"	 ed         Z#	 ed         Z$	 ed         Z%	 ed         Z&	 ed         Z'	 ed         Z(	  G d ded          Z) G d d e          Z* G d! d"e          Z+eee+f         Z, G d# d$e          Z- G d% d&e-          Z. G d' d(e-          Z/ G d) d*e          Z0 G d+ d,e          Z1 G d- d.e          Z2 G d/ d0e          Z3 G d1 d2e          Z4 G d3 d4e          Z5 G d5 d6e          Z6 G d7 d8e          Z7 G d9 d:ed          Z8ed;         Z9ed<         Z:ed=         Z;ed>         Z<ee=z  e>z  e?z  dz  Z@eee@ee@         z  f         eAeBee@f                  z  eBeBee@f         d?f         z  ez  eCz  ZD G d@ dAe          ZE G dB dCe          ZF G dD dEe          ZG G dF dGeFd          ZH G dH dIe          ZI G dJ dKe          ZJedL         ZK	  G dM dNe          ZL G dO dPe          ZM G dQ dRe          ZN G dS dTe          ZO G dU dVe          ZP G dW dXe          ZQ G dY dZe          ZRePeQz  eRz  ZS	  G d[ d\e          ZT G d] d^e          ZU G d_ d`e          ZV G da dbe          ZW G dc dde          ZX G de dfe          ZY G dg dhe          ZZ G di dje          Z[ G dk dle          Z\ G dm dne          Z] G do dpe          Z^ G dq dre          Z_eUeVz  eWz  eXz  eYz  eZz  e[z  e\z  e]z  e^z  e_z  Z`	  G ds dte          Za G du dved          Zb G dw dxe          Zc G dy dze          Zd G d{ d|e          Ze G d} d~e          Zf G d de          Zgdez  e=z  e>z  e?z  Zhdeid<   eeeeheAeh         deAd         f         f         Zjdeid<   edeez  efz  egz  ejz  Zkdeid<   ekZldeid<   dS )z3Data models for interacting with the LangGraph API.    )annotations)MappingSequence)Field)datetime)AnyClassVarLiteral
NamedTupleProtocol	TypeAliasUnion)NotRequired	TypedDictN)pendingrunningerrorsuccesstimeoutinterrupted)idlebusyr   r   )	run_modes	lifecyclestate_update)	valuesmessagesupdateseventstaskscheckpointsdebugcustomzmessages-tuple)cancelcontinue)reject	interruptrollbackenqueue)raise
do_nothing)deletekeep)syncasyncexit*)creater&   )r,   keep_latest)r'   r(   )r   r   all)assistant_idgraph_idname
created_at
updated_at)	thread_idstatusr8   r9   state_updated_at)cron_idr5   r:   r8   r9   next_run_dateend_time)ascdescc                  6    e Zd ZU dZded<   	 ded<   	 ded<   dS )	Configz!Configuration options for a call.	list[str]tagsintrecursion_limitdict[str, Any]configurableN__name__
__module____qualname____doc____annotations__     C:\Users\Dell Inspiron 16\Desktop\tws\AgrotaPowerBi\back-agrota-powerbi\mcp-client-agrota\venv\Lib\site-packages\langgraph_sdk/schema.pyrC   rC      sK         ++OOO
  !    rQ   rC   F)totalc                  B    e Zd ZU dZded<   	 ded<   	 ded<   	 ded<   d	S )

Checkpointz1Represents a checkpoint in the execution process.strr:   checkpoint_ns
str | Nonecheckpoint_iddict[str, Any] | Nonecheckpoint_mapNrJ   rP   rQ   rR   rU   rU      sN         ;;NNNKQ?))))BBrQ   rU   c                  Z    e Zd ZU dZded<   	 ded<   	 ded<   	 ded<   	 ded<   	 ded	<   d
S )GraphSchemaz0Defines the structure and properties of a graph.rV   r6   dict | Noneinput_schemaoutput_schemastate_schemaconfig_schemacontext_schemaNrJ   rP   rQ   rR   r]   r]      sx         ::MMM===== =rQ   r]   c                  ~    e Zd ZU dZded<   	 ded<   	 ded<   	 ded<   	 d	ed
<   	 ded<   	 ded<   	 ded<   	 ded<   dS )AssistantBasezBase model for an assistant.rV   r5   r6   rC   configContextcontextr   r8   JsonmetadatarF   versionr7   rX   descriptionNrJ   rP   rQ   rR   re   re      s         &&"MMMNNN.-NNN!LLL&III#**rQ   re   c                      e Zd ZdZdS )AssistantVersionz.Represents a specific version of an assistant.N)rK   rL   rM   rN   rP   rQ   rR   rn   rn     s        88DrQ   rn   c                      e Zd ZU dZded<   dS )	Assistantz3Represents an assistant with additional properties.r   r9   NrJ   rP   rQ   rR   rp   rp   	  s$         ==22rQ   rp   c                  *    e Zd ZU dZded<   	 ded<   dS )AssistantsSearchResponsez0Paginated response for assistant search results.zlist[Assistant]
assistantsrX   nextNrJ   rP   rQ   rR   rr   rr     s3         ::>KKrQ   rr   c                  *    e Zd ZU dZded<   	 ded<   dS )	Interruptz1Represents an interruption in the execution flow.r   valuerV   idNrJ   rP   rQ   rR   rv   rv     s-         ;;JJJ2GGGGGrQ   rv   c                  r    e Zd ZU dZded<   	 ded<   	 ded<   	 ded<   	 d	ed
<   	 ded<   	 ded<   	 ded<   dS )Threadz!Represents a conversation thread.rV   r:   r   r8   r9   ri   rj   ThreadStatusr;   r   zdict[str, list[Interrupt]]
interruptszNotRequired[dict[str, Any]]	extractedNrJ   rP   rQ   rR   rz   rz   "  s         ++NNN*/NNNILLL*****J****[[rQ   rz   c                  Z    e Zd ZU dZded<   ded<   ded<   ded<   d	ed
<   ded<   ded<   dS )
ThreadTaskz"Represents a task within a thread.rV   rx   r7   rX   r   list[Interrupt]r|   Checkpoint | None
checkpointzThreadState | NonestaterZ   resultNrJ   rP   rQ   rR   r   r   7  sf         ,,GGGIII!!!!!!!!!!rQ   r   c                  r    e Zd ZU dZded<   	 ded<   	 ded<   	 ded	<   	 d
ed<   	 ded<   	 ded<   	 ded<   dS )ThreadStatez!Represents the state of a thread.zlist[dict] | dict[str, Any]r   zSequence[str]rt   rU   r   ri   rj   rX   r8   r   parent_checkpointzSequence[ThreadTask]r    r   r|   NrJ   rP   rQ   rR   r   r   C  s         ++''''#NNN!%((((ST66rQ   r   c                      e Zd ZU dZded<   dS )ThreadUpdateStateResponsez7Represents the response from updating a thread's state.rU   r   NrJ   rP   rQ   rR   r   r   Y  s$         AA))rQ   r   c                  r    e Zd ZU dZded<   	 ded<   	 ded<   	 ded<   	 ded<   	 d	ed
<   	 ded<   	 ded<   dS )Runz"Represents a single execution run.rV   run_idr:   r5   r   r8   r9   	RunStatusr;   ri   rj   MultitaskStrategymultitask_strategyNrJ   rP   rQ   rR   r   r   `  s         ,,KKKNNN3',kNNN))))@@rQ   r   c                      e Zd ZU dZded<   	 ded<   	 ded<   	 ded<   	 d	ed
<   	 ded<   	 ded<   	 ded<   	 ded<   	 ded<   	 ded<   	 d	ed<   	 ded<   	 ded<   dS )CronzRepresents a scheduled task.rV   r=   r5   rX   r:   zOnCompletionBehavior | Noneon_run_completedzdatetime | Noner?   scheduletimezoner   r8   r9   dictpayloaduser_idr>   rj   boolenabledNrJ   rP   rQ   rR   r   r   u  s         &&LLL"1111b0MMM+s(-MMM6"""""(NNN#MMM&&rQ   r   c                      e Zd ZU dZded<   	 ded<   	 ded<   	 ded<   	 d	ed
<   	 ded<   	 ded<   	 ded<   	 ded<   	 ded<   	 ded<   	 ded<   	 ded<   	 ded<   	 ded<   	 ded<   dS )
CronUpdatez9Payload for updating a cron job. All fields are optional.rV   r   r   r   r?   InputinputrH   rj   rC   rf   rg   rh   webhookzAll | list[str]interrupt_beforeinterrupt_afterOnCompletionBehaviorr   r   r   zStreamMode | list[StreamMode]stream_modestream_subgraphsstream_resumable
Durability
durabilityNrJ   rP   rQ   rR   r   r     s         CCMMM3MMMH0LLL!2NNN.0LLL;%%%%B$$$$A****=MMM)....$2SRRrQ   r   )
r5   r6   r7   rl   rf   rh   r8   r9   rj   rk   )	r:   r8   r9   rj   rf   rh   r;   r   r|   )	r   r:   r5   r8   r9   r;   rj   kwargsr   )r=   r5   r:   r?   r   r   r8   r9   r   r   r>   rj   nowr   r   .c                      e Zd ZU dZded<   	 ded<   	 ded<   	 ded<   	 d	ed
<   	 ded<   	 ded<   	 ded<   	 ded<   	 ded<   	 ded<   dS )	RunCreatez7Defines the parameters for initiating a background run.rX   r:   rV   r5   r^   r   rj   zConfig | Nonerf   zContext | Nonerh   rY   zlist[str] | Noner   r   r   zMultitaskStrategy | Noner   NrJ   rP   rQ   rR   r   r      s         AAU>)8,(8&&&&;%%%%:E0000CCrQ   r   c                  N    e Zd ZU dZded<   	 ded<   	 ded<   	 ded	<   	 ded
<   dS )ItemzyRepresents a single document or data entry in the graph's Store.

    Items are used to store cross-thread memories.
    rD   	namespacerV   keyrH   rw   r   r8   r9   NrJ   rP   rQ   rR   r   r     sg          
 XHHH D277rQ   r   c                      e Zd ZU dZded<   dS )ListNamespaceResponsez*Response structure for listing namespaces.zlist[list[str]]
namespacesNrJ   rP   rQ   rR   r   r   0  s$         44JJrQ   r   c                      e Zd ZU dZded<   dS )
SearchItemzItem with an optional relevance score from search operations.

    Attributes:
        score (Optional[float]): Relevance/similarity score. Included when
            searching a compatible store with a natural language query.
    zfloat | NonescoreNrJ   rP   rQ   rR   r   r   7  s*           rQ   r   c                      e Zd ZU dZded<   dS )SearchItemsResponsez'Response structure for searching items.zlist[SearchItem]itemsNrJ   rP   rQ   rR   r   r   B  s$         1177rQ   r   c                  :    e Zd ZU dZded<   	 ded<   	 dZded<   dS )	
StreamPartz'Represents a part of a stream response.rV   eventr   dataNrX   rx   )rK   rL   rM   rN   rO   rx   rP   rQ   rR   r   r   I  sA         11JJJ1JJJ5BrQ   r   )v1v2c                  B    e Zd ZU dZded<   	 ded<   	 ded<   	 ded<   d	S )
TaskPayloadzPayload for a task start event.rV   rx   r7   r   r   rD   triggersNrJ   rP   rQ   rR   r   r   _  sH         ))GGG*III*JJJ(VVrQ   r   c                  N    e Zd ZU dZded<   	 ded<   	 ded<   	 ded<   	 d	ed
<   dS )TaskResultPayloadz Payload for a task result event.rV   rx   r7   rX   r   list[dict[str, Any]]r|   rH   r   NrJ   rP   rQ   rR   r   r   l  sZ         **GGG*III-=$$$$AFFrQ   r   c                  Z    e Zd ZU dZded<   	 ded<   	 ded<   	 ded<   	 d	ed
<   	 ded<   dS )CheckpointTaskPayloada  A task entry within a `CheckpointPayload`.

    The keys present depend on the task's state:

    - **Error:** `id`, `name`, `error`, `state`
    - **Has result:** `id`, `name`, `result`, `interrupts`, `state`
    - **Pending:** `id`, `name`, `interrupts`, `state`
    rV   rx   r7   zNotRequired[str]r   zNotRequired[Any]r   z!NotRequired[list[dict[str, Any]]]r|   rZ   r   NrJ   rP   rQ   rR   r   r   {  so           GGG*III*9N1111V    CCrQ   r   c                  Z    e Zd ZU dZded<   	 ded<   	 ded<   	 ded<   	 ded	<   	 d
ed<   dS )CheckpointPayloadzPayload for a checkpoint event.rZ   rf   rH   rj   r   rD   rt   parent_configzlist[CheckpointTaskPayload]r    NrJ   rP   rQ   rR   r   r     sl         ))!!!![V>OOO7((((\&&&&88rQ   r   c                  >    e Zd ZU ded<   	 ded<   	 ded<   	 ded<   d	S )
_DebugCheckpointPayloadrF   steprV   	timestampzLiteral['checkpoint']typer   r   NrK   rL   rM   rO   rP   rQ   rR   r   r     sE         III1NNN9:!!rQ   r   c                  >    e Zd ZU ded<   	 ded<   	 ded<   	 ded<   d	S )
_DebugTaskPayloadrF   r   rV   r   zLiteral['task']r   r   r   Nr   rP   rQ   rR   r   r     sE         III1NNN94!!rQ   r   c                  >    e Zd ZU ded<   	 ded<   	 ded<   	 ded<   d	S )
_DebugTaskResultPayloadrF   r   rV   r   zLiteral['task_result']r   r   r   Nr   rP   rQ   rR   r   r     sE         III1NNN9    ;""rQ   r   c                      e Zd ZU dZded<   dS )RunMetadataPayloadz)Payload for the `metadata` control event.rV   r   NrJ   rP   rQ   rR   r   r     s!         33KKK++rQ   r   c                  B    e Zd ZU dZded<   	 ded<   	 ded<   	 ded	<   d
S )ValuesStreamPartz/Stream part emitted for `stream_mode="values"`.zLiteral['values']r   rD   nsrH   r   r   r|   NrJ   rP   rQ   rR   r   r     sN         99)MMME+$$$$<<rQ   r   c                  6    e Zd ZU dZded<   	 ded<   	 ded<   dS )	UpdatesStreamPartz0Stream part emitted for `stream_mode="updates"`.zLiteral['updates']r   rD   r   rH   r   NrJ   rP   rQ   rR   r   r     s?         ::)MMME11rQ   r   c                  6    e Zd ZU dZded<   	 ded<   	 ded<   dS )	MessagesPartialStreamPartzDStream part emitted for partial message chunks (`messages/partial`).zLiteral['messages/partial']r   rD   r   r   r   NrJ   rP   rQ   rR   r   r     s?         NN%%%%)MMME..rQ   r   c                  6    e Zd ZU dZded<   	 ded<   	 ded<   dS )	MessagesCompleteStreamPartz@Stream part emitted for complete messages (`messages/complete`).zLiteral['messages/complete']r   rD   r   r   r   NrJ   rP   rQ   rR   r   r     s?         JJ&&&&)MMME))rQ   r   c                  6    e Zd ZU dZded<   	 ded<   	 ded<   dS )	MessagesMetadataStreamPartz?Stream part emitted for message metadata (`messages/metadata`).zLiteral['messages/metadata']r   rD   r   rH   r   NrJ   rP   rQ   rR   r   r     s?         II&&&&)MMMERRrQ   r   c                  6    e Zd ZU dZded<   	 ded<   	 ded<   dS )	MessagesTupleStreamPartzMStream part emitted for `stream_mode="messages"` (raw message+metadata pair).zLiteral['messages']r   rD   r   r   r   NrJ   rP   rQ   rR   r   r     s?         WW)MMME>>rQ   r   c                  6    e Zd ZU dZded<   	 ded<   	 ded<   dS )	CustomStreamPartz/Stream part emitted for `stream_mode="custom"`.zLiteral['custom']r   rD   r   r   r   NrJ   rP   rQ   rR   r   r     s<         99)MMMEIIICCrQ   r   c                  6    e Zd ZU dZded<   	 ded<   	 ded<   dS )	CheckpointsStreamPartz4Stream part emitted for `stream_mode="checkpoints"`.zLiteral['checkpoints']r   rD   r   r   r   NrJ   rP   rQ   rR   r   r   "  s?         >>    )MMME!!rQ   r   c                  6    e Zd ZU dZded<   	 ded<   	 ded<   dS )	TasksStreamPartz.Stream part emitted for `stream_mode="tasks"`.zLiteral['tasks']r   rD   r   zTaskPayload | TaskResultPayloadr   NrJ   rP   rQ   rR   r   r   -  s?         88)MMME)))),,rQ   r   c                  6    e Zd ZU dZded<   	 ded<   	 ded<   dS )	DebugStreamPartz.Stream part emitted for `stream_mode="debug"`.zLiteral['debug']r   rD   r   DebugPayloadr   NrJ   rP   rQ   rR   r   r   8  s?         88)MMME""rQ   r   c                  6    e Zd ZU dZded<   	 ded<   	 ded<   dS )	MetadataStreamPartz3Control event with `run_id` and other run metadata.zLiteral['metadata']r   rD   r   r   r   NrJ   rP   rQ   rR   r   r   C  s?         ==)MMM0##rQ   r   c                  *    e Zd ZU dZded<   	 ded<   dS )SendzRepresents a message to be sent to a specific node in the graph.

    This type is used to explicitly send messages to nodes in the graph, typically
    used within Command objects to control graph execution flow.
    rV   noderZ   r   NrJ   rP   rQ   rR   r   r   a  s:           III=    7 7rQ   r   c                  6    e Zd ZU dZded<   	 ded<   	 ded<   dS )	Commanda  Represents one or more commands to control graph execution flow and state.

    This type defines the control commands that can be returned by nodes to influence
    graph execution. It lets you navigate to other nodes, update graph state,
    and resume from interruptions.
    z!Send | str | Sequence[Send | str]gotoz*dict[str, Any] | Sequence[tuple[str, Any]]updater   resumeNrJ   rP   rQ   rR   r   r   p  sQ           ,+++ 7666
 KKK rQ   r   c                  *    e Zd ZU dZded<   	 ded<   dS )RunCreateMetadataz$Metadata for a run creation request.rV   r   rX   r:   NrJ   rP   rQ   rR   r   r     s0         ..KKKrQ   r   c                  (    e Zd ZU dZded<   ded<   dS )_TypedDictLikeV1zbProtocol to represent types that behave like TypedDicts

    Version 1: using `ClassVar` for keys.zClassVar[frozenset[str]]__required_keys____optional_keys__NrJ   rP   rQ   rR   r   r     s6         - - 0/////////rQ   r   c                  (    e Zd ZU dZded<   ded<   dS )_TypedDictLikeV2zfProtocol to represent types that behave like TypedDicts

    Version 2: not using `ClassVar` for keys.zfrozenset[str]r   r   NrJ   rP   rQ   rR   r   r     s6         1 1 &%%%%%%%%%rQ   r   c                      e Zd ZU dZded<   dS )_DataclassLikezProtocol to represent types that behave like dataclasses.

    Inspired by the private _DataclassT from dataclasses that uses a similar protocol as a bound.
    zClassVar[dict[str, Field[Any]]]__dataclass_fields__NrJ   rP   rQ   rR   r   r     s*          
 :99999rQ   r   c                  0    e Zd ZU dZded<   ded<   dd
ZdS )_BaseModelLikezBProtocol to represent types that behave like Pydantic `BaseModel`.zClassVar[dict[str, Any]]model_configzClassVar[Any]__pydantic_core_schema__r   r   returnrH   c                    d S )NrP   )selfr   s     rR   
model_dumpz_BaseModelLike.model_dump  s	     rQ   N)r   r   r  rH   )rK   rL   rM   rN   rO   r  rP   rQ   rR   r  r    sH         LL****++++     rQ   r  r   	_JSONLike_JSONMapr   rg   )mrN   
__future__r   collections.abcr   r   dataclassesr   r   typingr   r	   r
   r   r   r   r   typing_extensionsr   r   r   rV   ri   r   r{   ThreadStreamMode
StreamModeDisconnectModer   OnConflictBehaviorr   r   AllIfNotExistsPruneStrategyCancelActionBulkCancelRunsStatusAssistantSortByThreadSortBy
CronSortBy	SortOrderrC   rU   r]   	Subgraphsre   rn   rp   rr   rv   rz   r   r   r   r   r   r   AssistantSelectFieldThreadSelectFieldRunSelectFieldCronSelectFieldrF   floatr   PrimitiveDatalisttuplebytesQueryParamTypesr   r   r   r   r   r   StreamVersionr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   StreamPartV2r   r   r   r   r   r   r  r  rO   r	  r   rg   rP   rQ   rR   <module>r)     s   9 9 9 " " " " " " - - - - - - - -                              5 4 4 4 4 4 4 4CH~ jVW	 => CD  


 -. HI  23  /0  ,-
@
 cl .() /0 ./ :;  B I 
 M"	
    Ye    .
C 
C 
C 
C 
C 
C 
C 
C= = = = =) = = =, k!"	+ + + + +I + + +.	 	 	 	 	} 	 	 	3 3 3 3 3 3 3 3L L L L Ly L L LH H H H H	 H H H\ \ \ \ \Y \ \ \*	" 	" 	" 	" 	" 	" 	" 	"7 7 7 7 7) 7 7 7,* * * * *	 * * *A A A A A) A A A*' ' ' ' '9 ' ' 'B"S "S "S "S "S% "S "S "S "SP 	  
  
 $ c	E!D(4/ C-!8889
5m#$%&E#}$%s*+, 

 	 D D D D D	 D D D68 8 8 8 89 8 8 8*K K K K KI K K K    U    8 8 8 8 8) 8 8 8        
#
W 
W 
W 
W 
W) 
W 
W 
WG G G G G	 G G GD D D D DI D D D09 9 9 9 9	 9 9 9"" " " " "i " " "" " " " "	 " " "# # # # #i # # # ')::=TT ?, , , , , , , ,
= 
= 
= 
= 
=y 
= 
= 
=2 2 2 2 2	 2 2 2/ / / / /	 / / /* * * * * * * *S S S S S S S S? ? ? ? ?i ? ? ?D D D D Dy D D D" " " " "I " " "- - - - -i - - -# # # # #i # # #$ $ $ $ $ $ $ $   !! !	!
     	 
 7 7 7 7 79 7 7 7    iu    6         	      0 0 0 0 0x 0 0 0& & & & &x & & &: : : : :X : : :	 	 	 	 	X 	 	 	 czC'%/$6	 6 6 6 6y$y/:tJ7GG	HH    
 ''.8>IHT            rQ   