jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

🐛 fix(loading.py): handle conversion of max_tokens from string to int if necessary

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-07-08 15:40:05 -03:00
commit bd9bd13f57
1 changed files with 5 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

13
src/backend/langflow/interface/initialize/loading.py
View file

@ -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)