pengjunjie/langflow
1
0
Fork 1
mirror of https://github.com/langflow-ai/langflow.git synced 2026-07-24 16:30:28 +08:00
Code Issues Packages Projects Releases Wiki Activity

🐛 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
parent 79261bf620
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)