65 lines
2.2 KiB
Python
65 lines
2.2 KiB
Python
"""Trust memory: verified vs unverified, provenance, confidence decay for AGI."""
|
|
|
|
from datetime import datetime, timezone
|
|
from typing import Any
|
|
|
|
from fusionagi._logger import logger
|
|
|
|
|
|
def _utc_now() -> datetime:
|
|
return datetime.now(timezone.utc)
|
|
|
|
|
|
class TrustMemory:
|
|
"""
|
|
Tracks verified vs unverified claims, source provenance, and optional
|
|
confidence decay over time.
|
|
"""
|
|
|
|
def __init__(self, decay_enabled: bool = False) -> None:
|
|
self._entries: dict[str, dict[str, Any]] = {} # claim_id -> {verified, source, confidence, created_at}
|
|
self._decay_enabled = decay_enabled
|
|
|
|
def add(
|
|
self,
|
|
claim_id: str,
|
|
verified: bool,
|
|
source: str = "",
|
|
confidence: float = 1.0,
|
|
metadata: dict[str, Any] | None = None,
|
|
) -> None:
|
|
"""Record a claim with verification status and provenance."""
|
|
self._entries[claim_id] = {
|
|
"verified": verified,
|
|
"source": source,
|
|
"confidence": confidence,
|
|
"created_at": _utc_now(),
|
|
"metadata": metadata or {},
|
|
}
|
|
logger.debug("Trust memory: claim added", extra={"claim_id": claim_id, "verified": verified})
|
|
|
|
def get(self, claim_id: str) -> dict[str, Any] | None:
|
|
"""Return trust entry or None. Applies decay if enabled."""
|
|
e = self._entries.get(claim_id)
|
|
if not e:
|
|
return None
|
|
if self._decay_enabled:
|
|
# Simple decay: reduce confidence by 0.01 per day (placeholder)
|
|
from datetime import timedelta
|
|
age_days = (_utc_now() - e["created_at"]).total_seconds() / 86400
|
|
e = dict(e)
|
|
e["confidence"] = max(0.0, e["confidence"] - 0.01 * age_days)
|
|
return e
|
|
|
|
def is_verified(self, claim_id: str) -> bool:
|
|
"""Return True if claim is marked verified."""
|
|
e = self._entries.get(claim_id)
|
|
return e.get("verified", False) if e else False
|
|
|
|
def set_verified(self, claim_id: str, verified: bool) -> bool:
|
|
"""Update verified status. Returns True if claim existed."""
|
|
if claim_id not in self._entries:
|
|
return False
|
|
self._entries[claim_id]["verified"] = verified
|
|
return True
|