Increasing timeout for remote diarization auth

Wordcab · Jun 28, 2024 · 53dc35f · 53dc35f
1 parent 0bfaeb3
commit 53dc35f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/wordcab_transcribe/services/asr_service.py b/src/wordcab_transcribe/services/asr_service.py
@@ -880,7 +880,8 @@ async def remote_diarization(
         if not settings.debug:
             headers = {"Content-Type": "application/x-www-form-urlencoded"}
             auth_url = f"{url}/api/v1/auth"
-            async with aiohttp.ClientSession() as session:
+            diarization_timeout = aiohttp.ClientTimeout(total=60)
+            async with AsyncLocationTrustedRedirectSession(timeout=diarization_timeout) as session:
                 async with session.post(
                     url=auth_url,
                     data={"username": settings.username, "password": settings.password},