Update api_routes_v2.py

api_routes_v2.py

@@ -640,11 +640,14 @@ def _build_api_result_summary(result: Dict[str, Any], session: Dict[str, Any]) -
     return {"summary": summary, "metadata": metadata}
 
 def _add_and_mirror_message(
-    chat_id: str, 
-    role: str, 
-    content: str, 
-    file_metadata: Optional[Dict[str, Any]] = None,
-    pipeline_result: Optional[Dict[str, Any]] = None  # NEW: Add pipeline_result parameter
+    chat_id: str,
+    role: str,
+    content: str,
+    *,
+    pipeline_id: Optional[str] = None,
+    pipeline_status: Optional[str] = None,
+    pipeline_result: Optional[dict] = None,
+    file_metadata: Optional[dict] = None,
 ):
     """
     V3 RULE: Append message to S3 conversation.
@@ -1446,6 +1449,17 @@ async def chat_unified(
                 # Build response based on actual success/failure
                 if is_success:
                     friendly = "🎉 Pipeline completed successfully!"
+                    
+                    # ✅ NEW: persist assistant message with pipeline result
+                    _add_and_mirror_message(
+                        chat_id,
+                        "assistant",
+                        friendly,
+                        pipeline_id=pipeline_id,
+                        pipeline_status="completed",
+                        pipeline_result=result
+                    )
+                    
                     output = {
                         "component_summary": "Pipeline executed successfully",
                         "steps": total_steps,
@@ -1522,6 +1536,16 @@ async def chat_unified(
                     else:
                         friendly = f"⚠️ Pipeline partially completed: {error_msg}"
                     
+                    # ✅ NEW: persist failure result
+                    _add_and_mirror_message(
+                        chat_id,
+                        "assistant",
+                        friendly,
+                        pipeline_id=pipeline_id,
+                        pipeline_status=pipeline_status,
+                        pipeline_result=result
+                    )
+                    
                     output = {
                         "component_summary": f"Pipeline {pipeline_status}",
                         "steps": total_steps,
@@ -1578,6 +1602,23 @@ async def chat_unified(
                         print(f"Warning: Could not get component error details: {comp_error}")
                 
                 friendly = f"❌ Pipeline execution failed: {str(e)}"
+                
+                error_result = {
+                    "pipeline_id": pipeline_id,
+                    "status": "failed",
+                    "error": str(e)
+                }
+                
+                # ✅ NEW: persist exception result
+                _add_and_mirror_message(
+                    chat_id,
+                    "assistant",
+                    friendly,
+                    pipeline_id=pipeline_id,
+                    pipeline_status="failed",
+                    pipeline_result=error_result
+                )
+                
                 api_data = {
                     "type": "error",
                     "error_code": "PIPELINE_EXECUTION_FAILED",