TravelContentCreator/examples/test_prompt_manager.py

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

"""
测试PromptManager组件

该脚本演示如何独立使用prompt_manager模块构建不同场景的提示词。
"""

import os
import sys
import tempfile
import json
from pathlib import Path

# 将项目根目录添加到PATH
project_root = str(Path(__file__).parent.parent.absolute())
if project_root not in sys.path:
    sys.path.append(project_root)

from utils.prompt_manager import PromptManager

def create_test_files():
    """创建测试用的提示词文件和资源文件"""
    # 创建临时目录
    temp_dir = tempfile.mkdtemp()
    
    # 创建提示词目录结构
    prompts_dir = os.path.join(temp_dir, "prompts")
    os.makedirs(prompts_dir, exist_ok=True)
    
    # 创建Style和Demand子目录
    style_dir = os.path.join(prompts_dir, "Style")
    demand_dir = os.path.join(prompts_dir, "Demand")
    refer_dir = os.path.join(prompts_dir, "Refer")
    
    os.makedirs(style_dir, exist_ok=True)
    os.makedirs(demand_dir, exist_ok=True)
    os.makedirs(refer_dir, exist_ok=True)
    
    # 创建Topic提示词文件
    topic_system_path = os.path.join(temp_dir, "topic_system.txt")
    with open(topic_system_path, 'w', encoding='utf-8') as f:
        f.write("您是旅游内容选题专家。请根据提供的信息，生成高质量的旅游选题。")
    
    topic_user_path = os.path.join(temp_dir, "topic_user.txt")
    with open(topic_user_path, 'w', encoding='utf-8') as f:
        f.write("生成{num}个旅游内容选题，日期为{date}。")
    
    # 创建Content提示词文件
    content_system_path = os.path.join(temp_dir, "content_system.txt")
    with open(content_system_path, 'w', encoding='utf-8') as f:
        f.write("您是旅游内容创作专家。请根据选题和参考资料，生成优质旅游内容。")
    
    # 创建Style文件
    style_file = os.path.join(style_dir, "摄影指南.txt")
    with open(style_file, 'w', encoding='utf-8') as f:
        f.write("这是一篇摄影风格的旅游文章，应该包含构图技巧、最佳拍摄时间和角度建议。")
    
    # 创建Demand文件
    demand_file = os.path.join(demand_dir, "亲子游.txt")
    with open(demand_file, 'w', encoding='utf-8') as f:
        f.write("这是一篇面向家庭旅游者的文章，应该包含适合儿童的活动和安全提示。")
    
    # 创建Resource目录和文件
    resource_dir = os.path.join(temp_dir, "resource")
    os.makedirs(resource_dir, exist_ok=True)
    
    # 创建一个景点描述文件
    desc_file = os.path.join(resource_dir, "泰宁古城描述.txt")
    with open(desc_file, 'w', encoding='utf-8') as f:
        f.write("泰宁古城是一座有着八百年历史的古城，保存了大量明清时期的建筑。")
    
    # 返回创建的文件路径
    return {
        "temp_dir": temp_dir,
        "prompts_dir": prompts_dir,
        "topic_system_path": topic_system_path,
        "topic_user_path": topic_user_path,
        "content_system_path": content_system_path,
        "resource_dir": resource_dir,
        "desc_file": desc_file,
        "style_file": style_file,
        "demand_file": demand_file
    }

def create_topic_item():
    """创建测试用的选题项"""
    return {
        "index": 1,
        "date": "2024-05-30",
        "logic": "这是一个著名的景点，有丰富的历史文化底蕴，适合文化和历史爱好者。",
        "object": "泰宁古城",
        "product": "文化体验套票.txt",
        "product_logic": "泰宁古城有多处明清古建筑，游客可以通过体验套票更全面了解当地历史文化。",
        "style": "摄影指南.txt",
        "style_logic": "古城拥有800多年历史，古建筑保存完好，适合摄影风格。",
        "target_audience": "亲子游.txt",
        "target_audience_logic": "古城内有适合家庭参观的展览和互动活动，老人孩子都能找到感兴趣的内容。"
    }

def main():
    """主函数"""
    print("=== 测试 PromptManager 组件 ===")
    
    # 创建测试文件
    test_files = create_test_files()
    print(f"创建临时测试文件夹: {test_files['temp_dir']}")
    
    try:
        # 创建资源文件配置
        resource_dir_config = [
            {
                "type": "Description",
                "file_path": [test_files["desc_file"]]
            }
        ]
        
        # 初始化PromptManager
        prompt_manager = PromptManager(
            topic_system_prompt_path=test_files["topic_system_path"],
            topic_user_prompt_path=test_files["topic_user_path"],
            content_system_prompt_path=test_files["content_system_path"],
            prompts_dir=test_files["prompts_dir"],
            resource_dir_config=resource_dir_config,
            topic_gen_num=3,
            topic_gen_date="2024-05-30"
        )
        
        print("\n测试1: 获取选题提示词")
        system_prompt, user_prompt = prompt_manager.get_topic_prompts()
        
        print(f"系统提示词 ({len(system_prompt)} 字符):")
        print(system_prompt)
        print(f"\n用户提示词 ({len(user_prompt)} 字符):")
        print(user_prompt)
        
        # 验证变量替换
        if "{num}" not in user_prompt and "3" in user_prompt:
            print("测试通过: {num} 成功替换为 3")
        else:
            print("测试失败: {num} 未被正确替换")
            
        if "{date}" not in user_prompt and "2024-05-30" in user_prompt:
            print("测试通过: {date} 成功替换为 2024-05-30")
        else:
            print("测试失败: {date} 未被正确替换")
        
        print("\n测试2: 获取内容生成提示词")
        topic_item = create_topic_item()
        
        system_prompt, user_prompt = prompt_manager.get_content_prompts(topic_item)
        
        print(f"系统提示词 ({len(system_prompt)} 字符):")
        print(system_prompt[:200] + "..." if len(system_prompt) > 200 else system_prompt)
        print(f"\n用户提示词 ({len(user_prompt)} 字符):")
        print(user_prompt[:200] + "..." if len(user_prompt) > 200 else user_prompt)
        
        # 验证样式和目标受众内容是否包含在提示词中
        with open(test_files["style_file"], 'r', encoding='utf-8') as f:
            style_content = f.read()
        
        with open(test_files["demand_file"], 'r', encoding='utf-8') as f:
            demand_content = f.read()
        
        if style_content in user_prompt:
            print("测试通过: 样式文件内容成功包含在用户提示词中")
        else:
            print("测试失败: 样式文件内容未包含在用户提示词中")
            
        if demand_content in user_prompt:
            print("测试通过: 目标受众文件内容成功包含在用户提示词中")
        else:
            print("测试失败: 目标受众文件内容未包含在用户提示词中")
            
        # 验证景点描述是否包含
        with open(test_files["desc_file"], 'r', encoding='utf-8') as f:
            desc_content = f.read()
            
        if desc_content in user_prompt:
            print("测试通过: 景点描述文件内容成功包含在用户提示词中")
        else:
            print("测试失败: 景点描述文件内容未包含在用户提示词中")
        
        print("\n测试3: 异常处理")
        # 测试不存在的风格文件
        non_existent_topic = topic_item.copy()
        non_existent_topic["style"] = "不存在的风格.txt"
        
        system_prompt, user_prompt = prompt_manager.get_content_prompts(non_existent_topic)
        print("获取提示词成功，即使风格文件不存在")
        
    except Exception as e:
        print(f"测试过程中出错: {e}")
        import traceback
        traceback.print_exc()
    
    finally:
        # 清理临时文件
        import shutil
        try:
            shutil.rmtree(test_files["temp_dir"])
            print(f"\n清理临时文件: {test_files['temp_dir']}")
        except Exception as e:
            print(f"清理临时文件失败: {e}")
    
    print("\n=== 测试完成 ===")

if __name__ == "__main__":
    main()