Updated xray tool implementation

Author: webarch-ai

lib/idp_common_pkg/idp_common/agents/error_analyzer/agent.py

@@ -15,13 +15,13 @@
 
 from ..common.strands_bedrock_model import create_strands_bedrock_model
 from .tools import (
+    analyze_document_trace,
+    analyze_system_performance,
     analyze_workflow_execution,
     fetch_document_record,
     fetch_recent_records,
     lambda_lookup,
     search_cloudwatch_logs,
-    xray_performance_analysis,
-    xray_trace,
 )
 
 logger = logging.getLogger(__name__)
@@ -54,8 +54,8 @@ def create_error_analyzer_agent(
         fetch_recent_records,
         lambda_lookup,
         analyze_workflow_execution,
-        xray_trace,
-        xray_performance_analysis,
+        analyze_document_trace,
+        analyze_system_performance,
     ]
     bedrock_model = create_strands_bedrock_model(
         model_id=config.agents.error_analyzer.model_id, boto_session=session

lib/idp_common_pkg/idp_common/agents/error_analyzer/tools/__init__.py

@@ -21,8 +21,8 @@
 from .lambda_tool import lambda_lookup
 from .stepfunction_tool import analyze_workflow_execution
 from .xray_tool import (
-    xray_performance_analysis,
-    xray_trace,
+    analyze_document_trace,
+    analyze_system_performance,
 )
 
 __all__ = [
@@ -31,6 +31,6 @@
     "fetch_document_record",
     "fetch_recent_records",
     "analyze_workflow_execution",
-    "xray_trace",
-    "xray_performance_analysis",
+    "analyze_document_trace",
+    "analyze_system_performance",
 ]