NeoBot/plugins/discord-cross/translator.py

# -*- coding: utf-8 -*-
"""
跨平台消息互通插件翻译模块
"""
from typing import Dict, List
from core.utils.logger import logger
from .config import config

# 翻译上下文缓存（每个通道15条消息）
TRANSLATION_CONTEXT_CACHE: Dict[str, List[Dict[str, str]]] = {}
MAX_CONTEXT_MESSAGES = 15

def get_translation_context(channel_id: int, direction: str) -> List[Dict[str, str]]:
    """获取翻译上下文缓存"""
    cache_key = f"{channel_id}_{direction}"
    return TRANSLATION_CONTEXT_CACHE.get(cache_key, [])

def add_translation_context(channel_id: int, direction: str, original: str, translated: str):
    """添加翻译到上下文缓存"""
    cache_key = f"{channel_id}_{direction}"
    if cache_key not in TRANSLATION_CONTEXT_CACHE:
        TRANSLATION_CONTEXT_CACHE[cache_key] = []
    
    TRANSLATION_CONTEXT_CACHE[cache_key].append({
        "original": original,
        "translated": translated
    })
    
    if len(TRANSLATION_CONTEXT_CACHE[cache_key]) > MAX_CONTEXT_MESSAGES:
        TRANSLATION_CONTEXT_CACHE[cache_key] = TRANSLATION_CONTEXT_CACHE[cache_key][-MAX_CONTEXT_MESSAGES:]

async def translate_with_deepseek(
    text: str, 
    target_lang: str = "zh-CN",
    channel_id: int = 0,
    direction: str = "en2zh"
) -> str:
    """使用 DeepSeek API 翻译文本"""
    if not config.ENABLE_TRANSLATION or not text.strip():
        return text
        
    if config.DEEPSEEK_API_KEY == "your-deepseek-api-key-here":
        logger.warning("[CrossPlatform] DeepSeek API 密钥未配置，跳过翻译")
        return text
    
    lang_name = "中文" if target_lang == "zh-CN" else "英文"
    
    messages = []
    context_ref = ""
    if channel_id > 0:
        context = get_translation_context(channel_id, direction)
        if context:
            context_ref = "\n\n参考之前的翻译：\n"
            for i, ctx in enumerate(context[-5:], 1):
                context_ref += f"{i}. 原文: {ctx['original'][:100]}\n   译文: {ctx['translated'][:100]}\n"
    
    system_prompt = f"""你是一个专业的翻译助手。请将以下文本翻译成{lang_name}。
只返回翻译后的文本，不要添加任何解释、注释或其他内容。避免翻译出仇视言论以及违反中国大陆相关法律法规的内容。如果有，请在翻译后有敏感的词语中把文本替换成井号（#）
保持原文的语气和格式。如果文本已经是目标语言，直接返回原文。{context_ref}"""
    
    messages.append({"role": "user", "content": text})
    
    try:
        from openai import AsyncOpenAI
        
        client = AsyncOpenAI(
            api_key=config.DEEPSEEK_API_KEY,
            base_url=config.DEEPSEEK_API_URL.replace("/chat/completions", "")
        )
        
        response = await client.chat.completions.create(
            model=config.DEEPSEEK_MODEL,
            messages=[{"role": "system", "content": system_prompt}] + messages,
            temperature=0.3,
            max_tokens=4000
        )
        
        translated_text = response.choices[0].message.content
        if translated_text:
            translated_text = translated_text.strip()
            logger.info(f"[CrossPlatform] 翻译成功: {text[:50]}... -> {translated_text[:50]}...")
            
            if channel_id > 0:
                add_translation_context(channel_id, direction, text, translated_text)
            
            return translated_text
        else:
            logger.warning("[CrossPlatform] DeepSeek 返回空翻译结果")
            return text
            
    except ImportError:
        logger.warning("[CrossPlatform] openai 库未安装，尝试使用同步请求")
        return await translate_with_deepseek_sync(text, target_lang, channel_id, direction)
    except Exception as e:
        logger.error(f"[CrossPlatform] 翻译失败: {e}")
        return text

async def translate_with_deepseek_sync(
    text: str, 
    target_lang: str = "zh-CN",
    channel_id: int = 0,
    direction: str = "en2zh"
) -> str:
    """使用同步请求的 DeepSeek 翻译（备用方案）"""
    if not config.ENABLE_TRANSLATION or not text.strip():
        return text
        
    if config.DEEPSEEK_API_KEY == "your-deepseek-api-key-here":
        return text
    
    lang_name = "中文" if target_lang == "zh-CN" else "英文"
    
    context_ref = ""
    if channel_id > 0:
        context = get_translation_context(channel_id, direction)
        if context:
            context_ref = "\n\n参考之前的翻译：\n"
            for i, ctx in enumerate(context[-5:], 1):
                context_ref += f"{i}. 原文: {ctx['original'][:100]}\n   译文: {ctx['translated'][:100]}\n"
    
    system_prompt = f"""你是一个专业的翻译助手。请将以下文本翻译成{lang_name}。
只返回翻译后的文本，不要添加任何解释、注释或其他内容。避免翻译出仇视言论以及违反中国大陆相关法律法规的内容。如果有，请在翻译后有敏感的词语中把文本替换成井号（#）
保持原文的语气和格式。如果文本已经是目标语言，直接返回原文。{context_ref}"""
    
    messages = [{"role": "user", "content": text}]
    
    try:
        from openai import OpenAI
        
        client = OpenAI(
            api_key=config.DEEPSEEK_API_KEY,
            base_url=config.DEEPSEEK_API_URL.replace("/chat/completions", "")
        )
        
        response = client.chat.completions.create(
            model=config.DEEPSEEK_MODEL,
            messages=[{"role": "system", "content": system_prompt}] + messages,
            temperature=0.3,
            max_tokens=4000
        )
        
        translated_text = response.choices[0].message.content
        if translated_text:
            translated_text = translated_text.strip()
            if channel_id > 0:
                add_translation_context(channel_id, direction, text, translated_text)
            return translated_text
        return text
            
    except Exception as e:
        logger.error(f"[CrossPlatform] 同步翻译失败: {e}")
        return text