Spaces:
Sleeping
Sleeping
File size: 1,658 Bytes
8c48bb1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 |
# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.
"""Utility functions for OpenEnv core."""
import asyncio
import concurrent.futures
def run_async_safely(coro):
"""
Run an async coroutine safely from any context.
This handles the case where we may already be inside an async event loop
(e.g., when called from an async framework). In that case, asyncio.run()
would fail, so we use a ThreadPoolExecutor to run in a separate thread.
Args:
coro: The coroutine to run
Returns:
The result of the coroutine
"""
try:
loop = asyncio.get_running_loop()
except RuntimeError:
loop = None
if loop is not None:
# Already in async context - run in a thread pool
with concurrent.futures.ThreadPoolExecutor() as pool:
future = pool.submit(asyncio.run, coro)
return future.result()
else:
# No async context - use asyncio.run() directly
return asyncio.run(coro)
def convert_to_ws_url(url: str) -> str:
"""
Convert an HTTP/HTTPS URL to a WS/WSS URL.
Args:
url: The URL to convert.
Returns:
The converted WebSocket URL.
"""
ws_url = url.rstrip("/")
if ws_url.startswith("http://"):
ws_url = "ws://" + ws_url[7:]
elif ws_url.startswith("https://"):
ws_url = "wss://" + ws_url[8:]
elif not ws_url.startswith("ws://") and not ws_url.startswith("wss://"):
ws_url = "ws://" + ws_url
return ws_url
|