Skip to content

Commit 3922977

Browse files
reset webrtc connections on reload
1 parent 419ca1d commit 3922977

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

server/api/api.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
from aiohttp import web
2+
import asyncio
23

34
from api.nodes.nodes import list_nodes, install_node, delete_node
45
from api.models.models import list_models, add_model, delete_model
@@ -43,7 +44,13 @@ async def reload(request):
4344

4445
#reload pipeline
4546
request.app["pipeline"] = Pipeline(cwd=request.app["workspace"], disable_cuda_malloc=True, gpu_only=True)
46-
47+
48+
#reset webrtc connections
49+
pcs = request.app["pcs"]
50+
coros = [pc.close() for pc in pcs]
51+
await asyncio.gather(*coros)
52+
pcs.clear()
53+
4754
return web.json_response({"success": True, "error": None})
4855

4956
async def nodes(request):

0 commit comments

Comments
 (0)