From 47f9d78a86a8ce3aa297241ce739ffde974b690a Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 30 Jun 2023 18:27:56 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(utils.py):=20add=20condition?= =?UTF-8?q?=20to=20check=20if=20langchain=5Fobject.prompt=20has=20a=20temp?= =?UTF-8?q?late=20attribute=20before=20adding=20it=20to=20input=5Fkeys=5Fr?= =?UTF-8?q?esponse=20The=20code=20now=20checks=20if=20the=20langchain=5Fob?= =?UTF-8?q?ject.prompt=20has=20a=20template=20attribute=20before=20adding?= =?UTF-8?q?=20it=20to=20the=20input=5Fkeys=5Fresponse.=20This=20prevents?= =?UTF-8?q?=20potential=20errors=20if=20the=20template=20attribute=20is=20?= =?UTF-8?q?not=20present=20in=20the=20prompt=20object.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/api/utils.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/backend/langflow/api/utils.py b/src/backend/langflow/api/utils.py index aa9598814..2384a4089 100644 --- a/src/backend/langflow/api/utils.py +++ b/src/backend/langflow/api/utils.py @@ -51,7 +51,9 @@ def build_input_keys_response(langchain_object, artifacts): # Add memory variables to memory_keys input_keys_response["memory_keys"] = langchain_object.memory.memory_variables - if hasattr(langchain_object, "prompt"): + if hasattr(langchain_object, "prompt") and hasattr( + langchain_object.prompt, "template" + ): input_keys_response["template"] = langchain_object.prompt.template return input_keys_response