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

Add comments to clarify the structure of the result dict

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-15 16:35:30 -03:00
commit bd5ecb8763
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -45,4 +45,6 @@ class RetrievalQAComponent(CustomComponent):
self.status = runnable self.status = runnable
result = runnable.invoke({input_key: inputs}) result = runnable.invoke({input_key: inputs})
result = result.content if hasattr(result, "content") else result 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") return result.get("result")