From 800f4b0a94d54c8c5832ada99f6ee92b4a40138b Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sat, 8 Jul 2023 16:45:59 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(loading.py):=20handle=20case?= =?UTF-8?q?=20when=20loading=20file=20into=20dict=20fails=20and=20raise=20?= =?UTF-8?q?ValueError=20with=20"Invalid=20file"=20message=20=E2=9C=A8=20fe?= =?UTF-8?q?at(loading.py):=20improve=20code=20readability=20by=20using=20w?= =?UTF-8?q?alrus=20operator=20to=20assign=20loaded=20file=20dict=20to=20pa?= =?UTF-8?q?rams["dict=5F"]=20variable?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/interface/initialize/loading.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/interface/initialize/loading.py b/src/backend/langflow/interface/initialize/loading.py index 9c900f18a..4e04fc8e6 100644 --- a/src/backend/langflow/interface/initialize/loading.py +++ b/src/backend/langflow/interface/initialize/loading.py @@ -254,12 +254,14 @@ def instantiate_prompt(node_type, class_object, params: Dict): def instantiate_tool(node_type, class_object: Type[BaseTool], params: Dict): if node_type == "JsonSpec": - params["dict_"] = load_file_into_dict(params.pop("path")) + if file_dict := load_file_into_dict(params.pop("path")): + params["dict_"] = file_dict + else: + raise ValueError("Invalid file") return class_object(**params) elif node_type == "PythonFunctionTool": params["func"] = get_function(params.get("code")) return class_object(**params) - # For backward compatibility elif node_type == "PythonFunction": function_string = params["code"] if isinstance(function_string, str):