fix timeout handling in AWSBedrockLLMService

This commit is contained in:
Eduards Klavins
2025-09-03 11:52:30 +03:00
parent e1477e79f0
commit 0a3e98857e

View File

@@ -880,7 +880,7 @@ class AWSBedrockLLMService(LLMService):
if self._retry_on_timeout:
try:
response = await asyncio.wait_for(
await client.converse_stream(**request_params), timeout=self._retry_timeout_secs
client.converse_stream(**request_params), timeout=self._retry_timeout_secs
)
return response
except (ReadTimeoutError, asyncio.TimeoutError) as e: