j\ UddlZddlZddlZddlmZmZmZmZddlm Z ddl m Z ejdZ hdZeeed<d eeefd eeeffd Zd e j$d efd Zdee j(ded e j(fdZy)N)AnyDictSetType)jsonable_encoder)messageszE^\d{4}-\d{2}-\d{2}T\d{2}:\d{2}:\d{2}(?:\.\d+)?(?:Z|[+-]\d{2}:\d{2})?$>end_time start_timelast_updated_atDATETIME_FIELD_NAMESobjreturnctD]m}|j|}t|ts%tj |s; t j j|jdd||<o|S#t$rY}wxYw)zObject hook for json.loads that converts ISO format strings to datetime objects. Only converts values for keys listed in DATETIME_FIELD_NAMES to avoid false conversions of ISO-like strings in arbitrary data fields. Zz+00:00) r get isinstancestr_ISO_DATETIME_PATTERNmatchdatetime fromisoformatreplace ValueError)rkeyvalues /Users/manta/Documents/Projects/TheRoad-I1/backend/.venv/lib/python3.12/site-packages/opik/message_processing/replay/message_serialization.pydatetime_object_hookrsx $  eS !&;&A&A%&H #,,::5==h;WXC $ J  s2A99 BBmessagecv|j}tj|}tj|S)aw Serializes a message object into a JSON string. This function converts a given message object, which must be a subclass of `messages.BaseMessage`, into a dictionary format suitable for database storage. The dictionary is first encoded via `jsonable_encoder.encode` (which converts datetime objects to ISO 8601 strings including timezone info via `datetime_utils.serialize_datetime`) and then serialized to a JSON string using the standard `json.dumps`. Args: message: The message object to be serialized. Returns: str: A JSON string representation of the message object. )as_db_message_dictrencodejsondumps)rdataencoded_data_dicts rserialize_messager'&s3"  % % 'D(//5 ::' (( message_classjson_strcftj|t}tj||S)a Deserializes a JSON string into an instance of a specified message class. This function takes a JSON string and transforms it into an instance of the provided message class. The conversion process includes mapping JSON structures into their equivalent class representations, with custom handling of date and time fields. Args: message_class: The class type into which the JSON string will be deserialized. This must be a subclass of `messages.T`. json_str: The JSON string to be deserialized. Returns: An instance of the provided message class populated with data from the JSON string. ) object_hook)r%r))r#loadsrr from_db_message_dict)r)r*r%s rdeserialize_messager/<s($ ::h,@ AD  ( (d- PPr()rr#retypingrrrropikrr compilerr r__annotations__r BaseMessager'Tr/r(rr9s ''!# L"Oc#hN d38n c3h  )x33)),QtHJJ'7Q3Q8::Qr(