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

refactor(run.py): simplify fix_memory_inputs function and handle cases where langchain_object has no prompt attribute

Browse source
This commit is contained in:
Gabriel Almeida 2023-04-19 00:48:06 -03:00
commit 8721083c25
1 changed files with 6 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/backend/langflow/interface/run.py
View file

@ -170,10 +170,12 @@ def fix_memory_inputs(langchain_object):
if langchain_object.memory.memory_key in langchain_object.input_variables:
return
except AttributeError:
if (
langchain_object.memory.memory_key
in langchain_object.prompt.input_variables
):
input_variables = (
langchain_object.prompt.input_variables
if hasattr(langchain_object, "prompt")
else langchain_object.input_keys
)
if langchain_object.memory.memory_key in input_variables:
return
possible_new_mem_key = get_memory_key(langchain_object)