:vj% dZddlmZddlZddlZddlmZddlmZm Z ddl m Z ddl mZdd lmZmZd Zdd Zd Zd ZdddddddddZdZdZdZdZdZdZdZy)z-Code for converting notebooks to and from v3.) annotationsN) get_logger)v3 validator)generate_corpus_id) NotebookNode)nbformatnbformat_minorcddlm}m} |||y#|$r%}tj d||Yd}~yd}~wwxYw)z(Log validation errors, if there are any.r)ValidationErrorvalidate)versionz"Notebook JSON is not valid v%i: %sN)r r rrerror)nbrr res d/workspace/moltex-experiments/.pixi/envs/default/lib/python3.12/site-packages/nbformat/v4/convert.py_warn_if_invalidrs?2MW% M ?!LLMs ?:?c(|s|d}|s(d|vr|dk(rd}tj|d}n|d}|dk(rt|||jdd}|jd d}|xsd|j_|xsd|j_t|_t|_gx|d <}|jd gD]&}|d D]}|jt|(|jjd d |jjdd t|t|S|dk(rN|tk(r|S|dkr |jD]}t|_ ||j_t|_|Std|z)aWConvert a notebook to latest v4. Parameters ---------- nb : NotebookNode The Python representation of the notebook to convert. from_version : int The original version of the notebook to convert. from_minor : int The original minor version of the notebook to convert (only relevant for v >= 3). r r zEThe v4 notebook does not include the nbformat minor, which is needed.r orig_nbformatNorig_nbformat_minorcells worksheetsname signaturezZCannot convert a notebook directly from v%s to v4. Try using the nbformat.convert module.)rr rpopmetadatarrr r append upgrade_cellrrandom_cell_idid ValueError) r from_version from_minormsgrrrwscells rupgrader,s *~  2 %q ]//44J,-Jq\*5  ff%:DA$1$6Q !*=*B ' *! 7 e&&r*B7  \$/0$+ #  R(X& q  'I >(*!+5 '*  13? @ c |jdtt|_|jdk(r~|j ddd|vr|j d|j d<|j dd|_|j dd|_t|j|_ |S|jd k(rcd |_|j d d }d jd|zdj|jddj|_|S|jdk(rd |_|S)zupgrade a cell from v3 to v4 heading cell: - -> markdown heading code cell: - remove language metadata - cell.input -> cell.source - cell.prompt_number -> cell.execution_count - update outputs r!codelanguager collapsedinput prompt_numberNheadingmarkdownlevelr z{hashes} {single_line}# source)hashes single_linehtml) setdefaultrr$r% cell_typer r!r9execution_countupgrade_outputsoutputsformatjoinget splitlines)r+r6s rr#r#fs OOJ /DG ~~ R $ )-+)>DMM+ &hhw+ #xx>&t||4  K 9 $#!$.55;(B!7!B!B!DE6   K 6 !# Kr-cP|jdk(rqd|_|jdd|_|jdd|_|j jdd|_t|j|_n|jd k(rr|jdd}d |vr\|jd rKtjd |}|J|j\}}d |_||_t||_|jdd|jdd|S)adowngrade a cell from v4 to v3 code cell: - set cell.language - cell.input <- cell.source - cell.prompt_number <- cell.execution_count - update outputs markdown cell: - single-line heading -> heading cell r/pythonr9rr?Nr1Fr5 r7z (#+)\s*(.*)r4r% attachments)r>r0r r2r3r!r1downgrade_outputsrArD startswithrematchgroupsr9lenr6)r+r9rMprefixtexts rdowngrade_cellrRs ~~  XXh+ !XX&7>**;>(6 : %(B' v &"3"3C"8HH^V4E$ $$ <<>LFD&DNDKVDJHHT4HH]D! Kr-z text/plainz text/htmlz image/svg+xml image/png image/jpegz text/latexapplication/jsonzapplication/javascript)rQr<svgpngjpeglatexjson javascriptcptjD]\}}||vs |j|||< |S)z4convert dict with v3 aliases to plain mime-type keys) _mime_mapitemsr )daliasmimes r to_mime_keyrbs6 ( t A:eeElAdG) Hr-c\i}tjD]\}}||vs ||||<|S)z.convert dict with mime-type keys to v3 aliases)r]r^)r_d2r`ras r from_mime_keyres7 B ( t 19$BuI) Ir-c4|ddvr|jdt|ddk(rd|d<|jdd|d<i}t|D]}|d vr|j|||<t |||d <t |j d |vrt j|d |d <d D]2}||vst||ts||jd ||<4|S|ddk(rd|d<|S|ddk(r|jdd|d<|S)zupgrade a single code cell output from v3 to v4 - pyout -> execute_result - pyerr -> error - output.type -> output.data.mime/type - mime-type keys - stream.stream -> stream.name output_type>pyout display_datar!rhexecute_resultr3Nr?>r!rgr?datarU)rSrTasciipyerrrstreamstdoutr) r=rr listrbr!rZloads isinstancebytesdecode)outputrkkeys rupgrade_outputrwsIm 99*ln5 - G +$4F= !(. ?D(IF$ % output.type - un-mime-type keys - stream.stream <- stream.name rg>rirjrjrhr?Nr3rkrUr!rrmrnr)r rZdumpsreupdaterD)rurks rdowngrade_outputr{sm BB - $4 4$+F= !&,jj1BD&IF? #zz&"%  %'+zz$7I2J'KD# $T" dfjjR01 M  ' ) '} M  ( *!::f-x Mr-c>|Dcgc] }t|c}Scc}w)z,upgrade outputs of a code cell from v3 to v4)rwrAops rr@r@s)0 12N2  11 1c>|Dcgc] }t|c}Scc}w)z.downgrade outputs of a code cell to v3 from v4)r{r}s rrJrJ s +2 37R R 7 33 3rc4|jtk7r|St|ttj|_tj|_|j dDcgc] }t |}}tj |g|_|jjddt|tj|jj dt|_ |jj dt|_ |Scc}w)zConvert a v4 notebook to v3. Parameters ---------- nb : NotebookNode The Python representation of the notebook to convert. r)rrrrr) r rrr r rR new_worksheetrr!r=rr)rr+rs r downgraders {{h R"++BK))B.0ffWo >od^D !oE >%%E23BMKK62&R%{{AB[[__-BNSB I ?s#D)NN)__doc__ __future__rrZrL traitlets.logrr rrnbformat.corpus.wordsrr$nbformat.notebooknodernbbaser rr,r#rRr]rbrerwr{r@rJrr-rrs3# $"F.,MFRB@       *   "J82 4 r-