i @ddlZddlmZerddlmZmZddlmZddlZddlm Z GddZ e jdge jd d ge jd d ge jd dgiZiZej#D]\ZZeD]Zeee< ddZddZdZdZdZy)N) TYPE_CHECKING)AnyCallable)Span)loggerceZdZdZdZdZdZy)GEN_AI_ALLOWED_MESSAGE_ROLESsystemuser assistanttoolN)__name__ __module__ __qualname__SYSTEMUSER ASSISTANTTOOLs/var/www/html/hwPaymentPortal-be-dev/hw-payment-portal-api/venv/lib/python3.12/site-packages/sentry_sdk/ai/utils.pyr r s F DI Drr r r humanr air tool_callc rt|dr t|jSt |tr4rt|dk(rt|dStfd|DSt |tr/|jDcic]\}}|t|c}}St |ttttfr|St|S#t$rM}t j d|t |ttttfr|n t|cYd}~Sd}~wwxYwcc}}w)N model_dumpunpackz+Could not convert pydantic data to JSON: %src38K|]}t|yw)rN)_normalize_data).0xrs r z"_normalize_data...sD!OAf55Ds)hasattrr!r Exceptionrwarning isinstanceintfloatboolstrlistlendictitems)datarekvs ` rr!r!!st\" T"4??#4VD D $ c$i1n"476: :DtDDD$CG::<P!Q?1V44PPdS%s$;<4K#d)K T NNH! L%dS%s,CD4#d) S TQs$CD3 D0#AD+%D0+D0ct||}t|tttt fr|j ||y|j |tj|y)Nr) r!r(r)r*r+r,set_datajsondumps)spankeyvaluer normalizeds rset_data_normalizedr=6sF v6J*sE456 c:& c4::j12rc.tj||S)z Normalize a message role to one of the 4 allowed gen_ai role values. Maps "ai" -> "assistant" and keeps other standard roles unchanged. )GEN_AI_MESSAGE_ROLE_MAPPINGget)roles rnormalize_message_rolerB?s ' * *4 66rcg}|D]Z}t|ts|j|%|j}d|vrt |d|d<|j|\|S)z Normalize roles in a list of messages to use standard gen_ai role values. Creates a deep copy to avoid modifying the original messages. rA)r(r/appendcopyrB)messagesnormalized_messagesmessagenormalized_messages rnormalize_message_rolesrJHss 7'4(  & &w / $\\^ W )?)P v &""#567 rctj}|duxr|jdu}|rtjStjS)N) sentry_sdkget_current_spancontaining_transaction start_spanstart_transaction) current_spantransaction_existss rget_start_span_functionrS[sI..0LD T\%H%HPT%T%7: XJr[s $'#!''( %%'8 **[$,? %% '< '#!!D!J!J!L?K#? 3>#K0?? L*37&Yr