🐛 fix(util.py): handle IndentationError when parsing class source code and log an error message
This commit is contained in:
parent
4d08d511e9
commit
2a53ac2688
1 changed files with 8 additions and 1 deletions
|
|
@ -3,6 +3,7 @@ import inspect
|
||||||
from typing import Dict, Union
|
from typing import Dict, Union
|
||||||
|
|
||||||
from langchain.agents.tools import Tool
|
from langchain.agents.tools import Tool
|
||||||
|
from langflow.utils.logger import logger
|
||||||
|
|
||||||
|
|
||||||
def get_func_tool_params(func, **kwargs) -> Union[Dict, None]:
|
def get_func_tool_params(func, **kwargs) -> Union[Dict, None]:
|
||||||
|
|
@ -57,7 +58,13 @@ def get_func_tool_params(func, **kwargs) -> Union[Dict, None]:
|
||||||
|
|
||||||
|
|
||||||
def get_class_tool_params(cls, **kwargs) -> Union[Dict, None]:
|
def get_class_tool_params(cls, **kwargs) -> Union[Dict, None]:
|
||||||
tree = ast.parse(inspect.getsource(cls))
|
try:
|
||||||
|
tree = ast.parse(inspect.getsource(cls))
|
||||||
|
except IndentationError:
|
||||||
|
logger.error(
|
||||||
|
f"Error parsing class {cls.__name__}. Make sure there are no tabs in the code."
|
||||||
|
)
|
||||||
|
return None
|
||||||
|
|
||||||
tool_params = {}
|
tool_params = {}
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue