Function test_batch_export

tests/test_trajectory.py:147–179

测试批量导出

()

    log.info(f"✓ 已导出 SFT 格式: {filepath}")
    
def test_batch_export():
    """测试批量导出"""
    log.info("=" * 60)
    log.info("测试 3: 批量导出")
    log.info("=" * 60)
    
    # 创建多个轨迹
    trajectories = []
    
    for i in range(3):
        manager = TrajectoryManager()
        manager.start_recording(inputs={"query": f"任务 {i}"})
        
        collector = manager.get_collector()
        collector.on_node_start(f"node_{i}", StepRole.SYSTEM_NODE.value)
        collector.on_node_end(output={"result": f"output_{i}"})
        
        request = DFRequest(target=f"任务 {i}")
        state = DFState(request=request)
        
        trajectory = manager.stop_recording(state, f"workflow_{i}")
        trajectories.append(trajectory)
    
    # 批量导出
    exporter = TrajectoryExporter()
    
    # 导出为 JSONL
    filepath = exporter.export_to_jsonl(trajectories, mode="raw")
    log.info(f"✓ 批量导出 JSONL: {filepath}")
    
    # 导出统计信息
    stats_path = exporter.export_statistics(trajectories)
    log.info(f"✓ 导出统计信息: {stats_path}")
    
def test_feedback():
    """测试用户反馈"""