first commit

evaluator/config/agents.py

@@ -0,0 +1,36 @@
+from __future__ import annotations
+from dataclasses import dataclass, field
+import yaml
+from pathlib import Path
+from evaluator.config.settings import settings
+
+@dataclass
+class AgentConfig:
+    agent_id: str
+    enabled: bool = True
+    days_back: int = 1
+    percentage: float = 1.0
+    langfuse_agent_aliases: list[str] = field(default_factory=list)
+    gcs_prefix: str = ""
+
+    @property
+    def aliases(self) -> set[str]:
+        return {self.agent_id, *self.langfuse_agent_aliases}
+
+
+def load_agents(path: str | None = None) -> list[AgentConfig]:
+    p = settings.path(path or settings.agents_config_path)
+    data = yaml.safe_load(p.read_text()) or {}
+    agents = []
+    for item in data.get("agents", []):
+        cfg = AgentConfig(
+            agent_id=item["agent_id"],
+            enabled=bool(item.get("enabled", True)),
+            days_back=int(item.get("days_back", item.get("daysBack", 1))),
+            percentage=float(item.get("percentage", 1.0)),
+            langfuse_agent_aliases=list(item.get("langfuse_agent_aliases", [])),
+            gcs_prefix=str(item.get("gcs_prefix", "")),
+        )
+        if cfg.enabled:
+            agents.append(cfg)
+    return agents