有没有办法在google.colab.kernel.comms对象上注册消息处理程序回调？

import asyncio
from google.colab import kernel

# 定义消息处理程序
async def message_handler(msg):
    # 处理收到的消息
    print('Received message:', msg)

# 注册消息处理程序回调
async def register_message_handler():
    # 获取Colab后台内核连接
    kernel._get_connection_info()
    # 连接到Colab后台内核
    shell_channel = kernel.connect_shell()
    # 创建消息处理程序
    message_channel = kernel.Comms(shell_channel)
    # 注册消息处理程序回调
    message_channel.register_callback(message_handler)
    # 等待消息
    await message_channel.wait_closed()

# 运行消息处理程序
asyncio.run(register_message_handler())