Update display name and return result as dictionary
This commit is contained in:
parent
5d7f00c5ad
commit
52494fbf74
1 changed files with 4 additions and 3 deletions
|
|
@ -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")
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue