From 1a0c5c497a6f57749eaecf2458b63b73dde134f5 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 28 Nov 2023 15:38:40 -0300 Subject: [PATCH] Add basic imports from typing module --- src/backend/langflow/utils/validate.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/backend/langflow/utils/validate.py b/src/backend/langflow/utils/validate.py index 0a33dfa01..2103c4b0f 100644 --- a/src/backend/langflow/utils/validate.py +++ b/src/backend/langflow/utils/validate.py @@ -235,6 +235,12 @@ def build_class_constructor(compiled_class, exec_globals, class_name): :param class_name: Name of the class :return: Constructor function for the class """ + # Add basic imports from typing module + # List, Dict, Tuple, Union, Optional + # to the global scope + for name in ["List", "Dict", "Tuple", "Union", "Optional"]: + exec_globals[name] = getattr(importlib.import_module("typing"), name) + exec(compiled_class, exec_globals, locals()) exec_globals[class_name] = locals()[class_name]