mirror of
https://github.com/invariantlabs-ai/invariant-gateway.git
synced 2026-05-25 08:14:02 +02:00
Hemang
43 lines
1.2 KiB
Python
43 lines
1.2 KiB
Python
"""Task utilities for running async functions"""
|
|
|
|
import asyncio
|
|
import concurrent.futures
|
|
|
|
from contextlib import redirect_stdout
|
|
from typing import Any
|
|
|
|
from gateway.mcp.log import MCP_LOG_FILE
|
|
|
|
|
|
def run_task_sync(async_func, *args, **kwargs) -> Any:
|
|
"""
|
|
Runs an asynchronous function synchronously in a separate
|
|
thread with its own event loop. This function blocks the calling
|
|
thread until completion or timeout (10 seconds).
|
|
|
|
Args:
|
|
async_func: The async function to run
|
|
*args: Positional arguments to pass to the async function
|
|
**kwargs: Keyword arguments to pass to the async function
|
|
|
|
Returns:
|
|
Any: The return value of the async function
|
|
"""
|
|
|
|
def run_in_new_loop():
|
|
loop = asyncio.new_event_loop()
|
|
try:
|
|
return loop.run_until_complete(
|
|
async_func(
|
|
*args,
|
|
**kwargs,
|
|
)
|
|
)
|
|
finally:
|
|
loop.close()
|
|
|
|
with redirect_stdout(MCP_LOG_FILE):
|
|
with concurrent.futures.ThreadPoolExecutor() as executor:
|
|
future = executor.submit(run_in_new_loop)
|
|
return future.result(timeout=10.0)
|