feat: Add run_id parameter to run_flow function (#3834)

* Add run_id parameter to set run ID in graph within load_and_run_flow function

* Add run_id parameter to run_flow call in FlowTool for tracking runs

* Add run_id parameter to CustomComponent's run_flow method

* fix: mypy error arg-type

---------

Co-authored-by: italojohnny <italojohnnydosanjos@gmail.com>

src/backend/base/langflow/base/tools/flow_tool.py

@@ -95,10 +95,12 @@ class FlowTool(BaseTool):
     ) -> str:
         """Use the tool asynchronously."""
         tweaks = self.build_tweaks_dict(args, kwargs)
+        run_id = self.graph.run_id if self.graph else None
         run_outputs = await run_flow(
             tweaks={key: {"input_value": value} for key, value in tweaks.items()},
             flow_id=self.flow_id,
             user_id=self.user_id,
+            run_id=run_id,
         )
         if not run_outputs:
             return "No output"

src/backend/base/langflow/custom/custom_component/custom_component.py

@@ -484,6 +484,7 @@ class CustomComponent(BaseComponent):
             flow_name=flow_name,
             tweaks=tweaks,
             user_id=str(self._user_id),
+            run_id=self.graph.run_id,
         )
 
     def list_flows(self) -> list[Data]:

src/backend/base/langflow/helpers/flow.py

@@ -74,10 +74,13 @@ async def run_flow(
     flow_name: Optional[str] = None,
     output_type: Optional[str] = "chat",
     user_id: Optional[str] = None,
+    run_id: Optional[str] = None,
 ) -> List[RunOutputs]:
     if user_id is None:
         raise ValueError("Session is invalid")
     graph = await load_flow(user_id, flow_id, flow_name, tweaks)
+    if run_id:
+        graph.set_run_id(UUID(run_id))
 
     if inputs is None:
         inputs = []