Updated error analyzer tools

Author: webarch-ai

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

@@ -15,9 +15,8 @@
 
 from ..common.strands_bedrock_model import create_strands_bedrock_model
 from .tools import (
-    dynamodb_query,
-    dynamodb_record,
-    dynamodb_status,
+    fetch_document_record,
+    fetch_recent_records,
     lambda_lookup,
     search_cloudwatch_logs,
     stepfunction_details,
@@ -51,9 +50,8 @@ def create_error_analyzer_agent(
     # Create agent with specific tools - let LLM choose directly
     tools = [
         search_cloudwatch_logs,
-        dynamodb_record,
-        dynamodb_status,
-        dynamodb_query,
+        fetch_document_record,
+        fetch_recent_records,
         lambda_lookup,
         stepfunction_details,
         xray_trace,

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

@@ -15,9 +15,8 @@
 
 from .cloudwatch_tool import search_cloudwatch_logs
 from .dynamodb_tool import (
-    dynamodb_query,
-    dynamodb_record,
-    dynamodb_status,
+    fetch_document_record,
+    fetch_recent_records,
 )
 from .lambda_tool import lambda_lookup
 from .stepfunction_tool import stepfunction_details
@@ -29,9 +28,8 @@
 __all__ = [
     "search_cloudwatch_logs",
     "lambda_lookup",
-    "dynamodb_record",
-    "dynamodb_status",
-    "dynamodb_query",
+    "fetch_document_record",
+    "fetch_recent_records",
     "stepfunction_details",
     "xray_trace",
     "xray_performance_analysis",

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

@@ -15,7 +15,7 @@
 from strands import tool
 
 from ..config import create_error_response, safe_int_conversion
-from .dynamodb_tool import dynamodb_record
+from .dynamodb_tool import fetch_document_record
 from .models import LogEvent
 from .xray_tool import extract_lambda_request_ids
 
@@ -238,7 +238,7 @@ def _get_document_context(document_id: str) -> Dict[str, Any]:
     """
     Get document context from DynamoDB and extract X-Ray information.
     """
-    dynamodb_response = dynamodb_record(document_id)
+    dynamodb_response = fetch_document_record(document_id)
     if not dynamodb_response.get("document_found"):
         return {
             "analysis_type": "document_not_found",