jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

refactor(validate.py): filter function objects by filename to ensure they are from the evaluated string

Browse source
This commit is contained in:
Gabriel Almeida 2023-04-18 21:27:47 -03:00
commit a0da9c6111
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/backend/langflow/utils/validate.py
View file

@ -59,7 +59,12 @@ def eval_function(function_string: str):
# Execute the code string in the new namespace # Execute the code string in the new namespace
exec(function_string, namespace) exec(function_string, namespace)
function_object = next( function_object = next(
(obj for name, obj in namespace.items() if isinstance(obj, types.FunctionType)), (
obj
for name, obj in namespace.items()
if isinstance(obj, types.FunctionType)
and obj.__code__.co_filename == "<string>"
),
None, None,
) )
if function_object is None: if function_object is None: