From 8a0630652f45570003cf4be2da8c8245d1394ad6 Mon Sep 17 00:00:00 2001 From: Tiago Santos Date: Wed, 20 Sep 2023 18:41:45 +0100 Subject: [PATCH] fix(util.py): fix IndentationError while parsing from source code. --- src/backend/langflow/interface/tools/util.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/interface/tools/util.py b/src/backend/langflow/interface/tools/util.py index 48d9368d1..8e4f582c1 100644 --- a/src/backend/langflow/interface/tools/util.py +++ b/src/backend/langflow/interface/tools/util.py @@ -1,12 +1,13 @@ import ast import inspect +import textwrap from typing import Dict, Union from langchain.agents.tools import Tool def get_func_tool_params(func, **kwargs) -> Union[Dict, None]: - tree = ast.parse(inspect.getsource(func)) + tree = ast.parse(textwrap.dedent(inspect.getsource(func))) # Iterate over the statements in the abstract syntax tree for node in ast.walk(tree): @@ -57,7 +58,7 @@ def get_func_tool_params(func, **kwargs) -> Union[Dict, None]: def get_class_tool_params(cls, **kwargs) -> Union[Dict, None]: - tree = ast.parse(inspect.getsource(cls)) + tree = ast.parse(textwrap.dedent(inspect.getsource(cls))) tool_params = {}