From bd5ecb876367799f6998450a5f82f67d3344c333 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 15 Feb 2024 16:35:30 -0300 Subject: [PATCH] Add comments to clarify the structure of the result dict --- src/backend/langflow/components/chains/RetrievalQA.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/backend/langflow/components/chains/RetrievalQA.py b/src/backend/langflow/components/chains/RetrievalQA.py index ba119ec2e..bfa3b2e11 100644 --- a/src/backend/langflow/components/chains/RetrievalQA.py +++ b/src/backend/langflow/components/chains/RetrievalQA.py @@ -45,4 +45,6 @@ class RetrievalQAComponent(CustomComponent): self.status = runnable result = runnable.invoke({input_key: inputs}) result = result.content if hasattr(result, "content") else result + # Result is a dict with keys "query", "result" and "source_documents" + # for now we just return the result return result.get("result")