Update display name and return result as dictionary

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-07 15:55:23 -03:00
commit 52494fbf74

View file

@ -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")