时序知识图谱RAG — 双层时序图增量更新与时间窗口检索

"""
时序知识图谱RAG系统 (TG-RAG)
功能：双层时序图构建 + 增量更新 + 时间窗口检索 + 冲突解决
基于 arXiv:2510.13590 (2025) + LedgerRAG (2026)
"""
import json
from dataclasses import dataclass, field
from typing import List, Dict, Optional, Tuple, Any
from datetime import datetime, timedelta
from collections import defaultdict
import warnings
warnings.filterwarnings('ignore')


@dataclass
class TemporalFact:
    """时序事实三元组"""
    fact_id: str
    subject: str
    relation: str
    value: Any
    valid_from: datetime
    valid_to: Optional[datetime] = None     # None = 持续有效
    source: str = ""
    confidence: float = 1.0

    @property
    def is_current(self) -> bool:
        """判断事实是否当前有效"""
        now = datetime.now()
        if self.valid_to and self.valid_to < now:
            return False
        return self.valid_from <= now

    @property
    def age_days(self) -> float:
        """事实年龄（天）"""
        return (datetime.now() - self.valid_from).days

    def overlaps_window(self, start: datetime, end: datetime) -> bool:
        """检查是否与时间窗口重叠"""
        fact_end = self.valid_to or datetime.max
        return self.valid_from <= end and fact_end >= start


@dataclass
class TimeNode:
    """时间图节点"""
    time_key: str               # 如 '2025-Q4', '2025-11', '2025-W47'
    granularity: str            # 'year', 'quarter', 'month', 'week', 'day'
    start: datetime
    end: datetime
    summary: str = ""           # 该时间段的知识摘要
    fact_ids: List[str] = field(default_factory=list)
    children_keys: List[str] = field(default_factory=list)
    parent_key: Optional[str] = None


class TemporalKnowledgeGraph:
    """