uPj(dZddlZddlmZdZdZy)zQ Utility functions for cleaning up async HTTP clients to prevent resource leaks. N)FinalcKddl}ddlm}t|jdi}|j D]\}}t ||r&t|dr |jd{9t|ds t|drt|dr |jn |j}t|dr9t|jd r# |jjd{t|d s|jr |jd{t|d s |jd{t|d rAt|d d}t ||r't|dr |jd{yyyy72#t$rYtwxYw7#t$rYwxYw7#t$rYwxYw7#t$rYwxYw7P#t$rYywxYww) a) Close all cached async HTTP clients to prevent resource leaks. This function iterates through all cached clients in litellm's in-memory cache and closes any aiohttp client sessions that are still open. Also closes the global base_llm_aiohttp_handler instance (issue #12443). rN)BaseLLMAIOHTTPHandler cache_dictclose_clientclient _transportaclosebase_llm_aiohttp_handler)litellm)litellm.llms.custom_httpx.aiohttp_handlerrgetattrin_memory_llm_clients_cacheitems isinstancehasattrr Exceptionrr r r is_closed)r rrkeyhandlerr base_handlers s/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/custom_httpx/async_client_cleanup.pyclose_litellm_async_clientsr sO C C\SUVJ"((*! W g4 5''7:S mmo%% Wi (GGX,F(/(CW__Fv|,9J9JH1U ++22444 vx(1A1A --/)) Wh ' nn&&&=!Jw23%g/I4P l$9 :w|U\?] "((***@^ :4C&  5  * '  +  sAG2F'F(F,AG2 F.)F,*F.. G2; G2 F?F=F?! G20GGG4G2=G#G!G#G2F F)%G2(F))G2,F.. F:7G29F::G2=F?? G G2 G  G2G GG2GG2!G## G/,G2.G//G2c4ddl}d}|j|y)z Register the async client cleanup function to run at exit. This ensures that all async HTTP clients are properly closed when the program exits. rNc tj}tj| |jt |j y#|j wxYw#t $rYywxYw)z Cleanup wrapper that creates a fresh event loop for atexit cleanup. At exit time, the main event loop is often already closed. Creating a new event loop ensures cleanup runs successfully (fixes issue #12443). N)asyncionew_event_loopset_event_looprun_until_completerrr)loops rcleanup_wrapperz6register_async_client_cleanup..cleanup_wrapperNsb "002D  " "4 ( ''(C(EF     s()A+AA+A((A++ A76A7)atexitregister)r#r"s rregister_async_client_cleanupr%Fs * OOO$)__doc__rtypingrrr%r&rr*s:z%r&