From 0cf1fb09a9618c452ba21445c35cd59c2a85d4a5 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 4 Aug 2023 00:20:16 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(prompt=5Frunner.py):=20fix?= =?UTF-8?q?=20issue=20with=20accessing=20result=20content=20when=20invokin?= =?UTF-8?q?g=20chain=20with=20an=20empty=20dict=20input?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/components/chains/prompt_runner.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/components/chains/prompt_runner.py b/src/backend/langflow/components/chains/prompt_runner.py index ebc1d0167..141941c38 100644 --- a/src/backend/langflow/components/chains/prompt_runner.py +++ b/src/backend/langflow/components/chains/prompt_runner.py @@ -25,7 +25,9 @@ class PromptRunner(CustomComponent): prompt: PromptTemplate, ) -> Document: chain = prompt | llm - result = chain.invoke() - result = result[chain.output_key] + # The input is an empty dict because the prompt is already filled + result = chain.invoke({}) + if hasattr(result, "content"): + result = result.content self.repr_value = result return Document(page_content=str(result))