mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-05-29 19:35:20 +02:00
Cut the docstrings and Field(description=...) text across the entire automations/ tree down to single-line intent statements, matching the multi_agent_chat conciseness style: - Module docstrings: one line stating what the file is. - Class docstrings: deleted when the class name + module docstring already cover intent; kept only where they add a constraint or rationale not visible in the signature. - Pydantic Field descriptions: short noun phrases / clauses, not full sentences. Reasoning that belonged in the design plan moved out of the code. - Enum values: per-value docstrings replaced with terse inline comments where the meaning isn't obvious from the name. Behaviour is unchanged. The same 33 files, same public surface, same imports — verified by re-running the 10-point registry smoke test and the 8-point schema round-trip / constraint suite from commits 9 and 10. LOC: 1180 → 691 (-42%).
23 lines
676 B
Python
23 lines
676 B
Python
"""In-memory capability registry. Populated once at process startup."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from .types import Capability
|
|
|
|
_REGISTRY: dict[str, Capability] = {}
|
|
|
|
|
|
def register_capability(capability: Capability) -> None:
|
|
"""Register a capability. Raises on duplicate id."""
|
|
if capability.id in _REGISTRY:
|
|
raise ValueError(f"Capability already registered: {capability.id!r}")
|
|
_REGISTRY[capability.id] = capability
|
|
|
|
|
|
def get_capability(capability_id: str) -> Capability | None:
|
|
return _REGISTRY.get(capability_id)
|
|
|
|
|
|
def all_capabilities() -> dict[str, Capability]:
|
|
"""Defensive snapshot of the registry."""
|
|
return dict(_REGISTRY)
|