From 52494fbf746957b13831122ba040be7f5335c009 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 7 Feb 2024 15:55:23 -0300 Subject: [PATCH] Update display name and return result as dictionary --- src/backend/langflow/components/chains/RetrievalQA.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/components/chains/RetrievalQA.py b/src/backend/langflow/components/chains/RetrievalQA.py index 3fc1933d9..ba119ec2e 100644 --- a/src/backend/langflow/components/chains/RetrievalQA.py +++ b/src/backend/langflow/components/chains/RetrievalQA.py @@ -8,7 +8,7 @@ from langflow.field_typing import BaseMemory, BaseRetriever, Text class RetrievalQAComponent(CustomComponent): - display_name = "RetrievalQA" + display_name = "Retrieval QA" description = "Chain for question-answering against an index." def build_config(self): @@ -42,6 +42,7 @@ class RetrievalQAComponent(CustomComponent): ) if isinstance(inputs, Document): inputs = inputs.page_content - + self.status = runnable result = runnable.invoke({input_key: inputs}) - return result.content if hasattr(result, "content") else result + result = result.content if hasattr(result, "content") else result + return result.get("result")