jndZddlZddlZddlZddlZddlmZddlmZm Z ddl m Z ddl mZddlmZded efd ZGd d ejZd e eeefd efdZded efdZdS)zUtility functions for Langfuse OpenTelemetry integration. This module provides utility functions for working with OpenTelemetry spans, including formatting and serialization of span data, and async execution helpers. N)sha256)Any Coroutine)trace)util) ReadableSpanspanreturnc|jrtj|jjnd}|jrt j|jnd}|jrt j|jnd}dt|j j j i}|j j r|j j |d<|j rUtj|j jtj|j jt!|j jdnd}t%j|jr|jnd}t%j|j|t|j||||||j||j||jt%j|j|d ddzS) N status_code description)trace_idspan_id trace_statez{}) namecontextkind parent_id start_timeend_timestatus attributeseventslinksresourceinstrumentationScope)indent ) parentotel_trace_apiformat_span_idr _start_timer ns_to_iso_str _end_timestr_statusr rr _contextformat_trace_idrreprrjsonloads_instrumentation_scopeto_jsondumps_namer_format_attributes _attributes_format_events_events _format_links_linksr)r rrrrrrs f/lsinfo/ai/hellotax_ai/data_center/backend/venv/lib/python3.11/site-packages/langfuse/_client/utils.pyspan_formatterr8s>BkS%dk&9:::t:>9IS#D$4555tJ59^Mt!$.111Hs4<3899F |9 $ 8} =  &6t}7MNN%4T]5JKK 9::     :151LV#++---RV  "DI&($ "55d6FGG--dl;;++DK88 Jt}'<'<'>'>??(<     " # cFeZdZdZdeeeefddffd ZddZxZS)_RunAsyncThreadzFHelper thread class for running async coroutines in a separate thread.coror Nc||_tj|_d|_d|_t dSN)r< contextvars copy_contextrresult exceptionsuper__init__)selfr< __class__s r7rDz_RunAsyncThread.__init__GsD "/11  +/ r9c |jtj|j|_dS#t $r}||_Yd}~dSd}~wwxYwr>)rrunasyncior<rA ExceptionrB)rEes r7rHz_RunAsyncThread.runNsZ ,**7; BBDKKK   DNNNNNNN s/3 AA  A)r N) __name__ __module__ __qualname____doc__rrrDrH __classcell__)rFs@r7r;r;DspPPYsC}5$r9r;r<c` tj}n$#t$rtj|cYSwxYw|r`|rLt |}|||jr|j|j Stj|S)a Safely run an async coroutine, handling existing event loops. This function detects if there's already a running event loop and uses a separate thread if needed to avoid the "asyncio.run() cannot be called from a running event loop" error. This is particularly useful in environments like Jupyter notebooks, FastAPI applications, or other async frameworks. Args: coro: The coroutine to run Returns: The result of the coroutine Raises: Any exception raised by the coroutine Example: ```python # Works in both sync and async contexts async def my_async_function(): await asyncio.sleep(1) return "done" result = run_async_safely(my_async_function()) ``` ) rIget_running_loop RuntimeErrorrH is_runningr;startjoinrBrA)r<loopthreads r7run_async_safelyrYUs6!')) !!!{4     ! !!! ! &&    #" "}{4   s 77valuect|dS)Nzutf-8)rencodedigesthex)rZs r7get_sha256_hash_hexr_s4 %,,w'' ( ( / / 1 1 5 5 7 77r9)rOrIr?r+ threadinghashlibrtypingrr opentelemetryrr!opentelemetry.sdkropentelemetry.sdk.tracerr&r8Threadr;rYr_r9r7rhsE  !!!!!!!!111111""""""000000..#....bi&"-!9S#s]3-!-!-!-!-!`8s8s888888r9