mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-11-04 07:30:21 +00:00
152fb8ac4f
Require ReMe 0.4.2.1 and verify isolated Home wheel installation in package CI. Share cancellation-safe cleanup with thread calls, retain AnyIO cancellation semantics, and record Bubblewrap mount targets during argument construction.
32 lines
1 KiB
Python
32 lines
1 KiB
Python
"""Small asyncio helpers."""
|
|
|
|
import asyncio
|
|
from collections.abc import Awaitable, Callable
|
|
from contextlib import suppress
|
|
from typing import Any
|
|
|
|
|
|
async def finish_cleanup(operation: Awaitable):
|
|
"""Finish cleanup before propagating any cancellation received while waiting."""
|
|
task = asyncio.ensure_future(operation)
|
|
cancellation: asyncio.CancelledError | None = None
|
|
while not task.done():
|
|
try:
|
|
await asyncio.shield(task)
|
|
except asyncio.CancelledError as exc:
|
|
if cancellation is None:
|
|
cancellation = exc
|
|
except Exception:
|
|
if cancellation is None:
|
|
raise
|
|
break
|
|
if cancellation is not None:
|
|
with suppress(Exception):
|
|
task.result()
|
|
raise cancellation
|
|
return task.result()
|
|
|
|
|
|
async def complete_in_thread(func: Callable[..., Any], /, *args) -> Any:
|
|
"""Finish a side-effecting thread call, including through repeated cancellation."""
|
|
return await finish_cleanup(asyncio.to_thread(func, *args))
|