FileToLink / Thunder /utils /keepalive.py
Hermes Agent
fix(keepalive,strm): eliminate 8KB header LineTooLong crash and prevent strm rewriting
d703274
Raw History Blame Contribute Delete
962 Bytes
# Thunder/utils/keepalive.py
import asyncio
import aiohttp
from Thunder.vars import Var
from Thunder.utils.logger import logger
async def ping_server():
try:
ping_url = f"http://127.0.0.1:{Var.PORT}/status"
async with aiohttp.ClientSession(
timeout=aiohttp.ClientTimeout(total=10)
) as session:
while True:
try:
await asyncio.sleep(Var.PING_INTERVAL)
async with session.get(ping_url, allow_redirects=False) as resp:
if resp.status not in (200, 302):
logger.warning(f"Ping to {ping_url} returned status {resp.status}.")
except asyncio.CancelledError:
break
except Exception as inner_e:
logger.warning(f"Keepalive ping note: {inner_e}")
except Exception as e:
logger.error(f"Error in ping_server: {e}", exc_info=True)