j ddlmZmZmZmZmZddlZddlm Z ddl m Z dee efdee ded d ee effd Z d ee efdee d ee jded d ee eff dZy))DictAnySetListLiteralN)jsonable_encoder) anonymizer kwargs_dictfields_to_anonymize object_type)spantracereturnctj|}tjj s|Stjj }t ||||S)a Encodes and anonymizes the data in the given dictionary based on the specified fields using registered anonymizers. If no anonymizers are registered, the function simply encodes the dictionary without anonymization. Args: kwargs_dict: The dictionary containing the data to encode and anonymize. fields_to_anonymize: The set of fields within the dictionary to anonymize. object_type: A string indicating the type of object ('span' or 'trace') that was used to create the kwargs_dict. This is passed to anonymizers to provide context about the source object. Returns: A dictionary that has been encoded and, if applicable, anonymized. )objr anonymizersr )r encodeopikhookshas_anonymizersget_anonymizersanonymize_encoded_obj)r r r encoded_objrs /Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/opik/message_processing/encoder_helpers.pyencode_and_anonymizer sU.#))+6K :: % % '**,,.K /  rrct|tr-|D](}||vs|D]}|j||||||<*|S)a Anonymizes specified fields in an encoded dictionary using the provided anonymizers. This function iterates over the given set of field names and applies each anonymizer to the corresponding field in the dictionary, if present. The anonymizers are expected to implement an `anonymize` method that takes the field value, field name, and object type as arguments. Only fields present in the dictionary and listed in `fields_to_anonymize` are anonymized. Args: obj: The encoded dictionary whose fields are to be anonymized. fields_to_anonymize: A set of field names within the dictionary to anonymize. anonymizers: A list of anonymizer instances to apply to each field. object_type: A string indicating the type of object ('span' or 'trace'), providing context for anonymization. Returns: The dictionary with specified fields anonymized using the provided anonymizers. ) field_namer ) isinstancedict anonymize)rr rr ranonymizer_instances rrr-sY0#t-JS +6'&9&C&CJ#-$/'D'C O,7. Jr)typingrrrrr opik.hooksrr r strr Anonymizerrrrr*s00#!c3h!S!)! #s(^ !H" c3h"S"j++,") "  #s(^ "r