o 0 i= @s$ddlZddlmZddlmZddlmZer.ddlmZmZm Z m Z m Z m Z ddl mZddlZddlmZdZGd d d Zejd gejd d gejddgejddgiZiZeD] \ZZeD]Zeee<qbq\d#ddZd#ddZddZ ddZ!ddZ"ddZ#efdd Z$efd!d"Z%dS)$N)deque) TYPE_CHECKING) getsizeof)AnyCallableDictListOptionalTuple)Span)loggeri Nc@seZdZdZdZdZdZdS)GEN_AI_ALLOWED_MESSAGE_ROLESsystemuser assistanttoolN)__name__ __module__ __qualname__SYSTEMUSER ASSISTANTTOOLrr_/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/sentry_sdk/ai/utils.pyr s r rrZhumanrZairZ tool_callTc st|dr6z t|dWSty5}ztd|t|ttt t fr'|nt |WYd}~Sd}~wwt|t rVrKt |dkrKt|ddSt fdd|DSt|t rffdd |DSt|ttt t frq|St |S) N model_dumpunpackz+Could not convert pydantic data to JSON: %src3s|] }t|dVqdS)rN_normalize_data).0xrrr 2sz"_normalize_data..csi|] \}}|t|dqS)rr)r!kvrrr 5sz#_normalize_data..)hasattrr r Exceptionr warning isinstanceintfloatboolstrlistlendictitems)datarerrrr %s  *  r cCsDt||d}t|ttttfr|||dS||t|dS)Nr) r r*r+r,r-r.set_datajsondumps)spankeyvaluer normalizedrrrset_data_normalized:s r<cCs t||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)rolerrrnormalize_message_roleCs r@cCsRg}|D]"}t|ts||q|}d|vr!t|d|d<||q|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. r?)r*r1appendcopyr@)messagesZnormalized_messagesmessageZnormalized_messagerrrnormalize_message_rolesLs   rEcCs*t}|duo |jdu}|rtjStjS)N) sentry_sdkZget_current_spanZcontaining_transactionZ start_spanZstart_transaction)Z current_spanZtransaction_existsrrrget_start_span_function_srGcCsZd}tt|dddD]}ttj||ddd}||7}||kr*|dSq dS)z Find the index of the first message that would exceed the max bytes limit. Compute the individual message sizes, and return the index of the first message from the back of the list that would exceed the max bytes limit. rr,: separatorsutf-8)ranger0r6r7encode)rC max_bytesZ running_sumidxsizerrr_find_truncation_indexhs rTcCsFtj|dd}t|d}||kr|dfSt||}||d|fS)NrIrLrNr)r6r7r0rPrT)rCrQZserialized_jsonZ current_sizeZtruncation_indexrrrtruncate_messages_by_sizeys  rUcCs2|sdSt||\}}|dkrt||j|j<|S)Nr)rUr0Z_gen_ai_original_message_countZspan_id)rCr8scoperQZtruncated_messages removed_countrrrtruncate_and_annotate_messagess rX)T)&r6 collectionsrtypingrsysrrrrrr r Zsentry_sdk.tracingr rFZsentry_sdk.utilsr ZMAX_GEN_AI_MESSAGE_BYTESr rrrrZ#GEN_AI_MESSAGE_ROLE_REVERSE_MAPPINGr=r2Z target_roleZ source_rolesZ source_roler r<r@rErGrTrUrXrrrrs<