Module core.utilities.profiling

Functions

def clear_profiling_data()
def start_new_step()
def time_profiler(event_name, prompt_size=None, html_size=None, full_step_profiling=False)

A context manager to profile the execution time of code blocks.

Parameters: - event_name: The name of the event being profiled. - prompt_size: Optional size of the prompt, if applicable. - html_size: Optional size of the HTML, if applicable. - full_step_profiling: Boolean indicating whether to profile full steps or individual events.

Classes

class ChartGenerator (agent_events, agent_steps)

Methods

def get_summary_df(self)
def plot_waterfall(self)