first commit

__spy.py

@@ -0,0 +1,50 @@
+#!/usr/bin/env python3
+"""
+__spy.py
+Lightweight global announcer for the currently loaded model.
+
+Usage:
+- Call set_model(model_name, model_obj, **params) when a model is loaded.
+- Retrieve with get_model(), get_model_name(), or get_info() anywhere.
+"""
+
+from __future__ import annotations
+from dataclasses import dataclass, asdict
+from typing import Any, Dict, Optional
+import threading
+
+@dataclass
+class SpyData:
+    model_name: str
+    model: Any
+    params: Dict[str, Any]
+
+    def to_dict(self) -> Dict[str, Any]:
+        d = asdict(self)
+        # Avoid serializing the raw model object
+        d["model"] = repr(self.model)
+        return d
+
+_lock = threading.RLock()
+_current: Optional[SpyData] = None
+
+def set_model(model_name: str, model: Any, **params: Any) -> None:
+    """Announce the current model and its load parameters."""
+    global _current
+    with _lock:
+        _current = SpyData(model_name=model_name, model=model, params=dict(params or {}))
+
+def get_model() -> Optional[Any]:
+    """Return the current model object, if any."""
+    with _lock:
+        return _current.model if _current else None
+
+def get_model_name() -> Optional[str]:
+    """Return the current model name, if any."""
+    with _lock:
+        return _current.model_name if _current else None
+
+def get_info() -> Optional[SpyData]:
+    """Return the full SpyData object, if any."""
+    with _lock:
+        return _current