`in%~dZddlmZddlZddlZddlZddlZddlmZddl m Z d"d Z Gd d Z e Z d#dZd$dZee _ee _ d%d&dZedeedeedeedeedeedeed eed!edS)'zError handling.) annotationsN)Any)core backtracestrreturnlist[tuple[str, int, str]]c|d}g}|dD]}tj||}|rv |d}t |d}|d}||||f#t$rYwxYw|S)zParse the backtrace string into a list of (filename, lineno, func). Parameters ---------- backtrace The backtrace string. Returns ------- result The list of (filename, lineno, func) z!File "(.+?)", line (\d+), in (.+) r)splitrematchstripgroupintappend ValueError)rpatternresultlinerfilenamelinenofuncs a/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/tvm_ffi/error.py_parse_backtracers3G F%%  $**,,//    ;;q>>U[[^^,,{{1~~ x67777       MsA$B++ B87B8c2eZdZdZddZdd Zdd ZddZdS)TracebackManagerz&Helper to manage traceback generation.r Noneci|_dS)z/Initialize the traceback manager and its cache.N) _code_cache)selfs r__init__zTracebackManager.__init__?sGIrrrrrtypes.CodeTypecX|||f}||jvr |j|Stjd|d}tj|D]0}t |drd|_t |drd|_1t||d}|||}||j|<|S)Nz _getframe()eval)rmode col_offsetrend_col_offset)co_nameco_firstlineno) r#astparsewalkhasattrr+r,compilereplace)r$rrrkeytreenode code_objects r_get_cached_code_objectz(TracebackManager._get_cached_code_objectCs& $" " "#C( (yGGGHTNN ( (Dt\** $"#t-.. (&'#dHf55 !))$v)NN +r&types.FrameTypecn||||}dtji}t|||S)z:Create a frame object from the filename, lineno, and func. _getframe)r9sysr<r))r$rrrr8contexts r _create_framezTracebackManager._create_frameYs9228VTJJ  .K'222r&tbtypes.TracebackType | Nonetypes.TracebackTypecNd d }|||||||S) aAppend a traceback to the given traceback. Parameters ---------- tb The traceback to append to. filename The filename of the traceback lineno The line number of the traceback func The function name of the traceback Returns ------- new_tb The new traceback with the appended frame. r@rAframer:rrr rBc:tj|||j|S)N)types TracebackTypef_lasti)r@rDrs rcreatez1TracebackManager.append_traceback..creates&r5%-HH Hr&)r@rArDr:rrr rB)r?)r$r@rrrrIs rappend_tracebackz!TracebackManager.append_tracebackbsC@ I I I I vb$,,XvtDDfMMMr&N)r r!)rrrrrrr r')rrrrrrr r:) r@rArrrrrrr rB)__name__ __module__ __qualname____doc__r%r9r?rJr&rr r <st00JJJJ,3333%N%N%N%N%N%Nr&r py_error BaseExceptionc|j} t|D]#\}}}t||||}$||~~S#~~wxYw)z3Append the backtrace to the py_error and return it.) __traceback__r_TRACEBACK_MANAGERrJwith_traceback)rPrr@rrrs r_with_append_backtracerVsw>  B &6y&A&A Q Q "Hfd#44R64PPBB&&r** bbHbs AAAr@rAc g}|N|j}|j}|jj}|jj}|d|d|d|d|j}|Ndt|S)z"Convert the traceback to a string.Nz File "z", line z, in r ) tb_frame tb_linenof_code co_filenamer-rtb_nextjoinreversed)r@linesrDrrfuncnames r_traceback_to_backtrace_strrbs E . <+<' KKK&KKxKKKLLL Z . 778E?? # ##r& name_or_clsstr | type | Nonecls type | Nonerc^tr }|jdfd }||S||S)aRegister an error class so it can be recognized by the ffi error handler. Parameters ---------- name_or_cls The name of the error class. cls The class to register. Returns ------- fregister Register function if f is not specified. Examples -------- .. code-block:: python import tvm_ffi # Register a custom Python exception so tvm_ffi.Error maps to it @tvm_ffi.error.register_error class MyError(RuntimeError): pass # Convert a Python exception to an FFI Error and back ffi_err = tvm_ffi.convert(MyError("boom")) py_err = ffi_err.py_error() assert isinstance(py_err, MyError) myclstyper cttrn|j}|tj|<|tj|<|S)z0Register the error class name with the FFI core.) isinstancerrKrERROR_NAME_TO_TYPEERROR_TYPE_TO_NAME)rherr_namercs rregisterz register_error..registers<",[#">">R;;EN,1))1& r&N)rhrir ri)callablerK)rcreros` rregister_errorrqsZL #l  { 8C==r& RuntimeErrorr TypeErrorAttributeErrorKeyError IndexErrorAssertionError MemoryError)rrr r )rPrQrrr rQ)r@rAr r)NN)rcrdrerfr r)rN __future__rr/rr=rFtypingrrXrrr rTrVrb_WITH_APPEND_BACKTRACE_TRACEBACK_TO_BACKTRACE_STRrqrrrrsrtrurvrwrxrOr&rr}s$"""""" :KNKNKNKNKNKNKNKN\&%''****Z $ $ $ $5#> &*33333l~|,,,|Z((({I&&&000z8$$$|Z(((000}k*****r&