diff --git a/python/packages/autogen-core/src/autogen_core/tools/_function_tool.py b/python/packages/autogen-core/src/autogen_core/tools/_function_tool.py index 985d7d1d1201..25185d7e563d 100644 --- a/python/packages/autogen-core/src/autogen_core/tools/_function_tool.py +++ b/python/packages/autogen-core/src/autogen_core/tools/_function_tool.py @@ -115,8 +115,9 @@ async def run(self, args: BaseModel, cancellation_token: CancellationToken) -> A else: result = await self._func(**kwargs) else: + loop = asyncio.get_running_loop() if self._has_cancellation_support: - result = await asyncio.get_event_loop().run_in_executor( + result = await loop.run_in_executor( None, functools.partial( self._func, @@ -125,7 +126,7 @@ async def run(self, args: BaseModel, cancellation_token: CancellationToken) -> A ), ) else: - future = asyncio.get_event_loop().run_in_executor(None, functools.partial(self._func, **kwargs)) + future = loop.run_in_executor(None, functools.partial(self._func, **kwargs)) cancellation_token.link_future(future) result = await future