"""
测试AI日志记录功能
"""
import sys
import os
from pathlib import Path

# 添加项目根目录到路径
project_root = Path(__file__).parent.parent
sys.path.insert(0, str(project_root))

from services.ai_logger import get_ai_logger

def test_logger():
    """测试日志记录器"""
    print("=" * 60)
    print("测试AI日志记录功能")
    print("=" * 60)
    
    # 获取日志记录器
    logger = get_ai_logger()
    
    # 测试记录一次对话
    print("\n1. 测试记录对话...")
    prompt = "请从以下文本中提取信息：张三，男，1980年5月出生，某公司总经理"
    api_request = {
        "endpoint": "http://test.example.com/v1/chat/completions",
        "model": "test-model",
        "messages": [
            {"role": "system", "content": "你是一个数据提取助手"},
            {"role": "user", "content": prompt}
        ],
        "temperature": 0.2,
        "max_tokens": 1000,
        "enable_thinking": False
    }
    api_response = {
        "choices": [
            {
                "message": {
                    "content": '{"target_name": "张三", "target_gender": "男"}'
                }
            }
        ]
    }
    extracted_data = {
        "target_name": "张三",
        "target_gender": "男",
        "target_date_of_birth": "1980年05月"
    }
    
    log_file = logger.log_conversation(
        prompt=prompt,
        api_request=api_request,
        api_response=api_response,
        extracted_data=extracted_data,
        error=None,
        session_id="test_session_001"
    )
    
    if log_file:
        print(f"✓ 日志已保存: {log_file}")
    else:
        print("✗ 日志保存失败")
        return
    
    # 测试读取日志
    print("\n2. 测试读取日志...")
    log_data = logger.read_log(log_file)
    if log_data:
        print(f"✓ 日志读取成功")
        print(f"  时间戳: {log_data['timestamp']}")
        print(f"  会话ID: {log_data['session_id']}")
        print(f"  成功: {log_data['success']}")
        print(f"  提取的字段数: {len(log_data.get('extracted_data', {}))}")
    else:
        print("✗ 日志读取失败")
        return
    
    # 测试获取最近的日志
    print("\n3. 测试获取最近的日志...")
    recent_logs = logger.get_recent_logs(limit=5)
    print(f"✓ 找到 {len(recent_logs)} 条最近的日志")
    for i, log_file in enumerate(recent_logs, 1):
        print(f"  {i}. {Path(log_file).name}")
    
    print("\n" + "=" * 60)
    print("测试完成！")
    print("=" * 60)
    print(f"\n日志目录: {logger.log_dir}")
    print(f"日志状态: {'启用' if logger.enabled else '禁用'}")

if __name__ == "__main__":
    test_logger()