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

fix: add check for input_variables

Browse Source
This commit is contained in:
Gabriel Almeida
2023-04-05 18:07:44 -03:00
parent 8eb4cdbbcb
commit dde9a5bde2
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/backend/langflow/interface/run.py
View File

@ -116,8 +116,15 @@ def fix_memory_inputs(langchain_object):
get_memory_key function and updates the memory keys using the update_memory_keys function.
"""
if hasattr(langchain_object, "memory") and langchain_object.memory is not None:
if langchain_object.memory.memory_key in langchain_object.input_variables:
return
try:
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
):
return
possible_new_mem_key = get_memory_key(langchain_object)
if possible_new_mem_key is not None: