6i ,`ddlmZddlZddlZddlmZerddlZndZddZ ddZy)) annotationsN)is_jmespath_availablecd|jr5|jjDcic] \}}| || c}}St|j}t |dkDrt d|dt |dk(rt d|j d|dScc}}w)Nz+Regex has multiple unnamed groups! Groups:  rzRegex has no capture groups: ) groupdictitemslistgroupslen ValueErrorgroup) node_matchkeyvalr s a/mnt/e/genesis-system/.venv/lib/python3.12/site-packages/transformers/utils/chat_parsing_utils.py_parse_re_matchrs)3)=)=)?)E)E)G[XS#3?S[[j'')* v;?KF8SUVW W [A ? @P@PQR@S?TUV Vay\s B,B,c  d|vr|dS|y|jd}d|vxs d|vxsd|v}|r!t|tstd|d||jd}|jd}|jd}|4t j ||tj }|syt|}|U|d k7rtd |d |t j||tj Dcgc] }t|}}|sy||d k7rtd |d|i}t j||tj D]A}t|} t| trd| vsd| vrtd| d|| d|| d<C|}|syd|vr|d} | dk(rt|tstd|d||jdi} | jd} | jdd} tj|}| *t tdtj | d|}|}ntd| d|d|vr2t|tstd |d|d||d}||vr||}d |vrht|tstd |d!|d||d }|j!D],\}}t j"|||tj }.|d k(ri}t|trLd"|vrtd#|d||d"j!D]\}}t%||d"|}||||<!|St|tr|jd"ij!D]6\}}d|vr |d||<||vrt%|||||<*d$|vs/|d$||<8d%|vr?|j!D],\}}||jd"ivst%||d%||<.|Std&||d k(r|sgSg}d'|vrEt|t&std(||D] }|j)t%||d'"|Sd)|vrt|t&s#t+|d)d*k(r|g}ntd(|t+|t+|d)k7r,td+t+|d,t+|d)d-|d|t-||d)D] \}}|j)t%||"|Std.||d/vr|t|tstd0|d1||d2k(r t/|S|d3k(r t1|S|d4k(r4|j3d5vry6|j3d7vrytd8||S||Std9|d:|cc}w#tj$r!}| r|}ntd|d|Yd}~d}~wwxYw);a This function takes content and a JSON schema which includes regex extractors, and recursively parses the content. The output should be a data structure matching the schema. Args: node_content: The content corresponding to this node. Usually a string, but can be something else if the parent node has multiple capture groups or named groups. In that case, we generally pass the capture groups straight through to the children of this node and don't do any parsing at this level. node_schema: The schema node controlling the parsing. Returns: The parsed data structure for the current node. constNtypezx-regexzx-regex-iteratorzx-regex-key-valuezHSchema node got a non-string input, but has a regex for parsing. Input: z Schema: )flagsarrayzSchema node with type z& cannot use x-regex-iterator. Schema: objectz' cannot use x-regex-key-value. Schema: rvaluezTRegex for x-regex-key-value must have named groups 'key' and 'value'. Match groups: zx-parserjsonz/Node has JSON parser but got non-string input: z x-parser-args transformallow_non_jsonFzNode has JSON parser but could not parse its contents as JSON. You can use the `allow_non_json` parser arg for nodes which may contain JSON or string content. Content: z Error: zChat response schema includes a jmespath transformation, but jmespath is not installed. You can install it with `pip install jmespath`.zUnknown parser z for schema node: z x-mappingz6 cannot use x-mapping on non-string content. Content: zx-mapping-regexz< cannot use x-mapping-regex on non-string content. Content: propertieszVObject node received string content but has no regex or parser to handle it. Content: defaultadditionalPropertiesz=Expected a dict or str for schema node with type object, got r z>Expected a list or regex for schema node with type array, got prefixItemsrzArray node has z items, but schema only has z prefixItems defined. Content: z?Array node has no items or prefixItems schema defined. Schema: )stringintegernumberbooleanz,Expected a string for schema node with type z, got r#r$r%)true1T)false0zInvalid boolean value: zUnsupported schema type z for node: )get isinstancestr TypeErrorresearchDOTALLrfinditerdictr rloadsJSONDecodeErrorjmespath ImportErrorr subrecursive_parser appendr zipintfloatlower) node_content node_schema node_type has_regex node_regexnode_regex_iteratornode_regex_to_dictroutput_content match_groupsparser parser_argsrr parsed_jsonemapping mapping_regexpattern replacement parsed_schemar child_nodechild_node_contentritem item_schemas rr8r8s*+7##'I[(s,>+,MsQdhsQsIL#6 "^$"m %  +J%//*<=$)<=YYzK# 4I(__-=uEN "jj6 $#%b'ook+.F T &Lvh6H VW Wk!,,( 4(>*&-)  k* 7 ""<0LK',,( 4(>*&-)  $$56 $1$7$7$9 W G[66'; BIIVL WH lC (;.  ,~.*m- $/|#<#B#B#D <Z%4\;|C\]`Ca%b"%1);M#& <!  d +#.??<#D#J#J#L ?Zj()3G)M#&  ?&4"."4"4"6iJC+//,"CC-L#&deqdr$stt< C M(B$CC %c,&7%88T;}5678 ,~.*m- &){=7Q%R I!k$$_T;%GH I _`k_lmn n @ @,,J9+U[\h[ijk k  !|$ $ ( "& & ) #!!#}4##%7 #:<.!IJJ   29+[WXX L'' !".K$FGSFTT_`a_bc  s:V??WW8W33W8)rzre.Matchreturnz dict | str)r>zstr | list | dictr?r2) __future__rrr.transformers.utilsrr5rr8rrYsA" 4H QY#QYQYrX