
    *jh              
      ,   d dl mZ d dlmZ d dlmZmZmZmZ d dl	m
Z
mZ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 d d	lmZmZmZmZ d
Z  G d deee
f         eeeeef                   Z!e"e"e#df         e#e
f         Z$ G d d          Z%dS )    )annotations)abstractmethod)AsyncIteratorCallableIteratorSequence)AnyGenericcast)RunnableRunnableConfig)Graph)Self)AllCommandStateSnapshotStateUpdate
StreamMode)ContextTInputTOutputTStateT)PregelProtocolStreamProtocolc                     e Zd Ze	 d=d>d            Ze	 d=d	d
d?d            Ze	 d=d	d
d?d            Zed	dd@d            Zed	dd@d            ZedddddAd            Z	edddddBd             Z
edCd#            ZedCd$            Ze	 d=dDd)            Ze	 d=dDd*            Ze	 d=ddddd	d+dEd6            Ze	 d=ddddd	d+dFd8            Ze	 d=dddd9dGd;            Ze	 d=dddd9dGd<            ZdS )Hr   NconfigRunnableConfig | Nonekwargsr	   returnr   c                    d S N )selfr   r   s      c/lsinfo/ai/hellotax_ai/base_platform/venv/lib/python3.11/site-packages/langgraph/pregel/protocol.pywith_configzPregelProtocol.with_config   s	     s    F)xrayr'   
int | boolDrawableGraphc                   d S r!   r"   r#   r   r'   s      r$   	get_graphzPregelProtocol.get_graph   s	     r&   c               
   K   d S r!   r"   r+   s      r$   
aget_graphzPregelProtocol.aget_graph   s       r&   )	subgraphsr   r/   boolr   c                   d S r!   r"   r#   r   r/   s      r$   	get_statezPregelProtocol.get_state'   s	     r&   c               
   K   d S r!   r"   r2   s      r$   
aget_statezPregelProtocol.aget_state,   s       r&   )filterbeforelimitr6   dict[str, Any] | Noner7   r8   
int | NoneIterator[StateSnapshot]c                   d S r!   r"   r#   r   r6   r7   r8   s        r$   get_state_historyz PregelProtocol.get_state_history1   s	     #&#r&   AsyncIterator[StateSnapshot]c                   d S r!   r"   r=   s        r$   aget_state_historyz!PregelProtocol.aget_state_history;   s	     (+sr&   updatesSequence[Sequence[StateUpdate]]c                    d S r!   r"   r#   r   rB   s      r$   bulk_update_statez PregelProtocol.bulk_update_stateE   s	    
 r&   c                
   K   d S r!   r"   rE   s      r$   abulk_update_statez!PregelProtocol.abulk_update_stateL   s      
 r&   valuesdict[str, Any] | Any | Noneas_node
str | Nonec                    d S r!   r"   r#   r   rI   rK   s       r$   update_statezPregelProtocol.update_stateS   s	     r&   c                
   K   d S r!   r"   rN   s       r$   aupdate_statezPregelProtocol.aupdate_state[   s       r&   )contextstream_modeinterrupt_beforeinterrupt_afterr/   inputInputT | Command | NonerR   ContextT | NonerS   $StreamMode | list[StreamMode] | NonerT   All | Sequence[str] | NonerU   Iterator[dict[str, Any] | Any]c                   d S r!   r"   r#   rV   r   rR   rS   rT   rU   r/   s           r$   streamzPregelProtocol.streamc   s	     *-r&   #AsyncIterator[dict[str, Any] | Any]c                   d S r!   r"   r]   s           r$   astreamzPregelProtocol.astreamp   s	     /2cr&   )rR   rT   rU   dict[str, Any] | Anyc                   d S r!   r"   r#   rV   r   rR   rT   rU   s         r$   invokezPregelProtocol.invoke}   s	      #sr&   c               
   K   d S r!   r"   rd   s         r$   ainvokezPregelProtocol.ainvoke   s        #sr&   r!   )r   r   r   r	   r   r   )r   r   r'   r(   r   r)   )r   r   r/   r0   r   r   )
r   r   r6   r9   r7   r   r8   r:   r   r;   )
r   r   r6   r9   r7   r   r8   r:   r   r?   )r   r   rB   rC   r   r   )r   r   rI   rJ   rK   rL   r   r   )rV   rW   r   r   rR   rX   rS   rY   rT   rZ   rU   rZ   r/   r0   r   r[   )rV   rW   r   r   rR   rX   rS   rY   rT   rZ   rU   rZ   r/   r0   r   r_   )rV   rW   r   r   rR   rX   rT   rZ   rU   rZ   r   rb   )__name__
__module____qualname__r   r%   r,   r.   r3   r5   r>   rA   rF   rH   rO   rQ   r^   ra   re   rg   r"   r&   r$   r   r      s       .2    ^  )- !	     ^  )- !	     ^ ;@     ^ ;@     ^ 
 )-(, & & & & & ^& 
 )-(, + + + + + ^+    ^    ^ 
 #	    ^ 
 #	    ^  )-
-
 $(<@7;6:
- 
- 
- 
- 
- ^
-  )-
2
 $(<@7;6:
2 
2 
2 
2 
2 ^
2  )-#
 $(7;6:# # # # # ^#  )-#
 $(7;6:# # # # # ^# # #r&   r   .c                  0    e Zd ZU dZded<   ded<   dd	Zd
S )r   )modes__call__set[StreamMode]rl   z#Callable[[Self, StreamChunk], None]rm   Callable[[StreamChunk], None]r   Nonec                p    t          t          t          t          gd f         |          | _        || _        d S r!   )r   r   r   StreamChunkrm   rl   )r#   rm   rl   s      r$   __init__zStreamProtocol.__init__   s/    
 Xt[&94&?@(KK


r&   N)rm   ro   rl   rn   r   rp   )rh   ri   rj   	__slots____annotations__rs   r"   r&   r$   r   r      sG         %I1111     r&   r   N)&
__future__r   abcr   collections.abcr   r   r   r   typingr	   r
   r   langchain_core.runnablesr   r   langchain_core.runnables.graphr   r)   typing_extensionsr   langgraph.typesr   r   r   r   r   langgraph.typingr   r   r   r   __all__r   tuplestrrr   r   r"   r&   r$   <module>r      s   " " " " " "       G G G G G G G G G G G G % % % % % % % % % % = = = = = = = = A A A A A A " " " " " " P P P P P P P P P P P P P P > > > > > > > > > > > >
.@# @# @# @# @#Xfck*GFHfg4U,V @# @# @#F E#s(OS#-.         r&   