Increased boto's read timeout to 10 mins.

adivekar-utexas · adivekar-utexas · commit 078367616050 · 2025-07-11T18:01:03.000+05:30
diff --git a/src/fmcore/algorithm/bedrock.py b/src/fmcore/algorithm/bedrock.py
@@ -36,6 +36,7 @@
     TextGenerationParams,
     TextGenerationParamsMapper,
 )
+from botocore.config import Config as BotoConfig
 
 
 class ConfigSelectionStrategy(AutoEnum):
@@ -711,6 +712,7 @@ def refresh_session(self):
             self.bedrock_client = self.boto3_session.client(
                 service_name="bedrock-runtime",
                 region_name=self.current_account_config.region_name,
+                config=BotoConfig(read_timeout=60 * 10),
             )
 
         def cleanup(self):

Original file line number	Diff line number	Diff line change
`@@ -36,6 +36,7 @@`
`36`	`36`	`TextGenerationParams,`
`37`	`37`	`TextGenerationParamsMapper,`
`38`	`38`	`)`
	`39`	`+from botocore.config import Config as BotoConfig`
`39`	`40`
`40`	`41`
`41`	`42`	`class ConfigSelectionStrategy(AutoEnum):`
`@@ -711,6 +712,7 @@ def refresh_session(self):`
`711`	`712`	`self.bedrock_client = self.boto3_session.client(`
`712`	`713`	`service_name="bedrock-runtime",`
`713`	`714`	`region_name=self.current_account_config.region_name,`
	`715`	`+ config=BotoConfig(read_timeout=60 * 10),`
`714`	`716`	`)`
`715`	`717`
`716`	`718`	`def cleanup(self):`