ai-agent-book 精选快照（<2MB 代码与文档，来自 github.com/bojieli/ai-agent-book）

chapter9/gaia-experience/AWorld/aworld/events/util.py

@@ -0,0 +1,54 @@
+# coding: utf-8
+# Copyright (c) 2025 inclusionAI.
+from typing import Callable, Any
+
+from aworld.core.context.base import Context
+from aworld.core.event import eventbus
+from aworld.core.event.base import Message, Constants
+from aworld.events.manager import EventManager
+from aworld.utils.common import sync_exec
+
+
+def subscribe(category: str, key: str = None):
+    """Subscribe the special event to handle.
+
+    Examples:
+        >>> cate = Constants.TOOL or Constants.AGENT; key = "topic"
+        >>> @subscribe(category=cate, key=key)
+        >>> def example(message: Message) -> Message | None:
+        >>>     print("do something")
+
+    Args:
+         category: Types of subscription events, the value is `agent` or `tool`, etc.
+         key: The index key of the handler.
+    """
+    def decorator(func: Callable[..., Any]) -> Callable[..., Any]:
+        topic = key
+        if not topic:
+            topic = category
+        sync_exec(eventbus.subscribe, category, topic, func)
+        return func
+
+    return decorator
+
+
+async def _send_message(msg: Message) -> str:
+    context = msg.context
+    if not context:
+        context = Context()
+
+    event_mng = context.event_manager
+    if not event_mng:
+        event_mng = EventManager(context)
+
+    await event_mng.emit_message(msg)
+    return msg.id
+
+
+async def send_message(msg: Message):
+    """Utility function of send event.
+
+    Args:
+        msg: The content and meta information to be sent.
+    """
+    await _send_message(msg)