nat.plugins.crewai.crewai_callback_handler#

Attributes#

Classes#

CrewAIProfilerHandler

A callback manager/handler for CrewAI that intercepts calls to:

Module Contents#

logger#

class CrewAIProfilerHandler#

Bases: nat.profiler.callbacks.base_callback_class.BaseProfilerCallback

A callback manager/handler for CrewAI that intercepts calls to:

ToolUsage._use
LLM Calls

to collect usage statistics (tokens, inputs, outputs, time intervals, etc.) and store them in NAT’s usage_stats queue for subsequent analysis.

_lock#

last_call_ts#

step_manager#

_original_tool_use = None#

_original_llm_call = None#

instrument() → None#: Monkey-patch the relevant CrewAI methods with usage-stat collection logic. Assumes the ‘crewai’ library is installed.

_tool_use_monkey_patch() → collections.abc.Callable[Ellipsis, Any]#: Returns a function that wraps calls to ToolUsage._use(…) with usage-logging.

_llm_call_monkey_patch() → collections.abc.Callable[Ellipsis, Any]#: Returns a function that wraps calls to litellm.completion(…) with usage-logging.