This commit is contained in:
Harrison Chase 2023-01-03 12:39:57 -08:00
parent 6953c2e707
commit 6fe6af7048

View File

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