NarratoAI/app/services/prompts/manager.py

#!/usr/bin/env python
# -*- coding: UTF-8 -*-

"""
@Project: NarratoAI
@File   : manager.py
@Author : AI Assistant
@Date   : 2025/1/7
@Description: 提示词管理器
"""

from typing import Dict, Any, List, Optional, Union
from loguru import logger

from .base import BasePrompt, ModelType, OutputFormat
from .registry import get_registry
from .template import get_renderer
from .validators import PromptOutputValidator
from .exceptions import (
    PromptNotFoundError,
    PromptValidationError,
    TemplateRenderError
)


class PromptManager:
    """提示词管理器 - 统一的提示词管理接口"""
    
    def __init__(self):
        self._registry = get_registry()
        self._renderer = get_renderer()
        
    @classmethod
    def get_prompt(cls, 
                   category: str, 
                   name: str, 
                   version: Optional[str] = None,
                   parameters: Optional[Dict[str, Any]] = None) -> str:
        """
        获取渲染后的提示词
        
        Args:
            category: 分类
            name: 名称
            version: 版本（可选，默认使用最新版本）
            parameters: 模板参数（可选）
            
        Returns:
            渲染后的提示词字符串
        """
        instance = cls()
        prompt_obj = instance._registry.get(category, name, version)
        
        try:
            rendered = prompt_obj.render(parameters)
            logger.debug(f"提示词渲染成功: {category}.{name} v{prompt_obj.version}")
            return rendered
        except Exception as e:
            logger.error(f"提示词渲染失败: {category}.{name} - {str(e)}")
            raise
            
    @classmethod
    def get_prompt_object(cls, 
                         category: str, 
                         name: str, 
                         version: Optional[str] = None) -> BasePrompt:
        """
        获取提示词对象
        
        Args:
            category: 分类
            name: 名称
            version: 版本（可选）
            
        Returns:
            提示词对象
        """
        instance = cls()
        return instance._registry.get(category, name, version)
        
    @classmethod
    def register_prompt(cls, prompt: BasePrompt, is_default: bool = True) -> None:
        """
        注册提示词
        
        Args:
            prompt: 提示词对象
            is_default: 是否设为默认版本
        """
        instance = cls()
        instance._registry.register(prompt, is_default)
        
    @classmethod
    def list_categories(cls) -> List[str]:
        """列出所有分类"""
        instance = cls()
        return instance._registry.list_categories()
        
    @classmethod
    def list_prompts(cls, category: str) -> List[str]:
        """列出指定分类下的所有提示词"""
        instance = cls()
        return instance._registry.list_prompts(category)
        
    @classmethod
    def list_versions(cls, category: str, name: str) -> List[str]:
        """列出指定提示词的所有版本"""
        instance = cls()
        return instance._registry.list_versions(category, name)
        
    @classmethod
    def exists(cls, category: str, name: str, version: Optional[str] = None) -> bool:
        """检查提示词是否存在"""
        instance = cls()
        return instance._registry.exists(category, name, version)
        
    @classmethod
    def search_prompts(cls,
                      keyword: str = None,
                      category: str = None,
                      model_type: ModelType = None,
                      output_format: OutputFormat = None) -> List[Dict[str, str]]:
        """
        搜索提示词
        
        Args:
            keyword: 关键词
            category: 分类过滤
            model_type: 模型类型过滤
            output_format: 输出格式过滤
            
        Returns:
            匹配的提示词列表
        """
        instance = cls()
        results = instance._registry.search(keyword, category, model_type, output_format)
        
        return [
            {
                "category": cat,
                "name": name,
                "version": ver,
                "full_name": f"{cat}.{name}",
                "identifier": f"{cat}.{name}@{ver}"
            }
            for cat, name, ver in results
        ]
        
    @classmethod
    def get_stats(cls) -> Dict[str, Any]:
        """获取统计信息"""
        instance = cls()
        registry_stats = instance._registry.get_stats()
        
        return {
            "registry": registry_stats,
            "categories": cls.list_categories(),
            "total_categories": registry_stats["categories"],
            "total_prompts": registry_stats["prompts"],
            "total_versions": registry_stats["versions"]
        }
        
    @classmethod
    def validate_output(cls,
                       output: Union[str, Dict],
                       category: str,
                       name: str,
                       version: Optional[str] = None) -> Any:
        """
        验证提示词输出
        
        Args:
            output: 输出内容
            category: 提示词分类
            name: 提示词名称
            version: 提示词版本
            
        Returns:
            验证后的数据
        """
        instance = cls()
        prompt_obj = instance._registry.get(category, name, version)
        
        # 根据输出格式进行验证
        output_format = prompt_obj.metadata.output_format
        
        try:
            if output_format == OutputFormat.JSON:
                # 特殊处理解说文案和剧情分析
                if "narration" in name.lower() or "script" in name.lower():
                    return PromptOutputValidator.validate_narration_script(output)
                elif "plot" in name.lower() or "analysis" in name.lower():
                    return PromptOutputValidator.validate_plot_analysis(output)
                else:
                    return PromptOutputValidator.validate_json(output)
            else:
                return PromptOutputValidator.validate_by_format(output, output_format)
                
        except Exception as e:
            logger.error(f"输出验证失败 {category}.{name}: {str(e)}")
            raise PromptValidationError(f"输出验证失败: {str(e)}")
            
    @classmethod
    def get_prompt_info(cls, category: str, name: str, version: Optional[str] = None) -> Dict[str, Any]:
        """
        获取提示词详细信息
        
        Args:
            category: 分类
            name: 名称
            version: 版本
            
        Returns:
            提示词详细信息
        """
        instance = cls()
        prompt_obj = instance._registry.get(category, name, version)
        
        return {
            "metadata": {
                "name": prompt_obj.metadata.name,
                "category": prompt_obj.metadata.category,
                "version": prompt_obj.metadata.version,
                "description": prompt_obj.metadata.description,
                "model_type": prompt_obj.metadata.model_type.value,
                "output_format": prompt_obj.metadata.output_format.value,
                "author": prompt_obj.metadata.author,
                "created_at": prompt_obj.metadata.created_at.isoformat(),
                "updated_at": prompt_obj.metadata.updated_at.isoformat(),
                "tags": prompt_obj.metadata.tags,
                "parameters": prompt_obj.metadata.parameters
            },
            "template_preview": prompt_obj.get_template()[:500] + "..." if len(prompt_obj.get_template()) > 500 else prompt_obj.get_template(),
            "system_prompt": prompt_obj.get_system_prompt(),
            "examples_count": len(prompt_obj.get_examples()),
            "has_parameters": bool(prompt_obj.metadata.parameters)
        }
        
    @classmethod
    def export_prompts(cls, category: Optional[str] = None) -> Dict[str, Any]:
        """
        导出提示词配置
        
        Args:
            category: 分类过滤（可选）
            
        Returns:
            提示词配置数据
        """
        instance = cls()
        categories = [category] if category else instance._registry.list_categories()
        
        export_data = {
            "version": "1.0.0",
            "exported_at": instance._get_current_time(),
            "categories": {}
        }
        
        for cat in categories:
            export_data["categories"][cat] = {}
            prompts = instance._registry.list_prompts(cat)
            
            for prompt_name in prompts:
                versions = instance._registry.list_versions(cat, prompt_name)
                export_data["categories"][cat][prompt_name] = {}
                
                for ver in versions:
                    prompt_obj = instance._registry.get(cat, prompt_name, ver)
                    export_data["categories"][cat][prompt_name][ver] = prompt_obj.to_dict()
                    
        return export_data
        
    def _get_current_time(self) -> str:
        """获取当前时间字符串"""
        from datetime import datetime
        return datetime.now().isoformat()


# 便捷函数
def get_prompt(category: str, name: str, version: str = None, **parameters) -> str:
    """获取提示词的便捷函数"""
    return PromptManager.get_prompt(category, name, version, parameters)


def validate_prompt_output(output: Union[str, Dict], category: str, name: str, version: str = None) -> Any:
    """验证提示词输出的便捷函数"""
    return PromptManager.validate_output(output, category, name, version)