lostecho/edge-tts
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix aiohttp timeout issue

Browse Source 
This addresses the issue described in https://github.com/rany2/edge-tts/issues/190#issuecomment-2116195342

Signed-off-by: rany <rany2@riseup.net>
This commit is contained in:
rany
2024-05-17 13:14:56 +03:00
parent 6d9299a669
commit 547d91dfd4
1 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/edge_tts/communicate.py
View File

@@ -260,7 +260,8 @@ class Communicate:
volume: str = "+0%", volume: str = "+0%",
pitch: str = "+0Hz", pitch: str = "+0Hz",
proxy: Optional[str] = None, proxy: Optional[str] = None,
receive_timeout: int = 5, connect_timeout: int = 10,
receive_timeout: int = 60,
): ):
""" """
Initializes the Communicate class. Initializes the Communicate class.
@@ -306,9 +307,14 @@ class Communicate:
raise TypeError("proxy must be str") raise TypeError("proxy must be str")
self.proxy: Optional[str] = proxy self.proxy: Optional[str] = proxy
if not isinstance(receive_timeout, int): if not isinstance(connect_timeout, int) or not isinstance(receive_timeout, int):
raise TypeError("receive_timeout must be int") raise TypeError("connect_timeout and receive_timeout must be int")
self.receive_timeout: int = receive_timeout self.session_timeout = aiohttp.ClientTimeout(
total=None,
connect=None,
sock_connect=connect_timeout,
sock_read=receive_timeout,
)
async def stream(self) -> AsyncGenerator[Dict[str, Any], None]: async def stream(self) -> AsyncGenerator[Dict[str, Any], None]:
"""Streams audio and metadata from the service.""" """Streams audio and metadata from the service."""
@@ -390,11 +396,11 @@ class Communicate:
ssl_ctx = ssl.create_default_context(cafile=certifi.where()) ssl_ctx = ssl.create_default_context(cafile=certifi.where())
async with aiohttp.ClientSession( async with aiohttp.ClientSession(
trust_env=True, trust_env=True,
timeout=self.session_timeout,
) as session, session.ws_connect( ) as session, session.ws_connect(
f"{WSS_URL}&ConnectionId={connect_id()}", f"{WSS_URL}&ConnectionId={connect_id()}",
compress=15, compress=15,
proxy=self.proxy, proxy=self.proxy,
receive_timeout=self.receive_timeout,
headers={ headers={
"Pragma": "no-cache", "Pragma": "no-cache",
"Cache-Control": "no-cache", "Cache-Control": "no-cache",