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

"""
测试AI审查引擎的RAG增强检查功能
"""

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

from core.construction_review.component.ai_review_engine import AIReviewEngine
from foundation.observability.logger.loggering import server_logger as logger

# 实例化AIReviewEngine
review_engine = AIReviewEngine()

# 记录开始时间
import time
start_time = time.time()
query = """ # 5. 1 防护栏杆5.1 防护栏杆\n\n## 部分 5\n\n5.1.7 防护栏杆使用过程中应避免在构件上额外施加长期的外力作用或施加振动荷载 不应悬挂重5.1.7 防护栏杆使用过程中应避免在构件上额外施加长期的外力作用或施加振动荷载，不应悬挂重物。 发现锈蚀、腐蚀、松动或损坏的,应及时进行维修、更换。物。发现锈蚀、腐蚀、松动或损坏的，应及时进行维修、更换。"""
unit_content= {
    "content" : query,
}
result = review_engine.rag_enhanced_check(unit_content)
print(result)

end_time = time.time()
elapsed_time = end_time - start_time
print(f"\nRAG增强检查完成，耗时: {elapsed_time:.2f}秒")