Update memory_drift_diagnostician.py

memory_drift_diagnostician.py

@@ -1,7 +1,7 @@
 import logging
 import numpy as np
 from typing import Dict, Any, List
-from agentic_reliability_framework.runtime.agents.base import BaseAgent
+from agentic_reliability_framework.runtime.agents.base import BaseAgent, AgentSpecialization
 from ai_event import AIEvent
 
 logger = logging.getLogger(__name__)