From ac9969624e884d7d38eaa138101470df0ca6ffd4 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 6 Jul 2023 15:32:18 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(loading.py):=20remove=20empt?= =?UTF-8?q?y=20input=5Fkey=20and=20output=5Fkey=20from=20params=20dictiona?= =?UTF-8?q?ry=20The=20code=20now=20removes=20the=20"input=5Fkey"=20and=20"?= =?UTF-8?q?output=5Fkey"=20keys=20from=20the=20params=20dictionary=20if=20?= =?UTF-8?q?they=20are=20empty=20strings.=20This=20ensures=20that=20the=20d?= =?UTF-8?q?ictionary=20does=20not=20contain=20unnecessary=20keys=20with=20?= =?UTF-8?q?empty=20values.?= 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, 6 insertions(+) diff --git a/src/backend/langflow/interface/initialize/loading.py b/src/backend/langflow/interface/initialize/loading.py index cbac8278c..cd5585656 100644 --- a/src/backend/langflow/interface/initialize/loading.py +++ b/src/backend/langflow/interface/initialize/loading.py @@ -111,6 +111,12 @@ def instantiate_llm(node_type, class_object, params: Dict): def instantiate_memory(node_type, class_object, params): + # process input_key and output_key to remove them if + # they are empty strings + for key in ["input_key", "output_key"]: + if key in params and not params[key]: + params.pop(key) + try: if "retriever" in params and hasattr(params["retriever"], "as_retriever"): params["retriever"] = params["retriever"].as_retriever()