g ddlmZddlmZddlmZddlmZmZddl m Z m Z m Z  ddl mZmZej aej addZddZdd Zdd Zdd Zdd Zy#e$rdd ZYwxYw)) annotations)asdict)partial)AnyCallable)ContextCarrierattach_context get_context)ProcessPoolExecutorThreadPoolExecutorctjturtt_tjt urt t_yy)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_p]/var/www/openai/venv/lib/python3.12/site-packages/logfire/_internal/integrations/executors.pyinstrument_executorsrs7  $ $ 5(0  %  % % 6)1  & 7rc\t|g|i|}t}t|t||dS)z[A wrapper around ThreadPoolExecutor.submit() that carries over OTEL context across threads.Ncarrierfn parent_config)rr r_run_with_contextsrargskwargsrs rrrs2 R )$ )& )-Q 17rY]^^rclt|g|i|}t}t|t||t S)z^A wrapper around ProcessPoolExecutor.submit() that carries over OTEL context across processes.r)rr rrserialize_configrs rrr!s5 R )$ )& )-Q 17rYiYkllrcl| t|t|5|cdddS#1swYyxYw)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_configr rs rrr's+  $ } - G $4% $ $s*3cy)Nrrrrrr4s rc$ddlm}t|S)N GLOBAL_CONFIG)configr)rr(s rr"r"8s& -  rc>ddlm}m}|js |di|yy)Nr')r) configurer)r*r)r, _initialized)r*r)r,s rr$r$As1  % %F &rN)returnNone)rr rCallable[..., Any]rrr r)rr rr0rrr r)rrrzCallable[[], Any]rzdict[str, Any] | Noner.r)r.dict[str, Any])r*r1r.r/) __future__r dataclassesr functoolsrtypingrrlogfire.propagaterr r concurrent.futuresr r rrrrrrr ImportErrorr"r$rrrr9sn" II, K&--M'..M2_ m "!   s0A A+*A+