init

wechat-bot/bot/wecom_bot.py

@@ -0,0 +1,104 @@
+"""
+企业微信桥接模块
+使用 wecom-aibot-python-sdk 与企业微信建立 WebSocket 长连接，
+接收用户消息后通过 ADK API 调用远程 Agent，再将 Agent 回复发送给用户。
+
+本模块仅负责：
+- 创建 WSClient 实例
+- 注册事件和消息处理器
+"""
+import json
+import logging
+import traceback
+
+from aibot import WSClient, WSClientOptions
+
+from config import Config
+from bot.agent_client import close_http_client
+from bot.handlers import handle_message
+from bot.message_parser import extract_metadata
+
+logger = logging.getLogger(__name__)
+
+
+def create_bot_client() -> WSClient:
+    """
+    创建企业微信机器人 WebSocket 客户端，并注册所有事件和消息处理器。
+    """
+    ws_client = WSClient(
+        WSClientOptions(
+            bot_id=Config.WECHAT_BOT_ID,
+            secret=Config.WECHAT_BOT_SECRET,
+        )
+    )
+
+    # ---- 事件处理器 ----
+
+    @ws_client.on("authenticated")
+    def on_authenticated():
+        logger.info("企业微信机器人认证成功")
+
+    @ws_client.on("event.enter_chat")
+    async def on_enter_chat(frame):
+        """用户进入会话，发送欢迎语"""
+        meta = extract_metadata(frame)
+        user_id = meta.get("user_id", "")
+        logger.info("用户进入会话: %s", user_id)
+
+        try:
+            await ws_client.reply_welcome(
+                frame,
+                {
+                    "msgtype": "text",
+                    "text": {
+                        "content": "您好！我是智能助手，有什么可以帮您的吗？"
+                    },
+                },
+            )
+        except Exception as e:
+            logger.error("发送欢迎语失败: %s", e)
+
+    @ws_client.on("event.feedback_event")
+    async def on_feedback(frame):
+        """用户反馈事件"""
+        body = frame.get("body", {})
+        logger.info("收到用户反馈: %s", json.dumps(body, ensure_ascii=False))
+
+    @ws_client.on("disconnected")
+    def on_disconnected(reason):
+        logger.warning("连接被断开（可能被新连接踢掉）: %s", reason)
+
+    # ---- 消息处理器 ----
+
+    @ws_client.on("message.text")
+    async def on_text(frame):
+        await _safe_handle(ws_client, frame, "text")
+
+    @ws_client.on("message.image")
+    async def on_image(frame):
+        await _safe_handle(ws_client, frame, "image")
+
+    @ws_client.on("message.voice")
+    async def on_voice(frame):
+        await _safe_handle(ws_client, frame, "voice")
+
+    @ws_client.on("message.file")
+    async def on_file(frame):
+        # file 类型可能包含视频
+        body = frame.get("body", {})
+        msg_type = "video" if body.get("video") else "file"
+        await _safe_handle(ws_client, frame, msg_type)
+
+    @ws_client.on("message.mixed")
+    async def on_mixed(frame):
+        await _safe_handle(ws_client, frame, "mixed")
+
+    return ws_client
+
+
+async def _safe_handle(ws_client, frame: dict, msg_type: str):
+    """安全地处理消息，捕获异常防止崩溃"""
+    try:
+        await handle_message(ws_client, frame, msg_type)
+    except Exception as e:
+        logger.error("处理%s消息异常: %s\n%s", msg_type, e, traceback.format_exc())