ubi integration for agent

RAG/bedrock_agent.py

@@ -128,7 +128,7 @@ def query_(inputs):
     agent_answer = ""
     trace_id = str(uuid.uuid4()).replace("-", "")
     try:
-        for event in event_stream:
+        for i,event in enumerate(event_stream):
             if 'trace' in event: 
                
                 if('orchestrationTrace' not in event['trace']['trace']):