From bd9bd13f57221582387d1df719e8a6a4e2a81478 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sat, 8 Jul 2023 15:40:05 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(loading.py):=20handle=20conv?= =?UTF-8?q?ersion=20of=20max=5Ftokens=20from=20string=20to=20int=20if=20ne?= =?UTF-8?q?cessary?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../langflow/interface/initialize/loading.py | 13 +++++-------- 1 file changed, 5 insertions(+), 8 deletions(-) diff --git a/src/backend/langflow/interface/initialize/loading.py b/src/backend/langflow/interface/initialize/loading.py index df539d12c..9c900f18a 100644 --- a/src/backend/langflow/interface/initialize/loading.py +++ b/src/backend/langflow/interface/initialize/loading.py @@ -121,14 +121,11 @@ def instantiate_llm(node_type, class_object, params: Dict): if node_type == "VertexAI": return initialize_vertexai(class_object=class_object, params=params) # max_tokens sometimes is a string and should be an int - if ( - "max_tokens" in params - and isinstance(params["max_tokens"], str) - and params["max_tokens"].isdigit() - ): - params["max_tokens"] = int(params["max_tokens"]) - elif not isinstance(params["max_tokens"], int): - params.pop("max_tokens", None) + if "max_tokens" in params: + if isinstance(params["max_tokens"], str) and params["max_tokens"].isdigit(): + params["max_tokens"] = int(params["max_tokens"]) + elif not isinstance(params.get("max_tokens"), int): + params.pop("max_tokens", None) return class_object(**params)