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

"""
测试AI审查引擎的rag_enhanced_check方法
用于快速验证模块功能是否正常实现
"""

import sys
import os
import time

sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from core.base.task_models import TaskFileInfo
from core.construction_review.component.ai_review_engine import AIReviewEngine
from foundation.observability.logger.loggering import review_logger as logger



def test_rag_enhanced_check():
    """
    测试rag_enhanced_check方法的基本功能
    """
    print("=" * 60)
    print("测试rag_enhanced_check方法")
    print("=" * 60)

    # 准备测试数据
    query_content = """相关准备工作
1、技术准备
四川公路桥梁建设集团有限公司 镇广C4项目经理部JQ220t-40m架桥机安装及拆除专项施工方案
（1）应将架桥机随机文件和有关技术资料准备齐全，认真组织安装人员学习阅
读本方案，并以此为依据结合实际工况及地貌拟定有关安装施工方案。
（2）安装前，应对设备散件进行全面检查、清理，如发现有损伤、腐蚀或其它
缺陷，应在安装前予以处理，合格后方可安装。
（3）应对架桥机运梁轨道进行如下检查：
1）架桥机运梁轨道基础应有足够的承压能力，应能满足架桥机运梁平车载重安
全运行。轨道下部如有支垫是否具有强大的刚性和一定的密度，特殊端部的支垫必
须采用刚性支垫，严禁使用腐蚀的枕木。
2）轨道钢轨正面、侧面的不平度不应大于1/1500，全长范围内不应大于10毫
米。
3）轨道安装的允许偏差：
①轨道实际中心线对轨道设计中心线的位置偏移允许偏差为3毫米。
②轨距允许偏差为±5毫米。
③轨道纵向不平度应小于 1/1500，且全行程不超过10毫米。
④同一断面上两轨道的标高相对偏差不超过5毫米。
4）轨道接头应符合下列要求：
①接头左、右、上三面错位不应大于2毫米。
②两平行轨道接口的位置应错开，其错开距离不应等于架桥机运梁平车前后车
轮的轮距。
第19页
四川公路桥梁建设集团有限公司 镇广C4项目经理部JQ220t-40m架桥机安装及拆除专项施工方案
③接头间隙应为1~2毫米，伸缩缝接头间隙应符合设计要求，其偏差不应大于
±1毫米。
④轨道的悬空部位是否得到了加强。
5）施工期间的基本要求及安全规定
①架桥机安装属于高空作业，施工前技术员及现场指挥员向参加施工的所有人
员详细介绍安装工序、技术要求和指挥信号。
②严禁在施工的架桥机下面逗留通过，与安装施工无关人员不准擅自进入施工
现场。
③现场施工人员必须按有关安全规定佩戴好安全用品（安全带、安全帽、绝缘
鞋等）。
④施工现场使用的氧气瓶、乙炔瓶必须保证立放并固定好，两瓶之间距离不得
小于5m。
6）架桥机组装
①安装原则：按从下至上的顺序进行安装，且必须在下部结构安装固定牢固以
后，才能进行上部结构的安装。同时，在安装过程中，在运输及吊装能力允许的情
况下，尽量保持设备各部件的整体性，一方面能有效减少危险作业点，另一方面有
利于提高设备现场安装工效。
②安装场地：架桥拼装在桥台台背后路基上，拼装场地选址位于路基上，拼装
场地面积宽15m，长 100m，架桥机主体最大宽度为9m（移动轨道 18m），路基宽度
满足安装场地所需，路基及便道顶面承载力满足接卸拼装所需的承载要求。
③辅助机械设备：根据施工现场实际情况，选择1台25T汽车吊作为主要安装
设备，架桥机主梁拼装采用分节进行吊装，整机重量最大的组件为三节主梁同时起
吊，总重18t，采用2 台50t汽车吊满足吊装要求。"""
    unit_content = {
        "content": query_content
    }

    # 创建TaskFileInfo实例
    file_info = {
        'file_id': 'test_file_001',
        'user_id': 'test_user',
        'callback_task_id': 'test_task_001',
        'file_name': 'test.pdf',
        'file_type': 'pdf'
    }
    task_file_info = TaskFileInfo(file_info)

    # 创建AIReviewEngine实例
    review_engine = AIReviewEngine(task_file_info)

    start_time = time.time()
    result = review_engine.rag_enhanced_check(unit_content)
    logger.info(f"rag_enhanced_check_result {result}")
    end_time = time.time()

    # 输出结果
    print("\n[输出结果]")
    print(f"  file_name: {result.get('file_name')}")
    print(f"  text_content: {result.get('text_content')}")
    print(f"  metadata: {result.get('metadata')}")
    print(f"  retrieval_status: {result.get('retrieval_status', 'N/A')}")

    print(f"\n[执行时间] {end_time - start_time:.4f}秒")

    # 验证断言
    if result.get('retrieval_status') == 'no_results':
        print("\n[测试结果] 通过 - RAG检索无结果")
    else:
        assert 'file_name' in result, "应包含file_name字段"
        assert 'metadata' in result, "应包含metadata字段"
        print("\n[测试结果] 通过 - rag_enhanced_check方法功能正常")

    return result


if __name__ == "__main__":
    test_rag_enhanced_check()
