diff --git a/kb_workflow/kb_workflow.py b/kb_workflow/kb_workflow.py
index c61a560..c5fda0f 100644
--- a/kb_workflow/kb_workflow.py
+++ b/kb_workflow/kb_workflow.py
@@ -80,10 +80,16 @@ def process_command(
         # Run async pipeline
         result = asyncio.run(run_pipeline())
 
+        # Extract archive_file from results if available
+        archive_file = None
+        if result.results and "archive" in result.results:
+            archive_file = result.results["archive"].get("archive_file")
+
         return {
             "success": result.success,
             "error": result.error_message,
             "content_dir": str(result.content_dir) if result.content_dir else None,
+            "archive_file": archive_file,
             "results": result.results,
         }