|
|
|
@ -39,14 +39,25 @@ class FakeCallbackHandler(BaseModel, BaseCallbackHandler):
|
|
|
|
|
"""Whether to ignore agent callbacks."""
|
|
|
|
|
return self.ignore_agent_
|
|
|
|
|
|
|
|
|
|
# add finer-grained counters for easier debugging of failing tests
|
|
|
|
|
chain_starts: int = 0
|
|
|
|
|
chain_ends: int = 0
|
|
|
|
|
llm_starts: int = 0
|
|
|
|
|
llm_ends: int = 0
|
|
|
|
|
tool_starts: int = 0
|
|
|
|
|
tool_ends: int = 0
|
|
|
|
|
agent_ends: int = 0
|
|
|
|
|
|
|
|
|
|
def on_llm_start(
|
|
|
|
|
self, serialized: Dict[str, Any], prompts: List[str], **kwargs: Any
|
|
|
|
|
) -> None:
|
|
|
|
|
"""Run when LLM starts running."""
|
|
|
|
|
self.llm_starts += 1
|
|
|
|
|
self.starts += 1
|
|
|
|
|
|
|
|
|
|
def on_llm_end(self, response: LLMResult, **kwargs: Any) -> None:
|
|
|
|
|
"""Run when LLM ends running."""
|
|
|
|
|
self.llm_ends += 1
|
|
|
|
|
self.ends += 1
|
|
|
|
|
|
|
|
|
|
def on_llm_error(
|
|
|
|
@ -59,10 +70,12 @@ class FakeCallbackHandler(BaseModel, BaseCallbackHandler):
|
|
|
|
|
self, serialized: Dict[str, Any], inputs: Dict[str, Any], **kwargs: Any
|
|
|
|
|
) -> None:
|
|
|
|
|
"""Run when chain starts running."""
|
|
|
|
|
self.chain_starts += 1
|
|
|
|
|
self.starts += 1
|
|
|
|
|
|
|
|
|
|
def on_chain_end(self, outputs: Dict[str, Any], **kwargs: Any) -> None:
|
|
|
|
|
"""Run when chain ends running."""
|
|
|
|
|
self.chain_ends += 1
|
|
|
|
|
self.ends += 1
|
|
|
|
|
|
|
|
|
|
def on_chain_error(
|
|
|
|
@ -75,10 +88,12 @@ class FakeCallbackHandler(BaseModel, BaseCallbackHandler):
|
|
|
|
|
self, serialized: Dict[str, Any], action: AgentAction, **kwargs: Any
|
|
|
|
|
) -> None:
|
|
|
|
|
"""Run when tool starts running."""
|
|
|
|
|
self.tool_starts += 1
|
|
|
|
|
self.starts += 1
|
|
|
|
|
|
|
|
|
|
def on_tool_end(self, output: str, **kwargs: Any) -> None:
|
|
|
|
|
"""Run when tool ends running."""
|
|
|
|
|
self.tool_ends += 1
|
|
|
|
|
self.ends += 1
|
|
|
|
|
|
|
|
|
|
def on_tool_error(
|
|
|
|
@ -93,4 +108,5 @@ class FakeCallbackHandler(BaseModel, BaseCallbackHandler):
|
|
|
|
|
|
|
|
|
|
def on_agent_finish(self, finish: AgentFinish, **kwargs: Any) -> None:
|
|
|
|
|
"""Run when agent ends running."""
|
|
|
|
|
self.agent_ends += 1
|
|
|
|
|
self.ends += 1
|
|
|
|
|