oFjUddlmZddlmZmZddlmZmZddlmZddl m Z ddl m Z ddl Z ddlZddlmZejZdad ed <d Zd4dZd5dZd6dZd7dZd8dZd9d!Zd:d%Zd;dd/Zd?d1Z d@d2Z!dAd3Z"dS)B) annotations)MappingSequence)fields is_dataclass)datetime)Enum)PathN)Any Path | None _LOG_PATHzdata:folderstr | os.PathLike[str] | Nonereturnct5|sda ddddSt|}|ddt jd}|d|dtj dz at dd td ttcdddS#1swxYwYdS) NT)parentsexist_okz%Y%m%d_%H%M%S_%fllm_io__z.logzWanGP LLM I/O transcript Text is preserved verbatim. Non-text content is written using its known type/name, token IDs, or numeric values. Binary media is represented by source, media type, dimensions, and size; encoded binary bodies are not dumped. utf-8)encodingz![LLM I/O] Logging LLM traffic to ) _LOCKr r expanduserresolvemkdirrnowstrftimeosgetpid write_textprint)rtargetstamps &C:\Users\johnn\Wan2GP\shared\llm_io.pyconfigure_llm_ior%sa  If((**2244 TD 111 ''(:;;@u@@ry{{@@@@  A    =)==>>>sC.CC..C25C2cRt5tcdddS#1swxYwYdSNrr r$get_llm_io_pathr+&so s   boolcVt5tducdddS#1swxYwYdSr'r(r)r*r$llm_io_enabledr.+st %%$%%%%%%%%%%%%%%%%%%s "" directionstrenginestreampayloadr metadataNonec t5t ddddSt|pd}|dkrdn |dkrdn|}dt jdd |d |d |}d |d g}|r7| t|| d | t|t ddd5} | d|dzdddn #1swxYwYddddS#1swxYwYdS)NOUTu OUT → LLMINu IN ← LLM[ milliseconds)timespecz] [z ] engine=z stream=zd====================================================================================================zd----------------------------------------------------------------------------------------------------ar )rnewlinez )rr r0stripupperrr astimezone isoformatextend_renderappendopenwritejoin) r/r1r2r3r4 direction_keydirection_labelheaderlineswriters r$ log_llm_iorO0s=  4 4    4 4 4 4 4 4 4 4IO,,2244::<< +8E+A+A--WdhlWlWl||rKX\^^..00::N:SSKKXgKKrxKKCIKKFI.  $ LL** + + + LL # # # WW%%&&& ^^C'4^ @ @ 4F LL5))F2 3 3 3 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4 4s; E7D E7',E E7E# #E7&E# 'E77E;>E;mediadict[str, Any]c^t|drt|drt|ttjfs|j\}}tt |ddpdtt |ddpt|jt|t|t|j dSt| }t|| jd} d d lm}||5}||p|jpd |j|j|j d dddn #1swxYwYn#t.t0f$rd |d<YnwxYw|S)Nsizemodefilenamer7zformat)source media_typewidthheightrT)rWbytesr)Imageimage)rXrYrZrTfilerX)hasattr isinstancer0rPathLikerSgetattrtype__name__intrTr rstatst_sizePILr\rGupdateget_format_mimetyperVrYrZ ImportErrorOSError)rPrYrZrW descriptorr\r]s r$media_descriptorrn@suf '%"8"8 ETWY[YdSeAfAf  v'%R88ORSSZZ&kk OO    %[[ " "F,/KK&++--BW!X!XJ* ZZ   h5   U-F-F-H-H-cEL-c\cnsnyFKFR\a\fgg h h h h h h h h h h h h h h h h h h  !***#) <   * s7F4AF9 FF  F F  FF*)F* tokenizerdict[str, dict[str, Any]]ci}dD]}t||dd}|t||dd}|2t|dr"|t|}t|t |p|d||<|S)N)boseospadunksepclsmask _token_id_tokenconvert_ids_to_tokensidname)rbr_r{rer0)roresultr~token_id token_names r$known_token_idsrVs FBNN9&8&8&8$??   Y4>>  ')5L"M"M "88XGGJ!(mmS9Kt5L5LMMt Mr*r int | Nonedict[str, Any] | Nonec|dSt|dr"|t|nd}t|t|pddS)Nr{r7z r|)r_r{rer0)rorrs r$token_id_descriptorrcs]tCJ9VmCnCnv00X???tvJh--Z-F;)G)G H HHr*valueindentreseenset[int] | None list[str]cdz}|tn|}t|}t|trt ||S|"t|t t tfr|t|zgSt|}||vr|dt|j dzgSt|tr>| |g}| D]\}}t||}|t|zdz}t|rt!|d|} t#| dkr"||dz| dz||dz|fd| D|||t!|d z||||p|d zgSt|t*rt|tt,t.fs| |g}|D]m}t!|d z|} ||d z| rd| dznd z|| ddn|||p|d zgS|t|zgS)N :rz |c3.K|]}ddzz|zVdS)rNr)).0liners r$ z_render..s1 P Pt !3d!: P P P P P Pr*rz-r7z)set _plain_valuer`r0 _render_textr,refloatr}rcrdradditems_binary_descriptor _is_scalarrElenrFrDremoverr[ bytearraylstrip) rrrprefixidentityrMkeyitemlabelrendered item_liness ` r$rErEjs 6\FL3555dD   E%+E6*** } 54e*<==}U#$$%yyH4IT%[[1IIIIJJ%!!5  > >IC%c400DSXX%+E$ >"4D11x==A%%LLx{!:;;;;LL...LL P P P Px P P PPPPP U### WT6A:t<<==== H4"3344%""6:ec5)=T+U+U6  ) )D vz488J LL#:)]# 1 0D0D0F0F*F*F[]^ _ _ _ LLABB ( ( ( ( H5"4455 SZZ   r*rc^d|vr|zgSfd|DpgS)Nr>cg|]}|zSr)r))rrrs r$ z _render_text..s 9 9 9dFTM 9 9 9r*) splitlines)rrs `r$rrsH 5 9 9 9 9e&6&6&8&8 9 9 9 EfXEr*cttr'tjdjdjdStt rtSttttfr)dtjdtddStrFtts1dtjifd tDStd rutd retjtjtjd }td rtj|d <|StdrAdtjidt'DSS)N.z ()rz: ,z bytes> content_typecFi|]}|jt|jSr))r~rb)rfieldrs r$ z _plain_value..s,8{8{8{diWUTYT^E_E_8{8{8{r*shapedtype)rrrdevice__dict__cDi|]\}}|d||S)r) startswith)rrrs r$rz _plain_value..sT9D9D9DysDoro}o}BpCpC9Dd9D9D9Dr*)r`r rcrdr~rr r0r[r memoryviewrrrr_tuplerrrvarsr)rrms` r$rrs%Fu++&EEEEu{EEEE%5zz%%J788A@4;;'@@3u::@@@@@E}:eT#:#:}U 4|8{8{8{8{mstymzmz8{8{8{||ug75'#:#:&*5kk&:U5;EWEWbefkfqbrbrss 5( # # 5#&u|#4#4Jx uj!!EU 4E9D9DTXY^T_T_TeTeTgTg9D9D9DE E Lr*rct|}t|trJ|tr0|d\}}}d|dt |ddS|dkrKt|tr6t |dkr#t|rdt |ddS|S) Nrrz; encoded characters=>dataz)r0casefoldr`r_DATA_URI_PREFIX partitionr_looks_like_encoded_binary)rrkey_name media_headerrencodeds r$rrs3xx  ""H%H%"2"23C"D"DH#(??3#7#7 aG<GGc'llGGGGG6j44Uc9I9INhinNoNo9I>CJJ>>>>> Lr*c|dddd}t|dzdkotd|DS)N r7r>rc3FK|]}|p|dvVdS)z+/=N)isalnum)r characters r$rz-_looks_like_encoded_binary..s9(l(lW`):):)<)<)R U@R(l(l(l(l(l(lr*)replacerall)rcompacts r$rrsXmmD"%%--dB77G w<rs"""""""--------,,,,,,,,    ( %%%% 4 4 4 4 ,    IIII&!&!&!&!&!RFFFF &mmmm GGGGGGr*