improve retry mechanism

kazmer97 · kazmer97 · commit 24c9d87cbc74 · 2025-12-15T12:23:50.000Z
diff --git a/lib/idp_common_pkg/idp_common/utils/bedrock_utils.py b/lib/idp_common_pkg/idp_common/utils/bedrock_utils.py
@@ -19,6 +19,7 @@
     InvokeModelRequestTypeDef,
     InvokeModelResponseTypeDef,
 )
+from strands.models.bedrock import ModelThrottledException
 
 # Optional import for strands-agents (may not be installed in all environments)
 try:

Original file line number	Diff line number	Diff line change
`@@ -19,6 +19,7 @@`
`19`	`19`	`InvokeModelRequestTypeDef,`
`20`	`20`	`InvokeModelResponseTypeDef,`
`21`	`21`	`)`
	`22`	`+from strands.models.bedrock import ModelThrottledException`
`22`	`23`
`23`	`24`	`# Optional import for strands-agents (may not be installed in all environments)`
`24`	`25`	`try:`