github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-17 10:13:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

cr

Browse Source
This commit is contained in:
Harrison Chase 2023-01-03 12:39:57 -08:00
parent 6953c2e707
commit 6fe6af7048
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tests/unit_tests/callbacks/fake_callback_handler.py
View File

@ -2,7 +2,7 @@
from typing import Any, Dict, List from typing import Any, Dict, List
from langchain.callbacks.base import BaseCallbackHandler from langchain.callbacks.base import BaseCallbackHandler
from langchain.schema import AgentAction, LLMResult, AgentFinish from langchain.schema import AgentAction, AgentFinish, LLMResult
class FakeCallbackHandler(BaseCallbackHandler): class FakeCallbackHandler(BaseCallbackHandler):
@ -66,4 +66,3 @@ class FakeCallbackHandler(BaseCallbackHandler):
def on_agent_end(self, finish: AgentFinish, **kwargs: Any) -> None: def on_agent_end(self, finish: AgentFinish, **kwargs: Any) -> None:
"""Run when agent ends running.""" """Run when agent ends running."""
self.ends += 1 self.ends += 1