From cb8ecaccc7159aa354579ed1b6ec8310d668eb11 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Tue, 3 Oct 2023 13:33:54 +0100 Subject: [PATCH] Dumbness --- .../langchain/langchain/tools/pandas_eval/pandas_eval.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/libs/langchain/langchain/tools/pandas_eval/pandas_eval.py b/libs/langchain/langchain/tools/pandas_eval/pandas_eval.py index 2afa4a94f6e..be7bd0fbf68 100644 --- a/libs/langchain/langchain/tools/pandas_eval/pandas_eval.py +++ b/libs/langchain/langchain/tools/pandas_eval/pandas_eval.py @@ -1,5 +1,5 @@ from functools import partial -from typing import Any, Dict +from typing import Any, Dict, Optional from typing_extensions import TypeAlias @@ -62,7 +62,10 @@ class PandasEvalTool(BaseTool): def _run( self, query: str, - run_manager: CallbackManagerForToolRun, + run_manager: Optional[CallbackManagerForToolRun] = None, ) -> Any: chain = get_pandas_eval_chain(self.model, self.dfs) - return chain.invoke({"input": query}, {"callbacks": run_manager.get_child()}) + return chain.invoke( + {"input": query}, + {"callbacks": run_manager.get_child()} if run_manager else {}, + )