From dde9a5bde2df22eb59911d4e363410ad47ed6002 Mon Sep 17 00:00:00 2001 From: Gabriel Almeida Date: Wed, 5 Apr 2023 18:07:44 -0300 Subject: [PATCH] fix: add check for input_variables --- src/backend/langflow/interface/run.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/interface/run.py b/src/backend/langflow/interface/run.py index 0eaea19a76..c413dfc4f4 100644 --- a/src/backend/langflow/interface/run.py +++ b/src/backend/langflow/interface/run.py @@ -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: