The saved filename should include the original file name instead of a generated one.

google-genai-bot · copybara-github · commit 505d9360072b · 2025-05-21T17:09:47.000-07:00
PiperOrigin-RevId: 761727488
diff --git a/src/google/adk/code_executors/vertex_ai_code_executor.py b/src/google/adk/code_executors/vertex_ai_code_executor.py
@@ -148,18 +148,15 @@ def execute_code(
     )
 
     # Save output file as artifacts.
-    current_timestamp = datetime.datetime.now().strftime('%Y%m%d_%H%M%S')
-    file_name_prefix = '%s_' % str(current_timestamp)
     saved_files = []
     file_count = 0
     for output_file in code_execution_result['output_files']:
       file_type = output_file['name'].split('.')[-1]
-      file_name = file_name_prefix + '%d.%s' % (file_count, file_type)
       if file_type in _SUPPORTED_IMAGE_TYPES:
         file_count += 1
         saved_files.append(
             File(
-                name='plot_' + file_name,
+                name=output_file['name'],
                 content=output_file['contents'],
                 mime_type=f'image/{file_type}',
             )
@@ -168,16 +165,16 @@ def execute_code(
         file_count += 1
         saved_files.append(
             File(
-                name='data_' + file_name,
+                name=output_file['name'],
                 content=output_file['contents'],
                 mime_type=f'text/{file_type}',
             )
         )
       else:
-        mime_type, _ = mimetypes.guess_type(file_name)
+        mime_type, _ = mimetypes.guess_type(output_file['name'])
         saved_files.append(
             File(
-                name=file_name,
+                name=output_file['name'],
                 content=output_file['contents'],
                 mime_type=mime_type,
             )
