Skip to content

Dev helper handlers

The handlers in here are just for development and testing purposes. They are not used in production.

E.g. Being easily able to print only certain parts of callbacks, or to record the call args of certain methods.

CaptureStartEndHandler

Bases: AsyncCallbackHandler

Useful in testing to capture the full call args of various methods.

on_chat_start_call_args = [] instance-attribute

on_chat_start_call_kwargs = [] instance-attribute

on_llm_end_call_args = [] instance-attribute

on_llm_end_call_kwargs = [] instance-attribute

__init__()

on_chat_model_start(*args, **kwargs) async

on_llm_end(*args, **kwargs) async

ExperimentalCallbackHandler

Bases: AsyncCallbackHandler

Using this when experimenting in jupyter to be able to see what is arriving at each method.

agent_action = agent_action instance-attribute

agent_finish = agent_finish instance-attribute

chain_end = chain_end instance-attribute

chain_error = chain_error instance-attribute

chain_start = chain_start instance-attribute

chat_model_start = chat_model_start instance-attribute

llm_end = llm_end instance-attribute

llm_error = llm_error instance-attribute

llm_new_token = llm_new_token instance-attribute

llm_start = llm_start instance-attribute

tool_end = tool_end instance-attribute

tool_error = tool_error instance-attribute

tool_start = tool_start instance-attribute

__init__(agent_action=False, agent_finish=False, tool_start=False, tool_end=False, tool_error=False, chain_start=False, chain_end=False, chain_error=False, chat_model_start=False, llm_start=False, llm_end=False, llm_error=False, llm_new_token=False)

on_agent_action(*args, **kwargs) async

on_agent_finish(*args, **kwargs) async

on_chain_end(*args, **kwargs) async

on_chain_error(*args, **kwargs) async

on_chain_start(*args, **kwargs) async

on_chat_model_start(*args, **kwargs) async

on_llm_end(*args, **kwargs) async

on_llm_error(*args, **kwargs) async

on_llm_new_token(*args, **kwargs) async

on_llm_start(*args, **kwargs) async

on_tool_end(*args, **kwargs) async

on_tool_error(*args, **kwargs) async

on_tool_start(*args, **kwargs) async

print_callback(method_name, args, kwargs) staticmethod