diff --git a/concordia/typing/component_v2.py b/concordia/typing/component_v2.py index b98abff..7781ef8 100644 --- a/concordia/typing/component_v2.py +++ b/concordia/typing/component_v2.py @@ -21,6 +21,7 @@ from typing import Any from concordia.typing import entity as entity_lib +import overrides ComponentName = str ComponentContext = str @@ -75,7 +76,7 @@ def get_component(self, component_name: str) -> "BaseComponent": raise NotImplementedError() -class BaseComponent: +class BaseComponent(overrides.EnforceOverrides): """A base class for components.""" def __init__(self): diff --git a/concordia/typing/memory.py b/concordia/typing/memory.py index f5f5bbf..81de365 100644 --- a/concordia/typing/memory.py +++ b/concordia/typing/memory.py @@ -17,6 +17,7 @@ import abc from collections.abc import Mapping, Sequence from typing import Any, Protocol +import overrides class MemoryScorer(Protocol): @@ -32,7 +33,7 @@ def __call__(self, query: str, text: str, **metadata: Any) -> float: """ -class MemoryBank(metaclass=abc.ABCMeta): +class MemoryBank(overrides.EnforceOverrides, metaclass=abc.ABCMeta): """Base class for memory banks.""" @abc.abstractmethod