improve typing

Author: kazmer97

lib/idp_common_pkg/idp_common/extraction/agentic_idp.py

@@ -1098,7 +1098,7 @@ async def structured_output_async(
         )
 
         review_response = await invoke_agent_with_retry(
-            agent=agent, input=review_prompt
+            agent=agent, input=[review_prompt]
         )
         logger.debug("Review response received", extra={"review_completed": True})
 

lib/idp_common_pkg/idp_common/utils/bedrock_utils.py

@@ -19,7 +19,16 @@
     InvokeModelRequestTypeDef,
     InvokeModelResponseTypeDef,
 )
-from strands.types.exceptions import ModelThrottledException
+
+# Optional import for strands-agents (may not be installed in all environments)
+try:
+    from strands.types.exceptions import ModelThrottledException
+
+    _STRANDS_AVAILABLE = True
+except ImportError:
+    _STRANDS_AVAILABLE = False
+    # Create a placeholder exception class that will never match
+    ModelThrottledException = type("ModelThrottledException", (Exception,), {})  # type: ignore[misc, assignment]
 
 # Configure logger
 logger = logging.getLogger(__name__)
@@ -42,8 +51,9 @@
 }
 
 # Default retryable exception types (caught by isinstance check)
+# Only include ModelThrottledException if strands is available
 DEFAULT_RETRYABLE_EXCEPTION_TYPES: tuple[type[Exception], ...] = (
-    ModelThrottledException,
+    (ModelThrottledException,) if _STRANDS_AVAILABLE else ()
 )