# tests/unit/test_trace_model.py
import sys
sys.path.insert(0, '/home/jang/Projects/ai-researcher')

import pytest

from src.models.trace import Trace


def test_trace_creation():
    """测试Trace创建"""
    trace = Trace(
        id="trace_123",
        task_id="task_456",
        query_analysis={"intent": "query_employee", "confidence": 0.95},
        data_sources=[{"skill": "data_query", "endpoint": "hr.api.com"}],
        reasoning_steps=[{"step": 1, "type": "extraction"}],
        confidence={"overall": 0.97},
    )

    assert trace.id == "trace_123"
    assert trace.task_id == "task_456"
    assert trace.query_analysis["intent"] == "query_employee"
    assert trace.data_sources == [{"skill": "data_query", "endpoint": "hr.api.com"}]
    assert trace.reasoning_steps == [{"step": 1, "type": "extraction"}]
    assert trace.confidence["overall"] == 0.97