moss-ai/agents/conductor_agent/wechat_tools.py

"""
微信相关工具
提供微信聊天记录获取和消息发送功能
"""

from langchain_core.tools import tool
from pydantic import BaseModel, Field
from typing import List, Optional
import logging
import sys
import os

# 添加父目录到路径以导入backend模块
sys.path.insert(0, os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), '..', 'app', 'backend-python'))

logger = logging.getLogger(__name__)


class WechatChatHistoryArgs(BaseModel):
    """获取微信聊天记录参数"""
    to_user: str = Field(..., description="好友或群聊的备注或昵称")
    target_date: str = Field(..., description="目标日期，格式为YY/M/D，如25/11/10表示2025年11月10日")


class WechatSendMessageArgs(BaseModel):
    """发送单条消息参数"""
    to_user: str = Field(..., description="好友或群聊的备注或昵称")
    message: str = Field(..., description="要发送的消息内容")


class WechatSendMultipleMessagesArgs(BaseModel):
    """发送多条消息给单个好友参数"""
    to_user: str = Field(..., description="好友或群聊的备注或昵称")
    messages: List[str] = Field(..., description="要发送的消息列表")


class WechatSendToMultipleFriendsArgs(BaseModel):
    """发送消息给多个好友参数"""
    to_users: List[str] = Field(..., description="好友或群聊的备注或昵称列表")
    message: str = Field(..., description="要发送的消息内容（单条消息会发给所有好友）")


@tool("get_wechat_chat_history", args_schema=WechatChatHistoryArgs, description="获取微信特定日期的聊天记录")
def get_wechat_chat_history(
    to_user: str,
    target_date: str
) -> str:
    """
    获取微信聊天记录
    
    Args:
        to_user: 好友或群聊的备注或昵称
        target_date: 目标日期，格式为YY/M/D，如25/11/10
        
    Returns:
        聊天记录的文本描述
    """
    try:
        # 导入MCP服务
        from services.wechat_mcp_service import get_wechat_mcp_service
        import asyncio
        
        mcp_service = get_wechat_mcp_service()
        
        # 调用MCP服务
        loop = asyncio.new_event_loop()
        asyncio.set_event_loop(loop)
        result = loop.run_until_complete(
            mcp_service.get_chat_history(to_user, target_date)
        )
        loop.close()
        
        if result and result.get("success"):
            data = result.get("data", "")
            if isinstance(data, str):
                return f"✅ 成功获取与 {to_user} 在 {target_date} 的聊天记录：\n\n{data}"
            else:
                return f"✅ 成功获取聊天记录：\n\n{str(data)}"
        else:
            error_msg = result.get("message", "未知错误") if result else "服务无响应"
            return f"❌ 获取聊天记录失败：{error_msg}\n提示：请确保微信桌面版已登录且窗口可以被操作"
        
    except Exception as e:
        logger.error(f"获取微信聊天记录失败: {e}", exc_info=True)
        return f"❌ 获取微信聊天记录时发生错误：{str(e)}"


@tool("send_wechat_message", args_schema=WechatSendMessageArgs, description="向单个微信好友发送单条消息")
def send_wechat_message(
    to_user: str,
    message: str
) -> str:
    """
    发送单条消息给单个微信好友
    
    Args:
        to_user: 好友或群聊的备注或昵称
        message: 要发送的消息内容
        
    Returns:
        发送结果描述
    """
    try:
        # 导入MCP服务
        from services.wechat_mcp_service import get_wechat_mcp_service
        import asyncio
        
        mcp_service = get_wechat_mcp_service()
        
        # 调用MCP服务
        loop = asyncio.new_event_loop()
        asyncio.set_event_loop(loop)
        result = loop.run_until_complete(
            mcp_service.send_message(to_user, message)
        )
        loop.close()
        
        if result and result.get("success"):
            return f"✅ 消息已成功发送给 {to_user}"
        else:
            error_msg = result.get("message", "未知错误") if result else "服务无响应"
            return f"❌ 消息发送失败：{error_msg}\n提示：请确保微信桌面版已登录且窗口可以被操作"
        
    except Exception as e:
        logger.error(f"发送微信消息失败: {e}", exc_info=True)
        return f"❌ 发送微信消息时发生错误：{str(e)}"


@tool("send_multiple_wechat_messages", args_schema=WechatSendMultipleMessagesArgs, description="向单个微信好友发送多条消息")
def send_multiple_wechat_messages(
    to_user: str,
    messages: List[str]
) -> str:
    """
    发送多条消息给单个微信好友
    
    Args:
        to_user: 好友或群聊的备注或昵称
        messages: 要发送的消息列表
        
    Returns:
        发送结果描述
    """
    try:
        # 导入MCP服务
        from services.wechat_mcp_service import get_wechat_mcp_service
        import asyncio
        
        mcp_service = get_wechat_mcp_service()
        
        # 调用MCP服务
        loop = asyncio.new_event_loop()
        asyncio.set_event_loop(loop)
        result = loop.run_until_complete(
            mcp_service.send_multiple_messages(to_user, messages)
        )
        loop.close()
        
        if result and result.get("success"):
            return f"✅ 已成功向 {to_user} 发送 {len(messages)} 条消息"
        else:
            error_msg = result.get("message", "未知错误") if result else "服务无响应"
            return f"❌ 批量消息发送失败：{error_msg}\n提示：请确保微信桌面版已登录且窗口可以被操作"
        
    except Exception as e:
        logger.error(f"批量发送微信消息失败: {e}", exc_info=True)
        return f"❌ 批量发送微信消息时发生错误：{str(e)}"


@tool("send_wechat_to_multiple_friends", args_schema=WechatSendToMultipleFriendsArgs, description="向多个微信好友发送消息")
def send_wechat_to_multiple_friends(
    to_users: List[str],
    message: str
) -> str:
    """
    向多个微信好友发送消息
    
    Args:
        to_users: 好友或群聊的备注或昵称列表
        message: 要发送的消息内容（单条消息会发给所有好友）
        
    Returns:
        发送结果描述
    """
    try:
        # 导入MCP服务
        from services.wechat_mcp_service import get_wechat_mcp_service
        import asyncio
        
        mcp_service = get_wechat_mcp_service()
        
        # 调用MCP服务
        loop = asyncio.new_event_loop()
        asyncio.set_event_loop(loop)
        result = loop.run_until_complete(
            mcp_service.send_to_multiple_friends(to_users, message)
        )
        loop.close()
        
        if result and result.get("success"):
            return f"✅ 消息已成功发送给 {len(to_users)} 位好友：{', '.join(to_users)}"
        else:
            error_msg = result.get("message", "未知错误") if result else "服务无响应"
            return f"❌ 群发消息失败：{error_msg}\n提示：请确保微信桌面版已登录且窗口可以被操作"
        
    except Exception as e:
        logger.error(f"群发微信消息失败: {e}", exc_info=True)
        return f"❌ 群发微信消息时发生错误：{str(e)}"


# 导出所有工具
__all__ = [
    'get_wechat_chat_history',
    'send_wechat_message',
    'send_multiple_wechat_messages',
    'send_wechat_to_multiple_friends'
]