From b9fa68dcfbf5e82a3d2ba6054f72a15895e89335 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@langflow.org>
Date: Thu, 16 May 2024 16:50:00 -0700
Subject: [PATCH] feat: Add get_final_results_only parameter to
 build_records_from_result_data method (#1912)

* feat: Add get_final_results_only parameter to build_records_from_result_data method

The `build_records_from_result_data` method in the `SubFlowComponent` class now accepts a `get_final_results_only` parameter. This parameter controls whether the output will contain all outputs from the flow or only the final results. This change allows for more flexibility in handling the result data.
---
 .../components/experimental/SubFlow.py        | 19 ++++++++++++++-----
 1 file changed, 14 insertions(+), 5 deletions(-)

diff --git a/src/backend/base/langflow/components/experimental/SubFlow.py b/src/backend/base/langflow/components/experimental/SubFlow.py
index 72e07c8e86..80e15c6ada 100644
--- a/src/backend/base/langflow/components/experimental/SubFlow.py
+++ b/src/backend/base/langflow/components/experimental/SubFlow.py
@@ -35,7 +35,7 @@ def update_build_config(self, build_config: dotdict, field_value: Any, field_nam
             build_config["flow_name"]["options"] = self.get_flow_names()
         # Clean up the build config
         for key in list(build_config.keys()):
-            if key not in self.field_order + ["code", "_type"]:
+            if key not in self.field_order + ["code", "_type", "get_final_results_only"]:
                 del build_config[key]
         if field_value is not None and field_name == "flow_name":
             try:
@@ -85,20 +85,29 @@ def build_config(self):
                 "display_name": "Tweaks",
                 "info": "Tweaks to apply to the flow.",
             },
+            "get_final_results_only": {
+                "display_name": "Get Final Results Only",
+                "info": "If False, the output will contain all outputs from the flow.",
+                "advanced": True,
+            },
         }
 
-    def build_records_from_result_data(self, result_data: ResultData) -> List[Record]:
+    def build_records_from_result_data(self, result_data: ResultData, get_final_results_only: bool) -> List[Record]:
         messages = result_data.messages
         if not messages:
             return []
         records = []
         for message in messages:
             message_dict = message if isinstance(message, dict) else message.model_dump()
-            record = Record(data={"result": result_data.model_dump(), "message": message_dict.get("message", "")})
+            if get_final_results_only:
+                result_data_dict = result_data.model_dump()
+                results = result_data_dict.get("results", {})
+                inner_result = results.get("result", {})
+            record = Record(data={"result": inner_result, "message": message_dict}, text_key="result")
             records.append(record)
         return records
 
-    async def build(self, flow_name: str, **kwargs) -> List[Record]:
+    async def build(self, flow_name: str, get_final_results_only: bool = True, **kwargs) -> List[Record]:
         tweaks = {key: {"input_value": value} for key, value in kwargs.items()}
         run_outputs: List[Optional[RunOutputs]] = await self.run_flow(
             tweaks=tweaks,
@@ -112,7 +121,7 @@ async def build(self, flow_name: str, **kwargs) -> List[Record]:
         if run_output is not None:
             for output in run_output.outputs:
                 if output:
-                    records.extend(self.build_records_from_result_data(output))
+                    records.extend(self.build_records_from_result_data(output, get_final_results_only))
 
         self.status = records
         logger.debug(records)