
    KKik
                    r    S SK Jr  S SKrS SKJr  S SKJrJrJr  S SK	J
r
  S SKJr  S SKJr   " S S	\
5      rg)
    )annotationsN)AsyncIterator)AnyLiteralcast)AsyncCallbackHandler)	LLMResult)overridec                      \ rS rSr% SrS\S'   S\S'   \SS j5       rSS jr\	        SS	 j5       r
\	SS
 j5       r\	SS j5       r\	SS j5       rSS jrSrg)AsyncIteratorCallbackHandler   z0Callback handler that returns an async iterator.zasyncio.Queue[str]queuezasyncio.Eventdonec                    g)zAlways verbose.T selfs    m/var/www/html/dynamic-report/venv/lib/python3.13/site-packages/langchain_classic/callbacks/streaming_aiter.pyalways_verbose+AsyncIteratorCallbackHandler.always_verbose   s         c                l    [         R                  " 5       U l        [         R                  " 5       U l        g)z)Instantiate AsyncIteratorCallbackHandler.N)asyncioQueuer   Eventr   r   s    r   __init__%AsyncIteratorCallbackHandler.__init__   s    ]]_
MMO	r   c                @   #    U R                   R                  5         g 7fN)r   clear)r   
serializedpromptskwargss       r   on_llm_start)AsyncIteratorCallbackHandler.on_llm_start   s      			   c                X   #    Ub#  US:w  a  U R                   R                  U5        g g g 7f)N )r   
put_nowait)r   tokenr#   s      r   on_llm_new_token-AsyncIteratorCallbackHandler.on_llm_new_token)   s+     "JJ!!%( "-s   (*c                @   #    U R                   R                  5         g 7fr   r   set)r   responser#   s      r   
on_llm_end'AsyncIteratorCallbackHandler.on_llm_end.        		r&   c                @   #    U R                   R                  5         g 7fr   r.   )r   errorr#   s      r   on_llm_error)AsyncIteratorCallbackHandler.on_llm_error2   r3   r&   c                 #    U R                   R                  5       (       a   U R                  R                  5       (       Gd$  [        R
                  " [        R                  " U R                   R                  5       5      [        R                  " U R                  R                  5       5      /[        R                  S9I Sh  vN u  pU(       a  UR                  5       R                  5         [        SUR                  5       R                  5       5      nUSL a  gU7v   U R                   R                  5       (       d  GM  U R                  R                  5       (       d  GM#  gg N7f)z)Asynchronous iterator that yields tokens.)return_whenNzstr | Literal[True]T)r   emptyr   is_setr   waitensure_futuregetFIRST_COMPLETEDpopcancelr   result)r   r   othertoken_or_dones       r   aiter"AsyncIteratorCallbackHandler.aiter8   s     **""$$DII,<,<,>,> !( ))$**..*:;))$))..*:;	 $33! KD 		""$ !!6
8I8I8KLM $  3 **""$$DII,<,<,>,>s   C E)E'A=E)E)%E))r   r   N)returnbool)rG   None)r!   zdict[str, Any]r"   z	list[str]r#   r   rG   rI   )r*   strr#   r   rG   rI   )r0   r	   r#   r   rG   rI   )r5   BaseExceptionr#   r   rG   rI   )rG   zAsyncIterator[str])__name__
__module____qualname____firstlineno____doc____annotations__propertyr   r   r
   r$   r+   r1   r6   rE   __static_attributes__r   r   r   r   r      s    :
 $
 "  	
 
  ) )    
 r   r   )
__future__r   r   collections.abcr   typingr   r   r   langchain_core.callbacksr   langchain_core.outputsr	   typing_extensionsr
   r   r   r   r   <module>rZ      s,    "  ) % % 9 , &
E #7 E r   