diff --git a/src/promptflow/promptflow/_sdk/operations/_test_submitter.py b/src/promptflow/promptflow/_sdk/operations/_test_submitter.py index 12757f346ee..bd4d653e1af 100644 --- a/src/promptflow/promptflow/_sdk/operations/_test_submitter.py +++ b/src/promptflow/promptflow/_sdk/operations/_test_submitter.py @@ -206,6 +206,9 @@ def node_test( connections=connections, working_dir=self.flow.code, ) + result.output = persist_multimedia_data( + result.output, base_dir=self.flow.code, sub_dir=Path(".promptflow/output") + ) return result def _chat_flow(self, inputs, chat_history_name, environment_variables: dict = None, show_step_output=False):