src/tools/interfaces.py

# src/tools/interfaces.py
from abc import ABC, abstractmethod
from typing import List, Dict, Any, Optional
from domain.training.charts import ChartSpec
import matplotlib.figure


class IFeatureStore(ABC):
    @abstractmethod
    def put_features(self, run_id: str, features: Dict[str, Any]):
        pass

    @abstractmethod
    def get_features(self, run_id: str) -> Optional[Dict[str, Any]]:
        pass

    @abstractmethod
    def put_weekly_summary(self, week_key: str, summary: Dict[str, Any]):
        pass

    @abstractmethod
    def get_weekly_summary(self, week_key: str) -> Optional[Dict[str, Any]]:
        pass

    @abstractmethod
    def get_all_features(self) -> List[Dict[str, Any]]:
        pass

    @abstractmethod
    def clear(self):
        pass


class IPlanStore(ABC):
    @abstractmethod
    def save_plan(self, plan_id: str, plan_obj: str):
        pass

    @abstractmethod
    def get_plan(self, plan_id: str) -> Optional[str]:
        pass

    @abstractmethod
    def get_latest_plan(self) -> Optional[str]:
        pass

    @abstractmethod
    def clear(self):
        pass


class IVisualizationExecutor(ABC):
    @abstractmethod
    def render_chart(
        self, chart_spec: ChartSpec, features: List[Dict[str, Any]]
    ) -> matplotlib.figure.Figure:
        pass