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

🐛 fix(utils.py): add condition to check if prompt has 'partial' attribute before calling it to prevent error

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-14 17:17:57 -03:00
commit e0f625e793
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/backend/langflow/interface/initialize/utils.py
View file

@ -51,7 +51,9 @@ def handle_partial_variables(prompt, format_kwargs: Dict):
} }
# Remove handle_keys otherwise LangChain raises an error # Remove handle_keys otherwise LangChain raises an error
partial_variables.pop("handle_keys", None) partial_variables.pop("handle_keys", None)
return prompt.partial(**partial_variables) if partial_variables and hasattr(prompt, "partial"):
return prompt.partial(**partial_variables)
return prompt
def handle_variable(params: Dict, input_variable: str, format_kwargs: Dict): def handle_variable(params: Dict, input_variable: str, format_kwargs: Dict):