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

🐛 fix(loading.py): convert "max_tokens" parameter from string to int if it is a string and can be converted to int

Browse source 
🐛 fix(loading.py): remove "max_tokens" parameter if it is not an integer
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-07-08 13:59:46 -03:00
commit 7e712b1be8
1 changed files with 9 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -120,6 +120,15 @@ def instantiate_llm(node_type, class_object, params: Dict):
# False if condition is True # False if condition is True
if node_type == "VertexAI": if node_type == "VertexAI":
return initialize_vertexai(class_object=class_object, params=params) 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)
return class_object(**params) return class_object(**params)