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

fix: Runflow component output error (#6942)

Browse source 
Co-authored-by: caojianwei-jk <caojianwei-jk@360shuke.com>
This commit is contained in:
jeevi cao 2025-05-10 04:35:47 +08:00 • committed by GitHub
commit 5026124953
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 9 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

17
src/backend/base/langflow/base/tools/run_flow.py
View file

@ -79,7 +79,9 @@ class RunFlowBaseComponent(Component):
if isinstance(first_output, Data): if isinstance(first_output, Data):
return first_output return first_output
message_data = first_output.outputs[0].results["message"].data # just adaptive output Message
_, message_result = next(iter(run_outputs[0].outputs[0].results.items()))
message_data = message_result.data
return Data(data=message_data) return Data(data=message_data)
async def dataframe_output(self) -> DataFrame: async def dataframe_output(self) -> DataFrame:
@ -90,21 +92,20 @@ class RunFlowBaseComponent(Component):
if isinstance(first_output, DataFrame): if isinstance(first_output, DataFrame):
return first_output return first_output
message_data = first_output.outputs[0].results["message"].data # just adaptive output Message
_, message_result = next(iter(run_outputs[0].outputs[0].results.items()))
message_data = message_result.data
return DataFrame(data=message_data if isinstance(message_data, list) else [message_data]) return DataFrame(data=message_data if isinstance(message_data, list) else [message_data])
async def message_output(self) -> Message: async def message_output(self) -> Message:
"""Return the message output.""" """Return the message output."""
run_outputs = await self.run_flow_with_tweaks() run_outputs = await self.run_flow_with_tweaks()
message_result = run_outputs[0].outputs[0].results["message"] _, message_result = next(iter(run_outputs[0].outputs[0].results.items()))
if isinstance(message_result, Message): if isinstance(message_result, Message):
return message_result return message_result
if isinstance(message_result, str): if isinstance(message_result, str):
return Message(content=message_result) return Message(text=message_result)
return Message(text=message_result.data["text"])
return Message(content=message_result.data["text"])
async def get_flow_names(self) -> list[str]: async def get_flow_names(self) -> list[str]:
# TODO: get flfow ID with flow name # TODO: get flfow ID with flow name