From 6abd5d5343cdb07be072dcd298c341ee769ef132 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 12 Oct 2023 16:29:30 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(base.py):=20handle=20cases?= =?UTF-8?q?=20where=20value=20is=20not=20a=20valid=20int=20or=20float=20to?= =?UTF-8?q?=20prevent=20ValueError=20and=20assign=20the=20original=20value?= =?UTF-8?q?=20instead?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/graph/vertex/base.py | 14 ++++++++++---- 1 file changed, 10 insertions(+), 4 deletions(-) diff --git a/src/backend/langflow/graph/vertex/base.py b/src/backend/langflow/graph/vertex/base.py index ad6fc952ff..de64b3598b 100644 --- a/src/backend/langflow/graph/vertex/base.py +++ b/src/backend/langflow/graph/vertex/base.py @@ -216,10 +216,16 @@ class Vertex: } elif isinstance(_value, dict): params[key] = _value - elif value.get("type") == "int": - params[key] = int(value.get("value")) - elif value.get("type") == "float": - params[key] = float(value.get("value")) + elif value.get("type") == "int" and value.get("value") is not None: + try: + params[key] = int(value.get("value")) + except ValueError: + params[key] = value.get("value") + elif value.get("type") == "float" and value.get("value") is not None: + try: + params[key] = float(value.get("value")) + except ValueError: + params[key] = value.get("value") else: params[key] = value.get("value")