o cBg@s<ddlZddlZddlZddlmZddlmZddlm Z ddl m Z m Z m Z mZmZmZmZmZmZddlmZddlmZddlmZdd lmZdd lmZmZed Zed Z Gd ddeee fZ!GdddeZ"Gddde Z#dej$dedde fde fddZ%de ee fdej&dej'de fddZ(dS)N) ExitStack) copy_context) TracebackType) AsyncContextManager AwaitableCallableContextManager CoroutineOptionalProtocolTypeVarcast)RunnableConfig)get_executor_for_config) ParamSpec) GraphBubbleUp)CONTEXT_NOT_SUPPORTEDrun_coroutine_threadsafePTc@sZeZdZddddddeeefdejdeede d e d e d ej d e j j efd dZdS)SubmitNFT__name____cancel_on_exit____reraise_on_exit__ __next_tick__fnargsrrrrkwargsreturncOsdSN)selfrrrrrrrr!r!`/var/www/html/development/chatbot/venv/lib/python3.10/site-packages/langgraph/pregel/executor.py__call__s zSubmit.__call__)r __module__ __qualname__rrrrr strboolr concurrentfuturesFuturer$r!r!r!r#rs,  rc@seZdZdZdeddfddZddddd d eeefd ej d e e d e de de dej dejjefddZdejjddfddZdefddZde eede ede ede e fddZdS)BackgroundExecutoraIA context manager that runs sync tasks in the background. Uses a thread pool executor to delegate tasks to separate threads. On exit, - cancels any (not yet started) tasks with `__cancel_on_exit__=True` - waits for all tasks to finish - re-raises the first exception from tasks with `__reraise_on_exit__=True`configrNcCs$t|_|jt||_i|_dSr )rstack enter_contextrexecutortasks)r"r-r!r!r#__init__3s zBackgroundExecutor.__init__FTrrrrrrrrc Osvt}|rttjjt|jjt|j |g|Ri|} n|jj|j |g|Ri|} ||f|j | <| |j | Sr ) rr r)r*r+rr0submit next_tickrunr1add_done_callbackdone) r"rrrrrrrctxtaskr!r!r#r39s   zBackgroundExecutor.submitr9cCsPz|Wnty|j|YdStyYdSw|j|dS)z3Remove the task from the tasks dict when it's done.N)resultrr1pop BaseException)r"r9r!r!r#r7Ps   zBackgroundExecutor.donecCs|jSr r3r"r!r!r# __enter__]szBackgroundExecutor.__enter__exc_type exc_value tracebackc Cs|j}|D] \}\}}|r|q dd|D}r%tj||j||||durP|D]\}\}} | s>q5z| Wq5tjj yOYq5wdSdS)NcSsh|]}|s|qSr!)r7).0tr!r!r# msz.BackgroundExecutor.__exit__..) r1copyitemscancelr)r*waitr.__exit__r:CancelledError) r"r@rArBr1r9rH_pendingreraiser!r!r#rJ`s&   zBackgroundExecutor.__exit__)rr%r&__doc__rr2rrrrr r'r(rr)r*r+r3r7rr?typer<rrJr!r!r!r#r,+sF     r,c@seZdZdZdeddfddZddddd d eeee fd ej d e e d e de de dejdeje fddZdejddfddZdefddZde eede ede eddfddZdS)AsyncBackgroundExecutoraSA context manager that runs async tasks in the background. Uses the current event loop to delegate tasks to asyncio tasks. On exit, - cancels any tasks with `__cancel_on_exit__=True` - waits for all tasks to finish - re-raises the first exception from tasks with `__reraise_on_exit__=True` ignoring CancelledErrorr-rNcCs@i|_t|_t|_|d}rt||_dSd|_dS)Nmax_concurrency) r1objectsentinelasyncioget_running_looploopget Semaphore semaphore)r"r-rRr!r!r#r2s  z AsyncBackgroundExecutor.__init__FTrrrrrrrrc Os|ttddtf||i|}|jrt|j|}tr$t||j||d} n t||j|t|d} ||f|j | <| |j | S)N)namelazy)r[contextr\) r r rrZgatedrrrWrr1r6r7) r"rrrrrrrcoror9r!r!r#r3s"    zAsyncBackgroundExecutor.submitr9cCsdz |}rt|tr|j|WdSWdS|j|WdStjy1|j|YdSwr ) exception isinstancerr1r;rUrK)r"r9excr!r!r#r7s  zAsyncBackgroundExecutor.donecs|jSr r=r>r!r!r# __aenter__sz"AsyncBackgroundExecutor.__aenter__r@rArBc s|j}|D]\}\}}|r||jq |r#t|IdH|durI|D]\}\}}|s4q+z |} r=| Wq+tjyHYq+wdSdSr ) r1rFrGrHrTrUrIr`rK) r"r@rArBr1r9rHrLrNrbr!r!r# __aexit__s*   z!AsyncBackgroundExecutor.__aexit__)rr%r&rOrr2rrrrrr r'r(rrUr+r3r7rrcrPr<rrdr!r!r!r#rQ}sF   rQrZr_rc sH|4IdH|IdHWdIdHS1IdHswYdS)zHA coroutine that waits for a semaphore before running another coroutine.Nr!)rZr_r!r!r#r^s0r^rrrcOstd||i|S)zPA function that yields control to other threads before running another function.r)timesleep)rrrr!r!r#r4s r4))rUconcurrent.futuresr)re contextlibr contextvarsrtypesrtypingrrrrr r r r r langchain_core.runnablesrlangchain_core.runnables.configrtyping_extensionsrlanggraph.errorsrlanggraph.utils.futurerrrrrr,rQrYr^rrr4r!r!r!r#s&   ,    R"\*