o i @sddlmZddlmZddlmZddlmZmZddl m Z m Z m Z z$ddl mZmZejaejad'd d Zd(ddZd)ddZd*ddZWneyVd'dd ZYnwd+d!d"Zd,d$d%Zd&S)-) annotations)asdict)partial)AnyCallable)ContextCarrierattach_context get_context)ProcessPoolExecutorThreadPoolExecutorreturnNonecCs(tjturtt_tjturtt_dSdS)zMonkey-patch `submit()` methods of `ThreadPoolExecutor` and `ProcessPoolExecutor` to carry over OTEL context across threads and processes. N)r submit submit_t_origsubmit_tr submit_p_origsubmit_prrn/var/www/html/karishye-ai-python/venv/lib/python3.10/site-packages/logfire/_internal/integrations/executors.pyinstrument_executorss   rsr fnCallable[..., Any]argsrkwargscOs.t|g|Ri|}t}t|t||ddS)z[A wrapper around ThreadPoolExecutor.submit() that carries over OTEL context across threads.Ncarrierfunc parent_config)rr r_run_with_contextrrrrrrrrrsrr cOs0t|g|Ri|}t}t|t||tdS)z^A wrapper around ProcessPoolExecutor.submit() that carries over OTEL context across processes.r)rr rrserialize_configr rrrr!srrrrCallable[[], Any]rdict[str, Any] | NonecCsB|durt|t| |WdS1swYdS)zA wrapper around a function that restores OTEL context from a carrier and then calls the function. This gets run from within a process / thread. N)deserialize_configrrrrrr's  $rcCsdS)Nrrrrrr6sdict[str, Any]cCsddlm}t|S)N GLOBAL_CONFIG)configr(rr'rrrr!:s r!r)cCs,ddlm}m}|js|di|dSdS)Nr&)r( configurer)r)r(r* _initialized)r)r(r*rrrr$Csr$N)r r )rr rrrrrr)rr rrrrrr)rrrr"rr#r r)r r%)r)r%r r ) __future__r dataclassesr functoolsrtypingrrlogfire.propagaterrr concurrent.futuresr r rrrrrrr ImportErrorr!r$rrrrs$