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

🐛 fix(types.py): handle case where prompt template is not present in PromptVertex

Browse source 
ℹ️ The code was modified to handle a case where the `template` attribute is not present in the `PromptVertex` class. If the `template` attribute is not found, the code checks if the `prompt` attribute is present and uses its `template` attribute instead. This change ensures that the code does not break when the `template` attribute is missing.
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-14 17:16:54 -03:00
commit 2144e1ec91
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/backend/langflow/graph/vertex/types.py
View file

@ -226,7 +226,12 @@ class PromptVertex(Vertex):
# so the prompt format doesn't break # so the prompt format doesn't break
artifacts.pop("handle_keys", None) artifacts.pop("handle_keys", None)
try: try:
template = self._built_object.template if not hasattr(self._built_object, "template") and hasattr(
self._built_object, "prompt"
):
template = self._built_object.prompt.template
else:
template = self._built_object.template
for key, value in artifacts.items(): for key, value in artifacts.items():
if value: if value:
replace_key = "{" + key + "}" replace_key = "{" + key + "}"