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

"""
测试多阶段召回功能
"""

import sys
import os
import time
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from foundation.ai.rag.retrieval.retrieval import retrieval_manager
from foundation.observability.logger.loggering import server_logger as logger


def test_multi_stage_recall(collection_name,query):
    """
    测试多阶段召回
    """

    try:
        start_time = time.time()
        results = retrieval_manager.multi_stage_recall(
            collection_name=collection_name,
            query_text=query,
            hybrid_top_k=10,
            top_k=1,
        )
        logger.info(f"返回结果:{results}")
        end_time = time.time()
        elapsed_time = end_time - start_time

        print(f"[OK] 召回完成，耗时: {elapsed_time:.2f}秒")
        print(f"[OK] 返回结果数量: {len(results)}")


    except Exception as e:
        print(f"[ERROR] 多阶段召回测试失败: {str(e)}")




def main():
    """
    主测试函数
    """
    print("开始多阶段召回功能测试")
    print("=" * 60)
    query = """按照《公路钢结构桥梁制造与安装施工规范》（JTG/T3651-2022）的要求，对圆\n柱头焊钉进行了焊接工艺评定试验，试验共计1 组，焊接9 个剪力钉，其中3 个用来\n做弯曲试验，3 个用来做拉伸试验，3 个备用。施焊参数见表4.4.3。\n表4.4.3 焊钉施焊参数\n编号\n焊接方法\n剪力钉直径\n（mm）\n伸出长度\n（mm）\n提升高度\n（mm）\n焊接时间\n（s）\n焊接电流\n（A）\nYJ-J1\n电弧螺柱焊\n22\n6\n3\n1.0\n1900~2200\n焊缝挤出焊脚均匀饱满，外观质量合格。\n4.4.3 圆柱头焊钉焊接工艺评定试验结果\n按规定对焊钉取3 根进行锤击30º弯曲试验，另外取3 根进行拉伸试验，试验结\n果见表4.4.4，接头力学性能满足技术要求。\n鹿寨至钦州港公路郁江特大桥钢箱梁\n焊接工艺评定总结报告\n42\n表4.4.4 焊钉锤击30º弯曲试验和拉伸试验结果\n规格\n锤击30º弯曲试验\n拉伸试验\nYJ-J1\nΦ22×140+\nQ355D-δ18\n编号\n焊缝状况\n编号\n拉力（KN）\n断口\n部位\n评定标准\n1\n完好\n1\n202.05\n母材\n拉力载荷不\n小于\n159.6KN，断\n口不在焊缝\n或热影响区\n2\n完好\n2\n201.50\n母材\n3\n完好\n3\n201.95\n母材\n鹿寨至钦州港公路郁江特大桥钢箱梁\n焊接工艺评定总结报告\n43\n五、焊接工艺评定试验结果分析\n5.1 焊缝强度\n1.1 验收标准：焊缝屈服强度和抗拉强度不低于母材标准值。\n材质\n板厚范围(mm)\nReL(MPa)\nRm(MPa)\nA(%)\nQ355D\nt≤16\n≥355\n≥470\n≥22\n16＜t≤40\n≥345\n≥470\n≥22\n1.2 根据表4.1.6 中试验结果整理出各类接头强度散点图如下，所有焊缝的屈服\n强度、抗拉强度均超过了母材标准值。\n5.2 焊缝金属延伸率\n2.1 验收标准：不低于母材标准值。\n2.2 结果分析：根据表4.1.6 中试验结果整理出散点图如下，所有焊缝的延伸率\n均在母材标准要求值之上。\n鹿寨至钦州港公路郁江特大桥钢箱梁\n焊接工艺评定总结报告\n44\n5.3 接头韧性\n3.1 验收标准：根据设计要求，对接接头及熔透角接接头焊缝中心及热影响区焊\n缝V 型缺口低温冲击试验结果的平均值不低于下表的规定值，且任一试验结果不得\n低于0.70 倍的标准值。\n钢材牌号\n试验温度（℃)\n冲击韧性（J）\nQ355D\n-20\n34\n注：板厚≤20mm 的薄板接头冲击功规定值为27J\n3.2 结果分析：从表4.1.6 中试验数据可以看出：所有试验结果中冲击功最小平均\n值为YJ-D2 试件的焊缝中心冲击功47J，大于验收值；单一试件的冲击功最小值为\nYJ-D2（板厚t=18mm）试件的焊缝中心冲击功31J，大于验收值。\n5.4 接头弯曲\n4.1 验收标准：对接接头弯曲180°，试样受拉面上的裂纹总长不大于试样宽度\n的15%，且单个裂纹长度不大于3mm，则判为合格。\n4.2 结果分析：所有对接接头的弯曲试验均完好未产生裂纹。\n5.5 接头硬度\n5.1\n验收标准：不大于HV380。\n5.2\n结果分析：根据表4.1.6 中试验结果整理出接头各区最高硬度曲线图如下，\n所有接头各区的硬度低于HV380。\n鹿寨至钦州港公路郁江特大桥钢箱梁\n焊接工艺评定总结报告\n45\n5.6 接头宏观断面腐蚀\n焊缝均熔合良好，未见焊接缺陷。\n鹿寨至钦州港公路郁江特大桥钢箱梁\n焊接工艺评定总结报告\n46\n六、结论\n从以上试验的焊缝检验和试验结果可以看出：焊缝内、外部质量良好，通过宏观"""
    collection_name = "first_bfp_collection"
    # 4. 测试多阶段召回
    test_multi_stage_recall(collection_name,query=query)





if __name__ == "__main__":
    main()