Update services/pipeline_executor.py

services/pipeline_executor.py

@@ -173,8 +173,8 @@ Execute each component by calling the corresponding tool."""
         }
         
         for iteration in range(max_iterations):
-            # Prepare messages
-            messages = [{"role": "user", "content": user_message}]
+            # Prepare messages (Bedrock converse API format)
+            messages = [{"role": "user", "content": [{"text": user_message}]}]
             messages.extend(conversation_history)
             
             # Call Bedrock directly using converse API
@@ -192,8 +192,8 @@ Execute each component by calling the corresponding tool."""
             assistant_message = response['output']['message']['content'][0]['text']
             print(f"\n🤖 Mistral Response (Iteration {iteration + 1}):\n{assistant_message}\n")
             
-            # Add to conversation
-            conversation_history.append({"role": "assistant", "content": assistant_message})
+            # Add to conversation (Bedrock converse API format)
+            conversation_history.append({"role": "assistant", "content": [{"text": assistant_message}]})
             
             # Check for Final Answer
             if "Final Answer:" in assistant_message or "final answer" in assistant_message.lower():
@@ -261,19 +261,19 @@ Execute each component by calling the corresponding tool."""
                             "executor": "bedrock"
                         }
                         
-                        # Add observation to conversation
+                        # Add observation to conversation (Bedrock converse API format)
                         observation_message = f"Observation: {observation}"
-                        conversation_history.append({"role": "user", "content": observation_message})
+                        conversation_history.append({"role": "user", "content": [{"text": observation_message}]})
                         
                     else:
                         # Unknown tool
                         error_msg = f"Unknown tool: {tool_name}"
-                        conversation_history.append({"role": "user", "content": f"Error: {error_msg}"})
+                        conversation_history.append({"role": "user", "content": [{"text": f"Error: {error_msg}"}]})
                         
                 except json.JSONDecodeError as e:
                     # Invalid JSON
                     error_msg = f"Invalid JSON in Action Input: {e}"
-                    conversation_history.append({"role": "user", "content": f"Error: {error_msg}"})
+                    conversation_history.append({"role": "user", "content": [{"text": f"Error: {error_msg}"}]})
             else:
                 # No action found - agent might be confused or done
                 if iteration > 0 and not has_called_tools: