Merge pull request #377 from cnoe-io/fix-bedrock

fix: correctly handle aws bedrock streaming format

Author: suwhang-cisco

ai_platform_engineering/multi_agents/platform_engineer/protocol_bindings/a2a/agent.py

@@ -105,10 +105,27 @@ async def stream(self, query, context_id, trace_id=None) -> AsyncIterable[dict[s
                       "content": "",
                   }
               elif isinstance(message, AIMessageChunk):
+                  # Normalize content to string (AWS Bedrock returns list, OpenAI returns string)
+                  content = message.content
+                  if isinstance(content, list):
+                      # If content is a list (AWS Bedrock), extract text from content blocks
+                      text_parts = []
+                      for item in content:
+                          if isinstance(item, dict):
+                              # Extract text from Bedrock content block: {"type": "text", "text": "..."}
+                              text_parts.append(item.get('text', ''))
+                          elif isinstance(item, str):
+                              text_parts.append(item)
+                          else:
+                              text_parts.append(str(item))
+                      content = ''.join(text_parts)
+                  elif not isinstance(content, str):
+                      content = str(content) if content else ''
+
                   yield {
                       "is_task_complete": False,
                       "require_user_input": False,
-                      "content": message.content,
+                      "content": content,
                   }
 
       except Exception as e:

ai_platform_engineering/multi_agents/platform_engineer/protocol_bindings/a2a/agent_executor.py

@@ -135,6 +135,24 @@ async def execute(
                     logger.debug("Executor: Received A2A Task event; enqueuing.")
                     await self._safe_enqueue_event(event_queue, event)
                     continue
+                # Normalize content to string (handle cases where AWS Bedrock returns list)
+                # This is due to AWS Bedrock having a different format for the content for streaming compared to Azure OpenAI.
+                content = event.get('content', '')
+                if isinstance(content, list):
+                    # If content is a list (AWS Bedrock), extract text from content blocks
+                    text_parts = []
+                    for item in content:
+                        if isinstance(item, dict):
+                            # Extract text from Bedrock content block: {"type": "text", "text": "..."}
+                            text_parts.append(item.get('text', ''))
+                        elif isinstance(item, str):
+                            text_parts.append(item)
+                        else:
+                            text_parts.append(str(item))
+                    content = ''.join(text_parts)
+                elif not isinstance(content, str):
+                    content = str(content) if content else ''
+
                 if event['is_task_complete']:
                   logger.info("Task complete event received. Enqueuing TaskArtifactUpdateEvent and TaskStatusUpdateEvent.")
                   await self._safe_enqueue_event(
@@ -147,7 +165,7 @@ async def execute(
                       artifact=new_text_artifact(
                         name='current_result',
                         description='Result of request to agent.',
-                        text=event['content'],
+                        text=content,
                       ),
                     )
                   )
@@ -169,7 +187,7 @@ async def execute(
                       status=TaskStatus(
                         state=TaskState.input_required,
                         message=new_agent_text_message(
-                          event['content'],
+                          content,
                           task.context_id,
                           task.id,
                         ),
@@ -188,7 +206,7 @@ async def execute(
                       status=TaskStatus(
                         state=TaskState.working,
                         message=new_agent_text_message(
-                          event['content'],
+                          content,
                           task.context_id,
                           task.id,
                         ),