o ưi @sUddlZddlZddlZddlmZmZmZmZmZm Z dZ ee e d<ddZ dee e ffdd Zd e dee fd d Z dd e d ee dee dee fddZddZdee efdee effddZdeee efde eee efeee efffddZdS)N)AnyDictFinalListOptionalTuplez~/.opik.configCONFIG_FILE_PATH_DEFAULTc Cst}gd}d}t||\}}t|d>|\}}t|d>|\}}|dO}||dkrH||dkrH||dkrH|d d krG|d d7<n |||df|dd<d |d B} td } |d d|dd|dd| dd|  S)N)rrrrl Ys ipri?iz>08x-z>04x)timetime_nsdivmodosurandomhex) nslastZ sixteen_secst1Zrest1t2Zrest2t3_Zt4ZrandrV/home/app/Keep/.python/lib/python3.10/site-packages/litellm/integrations/opik/utils.py create_uuid7 s$   ,rreturncsHtjt}t|fddD}d|vr"|dSiS)Ncsi|] }|t|qSr)dictitems).0sectionconfigrr .sz*_read_opik_config_file..Zopik)rpath expanduserr configparser ConfigParserreadsections)Z config_path config_valuesrr$r_read_opik_config_file(s   r.keycCsd}t||dS)NZopik_)rgetenvupper)r/Z env_prefixrrr_get_env_variable8sr2 user_value default_valuecCs:|dur|St|}|dur|St}||vr||S|S)z Get the configuration value of a variable, order priority is: 1. user provided value 2. environment variable 3. Opik configuration file 4. default value N)r2r.)r/r3r4Z env_valuer-rrrget_opik_config_variable=s r5cCsDi}|jdur |j|d<|jdur|j|d<|jdur |j|d<|S)Ncompletion_tokens prompt_tokens total_tokens)r6r7r8)usageZ usage_dictrrrcreate_usage_objectZs      r:xcCsdd|DS)zRemove None values from dict.cSsi|] \}}|dur||qS)Nr)r"kvrrrr&hsz!_remove_nulls..)r!)r;rrr _remove_nullsfsr>payloadcCs$dd|D}dd|D}||fS)aL Separate traces and spans from payload. Traces are identified by not having a "type" field. Spans are identified by having a "type" field. Args: payload: List of dicts containing trace and span data Returns: Tuple of (traces, spans) where both are lists of dicts with null values removed cSsg|] }d|vrt|qStyper>r"r;rrr zz5get_traces_and_spans_from_payload..cSsg|] }d|vrt|qSr@rBrCrrrrD{rEr)r?Ztracesspansrrr!get_traces_and_spans_from_payloadksrG)NN)r)rrtypingrrrrrrrstr__annotations__rr.r2r5r:r>rGrrrrs2   " "