mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-09-05 08:06:15 +00:00
a4efc0f776
* fix(bm25_index): 修正BM25索引计算中的文档长度归一化问题 修复了在计算BM25相似度时对文档长度进行不正确归一化的bug，确保所有查询都能得到准确的相关性评分。 * up * up * up * up * up * up * up * up * up * up * up * up * up * up * up * up * up * refactor(steps): Rename and adjust indexing step logic - Rename `scan_changes.py` and `reindex.py` to `clear_and_scan.py` - Update implementation details of `ScanChangesStep` and `ClearAndScanStep` - Modify the scheduling mechanism in `WatchChangesStep` - Adjust step registration and parameter configuration in config files - Update related tests to align with the new interface changes * up * feat(daily): replace daily CRUD operations with slug provisioning approach * refactor(tests): migrate CRUD step tests from HTTP server to direct LocalFileStore * up * up * up * up --------- Co-authored-by: huangsen <huangsen.huang@alibaba-inc.com>
84 lines
3.2 KiB
Python
84 lines
3.2 KiB
Python
"""Global registry mapping ``(ComponentEnum, name) -> component class``."""
|
|
|
|
from typing import Callable, TypeVar, cast
|
|
|
|
from .base_component import BaseComponent
|
|
from ..enumeration import ComponentEnum
|
|
from ..utils import get_logger
|
|
|
|
T = TypeVar("T", bound=BaseComponent)
|
|
|
|
|
|
class ComponentRegistry:
|
|
"""Two-level registry: ``component_type -> name -> class``.
|
|
|
|
Supports both direct calls — ``R.register(MyClass, "name")`` — and
|
|
decorator usage — ``@R.register("name")``.
|
|
"""
|
|
|
|
def __init__(self) -> None:
|
|
self._registry: dict[ComponentEnum, dict[str, type[BaseComponent]]] = {}
|
|
self.logger = get_logger()
|
|
|
|
def _do_register(self, cls: type[T], name: str) -> type[T]:
|
|
"""Insert `cls` under its ``component_type`` group; warn on overwrite."""
|
|
component_type = getattr(cls, "component_type", None)
|
|
if not isinstance(component_type, ComponentEnum):
|
|
raise TypeError(
|
|
f"{cls.__name__} must have a ComponentEnum 'component_type' attribute",
|
|
)
|
|
if not name:
|
|
raise ValueError("Component name cannot be empty")
|
|
|
|
group = self._registry.setdefault(component_type, {})
|
|
if name in group:
|
|
self.logger.warning(
|
|
f"Component '{name}' already registered for {component_type}, overwriting",
|
|
)
|
|
group[name] = cls
|
|
return cls
|
|
|
|
def register(
|
|
self,
|
|
cls_or_name: type[T] | str,
|
|
name: str | None = None,
|
|
) -> Callable[[type[T]], type[T]] | type[T]:
|
|
"""Register a component class directly, or return a decorator that does so."""
|
|
# Direct call: register(MyClass) or register(MyClass, "alias").
|
|
if isinstance(cls_or_name, type):
|
|
cls = cast(type[T], cls_or_name)
|
|
return self._do_register(cls, name if name is not None else cls.__name__)
|
|
|
|
# Decorator call: @R.register("alias") — must receive a string name.
|
|
if not isinstance(cls_or_name, str):
|
|
raise TypeError(f"Expected a class or string, got {type(cls_or_name).__name__}")
|
|
|
|
registration_name = cls_or_name
|
|
|
|
def decorator(decorated_cls: type[T]) -> type[T]:
|
|
return self._do_register(decorated_cls, registration_name)
|
|
|
|
return decorator
|
|
|
|
def get(self, component_type: ComponentEnum, name: str) -> type[BaseComponent] | None:
|
|
"""Look up a registered class; return None if not found."""
|
|
return self._registry.get(component_type, {}).get(name)
|
|
|
|
def get_all(self, component_type: ComponentEnum) -> dict[str, type[BaseComponent]]:
|
|
"""Return a shallow copy of all classes registered under `component_type`."""
|
|
return dict(self._registry.get(component_type, {}))
|
|
|
|
def unregister(self, component_type: ComponentEnum, name: str) -> bool:
|
|
"""Remove an entry; return True if it existed, False otherwise."""
|
|
if (group := self._registry.get(component_type)) and name in group:
|
|
del group[name]
|
|
return True
|
|
return False
|
|
|
|
def clear(self) -> None:
|
|
"""Drop every registered entry."""
|
|
self._registry.clear()
|
|
|
|
|
|
# Process-wide singleton used throughout the codebase.
|
|
R = ComponentRegistry()
|