ai-rasa-manager/services/test_rasa_ge...

import logging
import os
from rasa_generator_by_json import RasaFileGenerator

# 配置日志
logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
logger = logging.getLogger(__name__)

if __name__ == "__main__":
    try:
        # 创建生成器实例
        generator = RasaFileGenerator()
        
        # 定义flow.json文件路径
        flow_json_path = '/home/rasa_manager/services/flow.json'
        
        # 检查文件是否存在
        if not os.path.exists(flow_json_path):
            raise Exception(f"flow.json文件不存在: {flow_json_path}")
        
        # 读取flow.json文件内容
        with open(flow_json_path, 'r', encoding='utf-8') as f:
            flow_json_str = f.read()
        
        logger.info(f"开始从JSON字符串生成Rasa文件...")
        
        # 生成所有文件
        result = generator.generate_all_files(flow_json_str, "酒店预定")
        
        if result['success']:
            logger.info(f"文件生成成功！输出目录: {result['output_dir']}")
            logger.info(f"生成的文件:")
            for file_type, file_path in result['files'].items():
                logger.info(f"  - {file_type}: {file_path}")
        else:
            logger.error(f"文件生成失败: {result['message']}")
            
    except Exception as e:
        logger.error(f"测试过程中发生错误: {str(e)}")