fix(bedrock): refectored using _attributes functions

[satya-turing]

This MR Updates the AWS Bedrock Agent Instrumentation for Multi agent collaboration.

The OpenInference instrumentation for AWS Bedrock implements a sophisticated trace ID-based system to track and correlate events across the lifecycle of agent interactions. This implementation enables detailed observability into the execution flow of Bedrock agents, including model invocations, tool calls, knowledge base lookups, and agent collaborations.

1. SpanManager

The SpanManager class serves as the central component for managing OpenTelemetry spans in the Bedrock instrumentation:

• Hierarchical Span Structure: Maintains a multi-level hierarchy of spans:

  • Chain Spans: Parent spans representing major trace events (e.g., preprocessing, orchestration, postprocessing)
  • Child Spans: Representing specific operations like LLM invocations, tool calls, agent_collaborator etc.

• Trace ID Generation and Management:

  • Extracts trace IDs for each event
  • Uses the format {trace_event}-{trace_id} to ensure uniqueness
  • Maintains mappings between trace IDs and their corresponding spans

• Parent-Child Relationship:

  • Tracks parent-child relationships between spans using the trace ID
  • Allows for proper nesting and context propagation

2. ResponseAccumulator

The _ResponseAccumulator class processes responses from the Bedrock service:

• Event Stream Processing: Handles the event stream from Bedrock's response
• Trace Event Routing: Routes different trace events to appropriate handlers based on event type
• Span Creation and Attribute Setting: Creates spans and sets attributes based on the trace data

3. AttributeExtractor

The AttributeExtractor class extracts and processes attributes from various types of trace data:

• Specialized Extractors: Contains methods for extracting attributes from different event types:

  • Model invocation input/output
  • Tool invocations (action groups, code interpreter, etc.)
  • Knowledge base lookups
  • Agent collaborator interactions

• Standardized Attribute Format: Converts Bedrock-specific data into OpenTelemetry-compatible attributes(we have used methods for creating span attribues using attributes.py)

Multi agent collaborator hierarchy example

cloases #1530

[satya-turing]

Changes availbale in #1656,

Closing This MR