pi*dZddlmZddlZddlZddlZddlmZmZ ddl Z dZ e Z d dZ d dZGd d Zy#e $rdZ YwxYw) a Memory monitoring utilities for measuring memory usage. Example usage: tracker = MemoryTracker("my_function") with tracker.monitor(): my_function() # Access data: tracker.rss_delta, tracker.duration, etc. # Get formatted string: tracker.get_summary() ) annotationsN)DictOptionalTFcptr! tj}|jSy#t$rYywxYw)z Get current available system memory in bytes. Used for memory threshold checking in parallel test execution. Returns: int or None: Available memory in bytes, or None if unavailable N) _HAS_PSUTILpsutilvirtual_memory available Exception)sys_mems X/opt/services/ai/voice_agent/venv/lib/python3.12/site-packages/numba/misc/memoryutils.pyget_available_memoryrsB ++-G$$ $     s ) 55c`i}trj tjtj}|j }|j |d<tj}|j|d<d|vrd|d<d|vrd|d<|S#tjtjf$rY9wxYw)a^ Get memory usage information needed for monitoring. Returns only RSS and available memory which are the fields actually used by the MemoryTracker. Returns: dict: Memory usage information including: - rss: Current process RSS (physical memory currently used) - available: Available system memory rssr N) rrProcessosgetpid memory_inforr r NoSuchProcess AccessDenied)rprocessmem_infor s r get_memory_usager/sK nnRYY[1G**,H!)K ++-G'.'8'8K $  K! E+%#' K $$f&9&9:   sA)B#B-,B-ceZdZUdZded<ded<ded<ded<d ed <d ed <ded <d ed<ddZejdZddZ y) MemoryTrackerz A simple memory monitor that tracks RSS delta and timing. Stores monitoring data in instance attributes for later access. Each instance is typically used for monitoring a single operation. intpidstrnamez float | None start_timeend_timezDict[str, int | None] | None start_memory end_memorydurationz int | None rss_deltactj|_||_d|_d|_d|_d|_d|_d|_ y)z6Initialize a MemoryTracker with empty monitoring data.N) rrrrr r!r"r#r$r%)selfrs r __init__zMemoryTracker.__init__dsA99;    c#Ktj|_t|_ |tj|_t|_|j|jz |_|jjdd}|j jdd}|r |r ||z |_yd|_y#tj|_t|_|j|jz |_|jjdd}|j jdd}|r |r ||z |_wd|_wxYww)a Context manager to monitor memory usage during function execution. Records start/end memory usage and timing, calculates RSS delta, and stores all data in instance attributes. Args: name (str): Name/identifier for the function or operation being monitored Yields: self: The MemoryTracker instance for accessing stored data rrN) timer rr"r!r#r$getr%)r' start_rssend_rsss r monitorzMemoryTracker.monitoros ))+,. ?J!IIKDM.0DO MMDOO;DM))--eQ7Ioo))%3G!*w '2DN<= N!IIKDM.0DO MMDOO;DM))--eQ7Ioo))%3G!*w '2DN<= Ns)ECBEBEEc J|j |j td|jjd}dd}t j dt j |j}|jjdd}|jjdd}d|jd|jd |d |jd d d ||d||d||jdd||g}dj|S)a Return a formatted summary of the memory monitoring data. Formats the stored monitoring data into a human-readable string containing name, PID, RSS delta, available memory, duration, and start time. Returns: str: Formatted summary string with monitoring results Note: Should be called after monitor() context has completed to ensure all data is available. z$Memory monitoring data not availabler c|y|dk(ryd}|r |dkrdnd}t|}dD]}|dkr ||d d |cS|dz}||d d S) z&Convert bytes to human readable formatzN/Arz0 B-+)BKBMBGBg@z.2f z TB)abs) bytes_val show_signsignunits r format_bytesz/MemoryTracker.get_summary..format_bytess A~D'!msII/ $v%"VIc?!D6::V#  $VIc?#. .r)z%H:%M:%SrrzName: zPID: zStart: z Duration: z.3fsz Start RSS: z End RSS: z RSS delta: T)r<zAvail memory: z | )F) r"r# ValueErrorr,r+strftime localtimer rrr$r%join)r'current_availabler?start_tsr-r.bufs r get_summaryzMemoryTracker.get_summarys%    $(?CD D OO// < /$==T^^DOO-LM%%))%3 //%%eQ/TYYK DHH: hZ s+1 -,y12 3 W-. /,t~~FG H\*;<= >  zz#r)N)rr)returnr) __name__ __module__ __qualname____doc____annotations__r( contextlibcontextmanagerr/rHr)r rrTs\ H I..,, ??@4r)r)rIz Optional[int])rIzDict[str, Optional[int]])rM __future__rrrOr+typingrrrr ImportError IS_SUPPORTEDrrrrQr)r rVs] #  !K  &"JppKs9AA