From 6cf9639128cbf151e3c94f675be5d8b048110a1a Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 28 Jun 2023 10:58:55 -0300 Subject: [PATCH] feat: add template to input_keys_response --- src/backend/langflow/api/utils.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/backend/langflow/api/utils.py b/src/backend/langflow/api/utils.py index cd7daa5d2..12cd3d76b 100644 --- a/src/backend/langflow/api/utils.py +++ b/src/backend/langflow/api/utils.py @@ -44,4 +44,7 @@ def build_input_keys_response(langchain_object): # Add memory variables to memory_keys input_keys_response["memory_keys"] = langchain_object.memory.memory_variables + if hasattr(langchain_object, "prompt"): + input_keys_response["template"] = langchain_object.prompt.template + return input_keys_response