jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Update display name and return result as dictionary

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-07 15:55:23 -03:00
commit 52494fbf74
1 changed files with 4 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/backend/langflow/components/chains/RetrievalQA.py
View file

@ -8,7 +8,7 @@ from langflow.field_typing import BaseMemory, BaseRetriever, Text
class RetrievalQAComponent(CustomComponent): class RetrievalQAComponent(CustomComponent):
display_name = "RetrievalQA" display_name = "Retrieval QA"
description = "Chain for question-answering against an index." description = "Chain for question-answering against an index."
def build_config(self): def build_config(self):
@ -42,6 +42,7 @@ class RetrievalQAComponent(CustomComponent):
) )
if isinstance(inputs, Document): if isinstance(inputs, Document):
inputs = inputs.page_content inputs = inputs.page_content
self.status = runnable
result = runnable.invoke({input_key: inputs}) 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")