pengjunjie/langflow
1
0
Fork 1
mirror of https://github.com/langflow-ai/langflow.git synced 2026-07-24 07:34:10 +08:00
Code Issues Packages Projects Releases Wiki Activity

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
parent 3b52cd0e6b
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
exec(function_string, namespace)
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,
)
if function_object is None: