From 8721083c25b27d6ef0f907f0b81e15001efcdcdb Mon Sep 17 00:00:00 2001 From: Gabriel Almeida Date: Wed, 19 Apr 2023 00:48:06 -0300 Subject: [PATCH] refactor(run.py): simplify fix_memory_inputs function and handle cases where langchain_object has no prompt attribute --- src/backend/langflow/interface/run.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/src/backend/langflow/interface/run.py b/src/backend/langflow/interface/run.py index c92f40fec..300e09e01 100644 --- a/src/backend/langflow/interface/run.py +++ b/src/backend/langflow/interface/run.py @@ -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)