Function test_feedback

tests/test_trajectory.py:181–216

测试用户反馈

()

    log.info(f"✓ 导出统计信息: {stats_path}")
    
def test_feedback():
    """测试用户反馈"""
    log.info("=" * 60)
    log.info("测试 4: 用户反馈")
    log.info("=" * 60)
    
    manager = TrajectoryManager()
    manager.start_recording(inputs={"query": "测试反馈"})
    
    collector = manager.get_collector()
    collector.on_node_start("test_node", StepRole.AGENT.value)
    collector.on_node_end(output={"result": "success"})
    
    request = DFRequest(target="测试反馈")
    state = DFState(request=request)
    
    trajectory = manager.stop_recording(state, "feedback_test")
    
    # 添加反馈
    manager.add_feedback(
        trajectory=trajectory,
        score=5,
        comment="非常好！",
        labels=["accurate", "helpful"]
    )
    
    assert trajectory.feedback is not None
    assert trajectory.feedback.score == 5
    assert "accurate" in trajectory.feedback.labels
    
    log.info(f"✓ 反馈评分: {trajectory.feedback.score}")
    log.info(f"✓ 反馈标签: {trajectory.feedback.labels}")
    
    # 导出包含反馈的轨迹
    filepath = manager.export(trajectory, format="json")
    log.info(f"✓ 已导出包含反馈的轨迹: {filepath}")
    
def main():
    """运行所有测试"""