Update agent.py

agent.py

@@ -9,7 +9,7 @@ def init_async():
     torch.classes.__path__ = [] # add this line to manually set it to empty. 
 
 
-def run_async_task(async_func, *args):
+def run_async_task(async_func, **kwargs):
     """
     Run an asynchronous function in a new event loop.
 
@@ -25,7 +25,9 @@ def run_async_task(async_func, *args):
 
     try:
         loop = asyncio.new_event_loop()
-        loop.run_until_complete(async_func(*args))
+        values = [val for val in kwargs.values()]
+
+        loop.run_until_complete(async_func(values))
     except:
         # Close the existing loop if open
         if loop is not None: