OtjJ ddlmZddlmZddlZddlmZddeedefdZddeedefd Z ddeed eefd Z ddeeded e fd Z ddeedede d dfdZ dS))Optional)protonN)flagssessionphasec|dStjr|dkrtdtjt j||S)z Retrieves profiling data for a given session. Args: session (Optional[int]): The session ID of the profiling session, or None if profiling is inactive. Returns: str: The profiling data in JSON format. NrEOnly one session can be retrieved when running from the command line.)r command_line ValueErrorjsonloads libprotonget_datarrs R/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/triton/profiler/data.pygetrsNt bgll`aaa :i(%88 9 99cv|dStjr|dkrtdtj||S)a Retrieves profiling data for a given session encoded with MessagePack. Args: session (Optional[int]): The session ID of the profiling session, or None if profiling is inactive. Returns: bytes: The profiling data encoded with MessagePack. Nrr )rr r rget_data_msgpackrs r get_msgpackrsDt bgll`aaa  %gu 5 55rreturnct|dStjr|dkrtdtj|S)z Advances the profiling phase for a given session. Args: session (Optional[int]): The session ID of the profiling session, or None if profiling is inactive. Returns: Optional[int]: The next phase number after advancing. NrzFOnly one session can advance phase when running from the command line.)rr r radvance_data_phase)rs r advance_phaser(sBt cgllabbb  ' 0 00rcv|dStjr|dkrtdtj||S)a  Checks if the profiling data for a given session and phase is complete. A "complete" phase is safe to read/clear because all device-side records for the phase have been flushed to the host and the phase will no longer receive new records. Args: session (Optional[int]): The session ID of the profiling session, or None if profiling is inactive. phase (int): The phase number to check. Defaults to 0. Returns: bool: True if the phase data is complete, False otherwise. NFrzVOnly one session can check phase completion status when running from the command line.)rr r ris_data_phase_completers ris_phase_completer9sDu sgllqrrr  +GU ; ;;rFclear_up_to_phasec||dStjr|dkrtdtj|||dS)aC Clears profiling data for a given session. Args: session (Optional[int]): The session ID of the profiling session, or None if profiling is inactive. phase (int): The phase number to clear. Defaults to 0. clear_up_to_phase (bool): If True, clear all phases up to and including `phase`. NrzCOnly one session can be cleared when running from the command line.)rr r r clear_data)rrrs rclearr"OsM `gll^___ %):;;;;;r)rr)r)rrF)typingrtriton._C.libprotonrrr rintrrrboolrr"rrr(sT333333 : :# :3 : : : : 66#636666"118C=1#1111"<