mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-09-17 23:51:19 +00:00
52f1a33b3a
Some checks are pending
Pre-commit / run (ubuntu-latest) (push) Waiting to run
- Add file_parser component with default implementation - Introduce SearchFilter schema for path and tag filtering - Implement filter functionality in BaseFileStore and LocalFileStore - Update file watcher to use parser-based filtering instead of suffix filters - Register new FILE_PARSER component enum - Add test_data directory to gitignore refactor: improve component imports and initialization - Fix relative imports in application.py - Add file_parser import to component init - Initialize registry dict when component type doesn't exist - Remove circular import in HttpService by using string annotation - Update config yaml to use proper component names refactor: enhance file watcher architecture - Replace MdFileWatcher with more flexible FullFileWatcher and LightFileWatcher - Remove suffix-based filtering in favor of parser-based approach - Update BaseFileWatcher to resolve parsers from app context - Remove unused watch_filter method refactor: update ReMe core functionality - Remove memory_path creation - Simplify dream and proactive methods to return empty strings - Update config defaults for HTTP service and component backends docs: update component configuration in paw.yaml - Change service backend from cmd to http - Rename components to use correct singular forms - Add default file parser and file watcher configurations - Set up local file store with default settings ``` Co-authored-by: huangsen <huangsen.huang@alibaba-inc.com>
80 lines
2.9 KiB
Python
80 lines
2.9 KiB
Python
"""
|
|
Component registry module.
|
|
|
|
Provides a global registry for managing component class registration and lookup.
|
|
Supports two registration methods:
|
|
1. Direct registration: R.register(MyClass, "name")
|
|
2. Decorator registration: @R.register("name")
|
|
"""
|
|
|
|
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:
|
|
"""Registry for managing component class registration and lookup."""
|
|
|
|
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]:
|
|
"""Register a component class with the given name."""
|
|
if not hasattr(cls, "component_type"):
|
|
raise TypeError(f"{cls.__name__} must have 'component_type' attribute")
|
|
if not name:
|
|
raise ValueError("Component name cannot be empty")
|
|
|
|
component_type = cls.component_type
|
|
if component_type not in self._registry:
|
|
self._registry[component_type] = {}
|
|
if name in self._registry[component_type]:
|
|
self.logger.warning(f"Component '{name}' already registered for {component_type}, overwriting")
|
|
|
|
self._registry[component_type][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. Supports direct and decorator modes."""
|
|
# Direct registration: R.register(MyClass, "name")
|
|
if isinstance(cls_or_name, type):
|
|
return self._do_register(cast(type[T], cls_or_name), name or cls_or_name.__name__)
|
|
|
|
# Decorator mode: @R.register("name")
|
|
decorator_name = cls_or_name
|
|
|
|
def decorator(decorated_cls: type[T]) -> type[T]:
|
|
return self._do_register(decorated_cls, decorator_name or decorated_cls.__name__)
|
|
|
|
return decorator
|
|
|
|
def get(self, component_type: ComponentEnum, name: str) -> type[BaseComponent] | None:
|
|
"""Get a registered component class by type and name."""
|
|
return self._registry.get(component_type, {}).get(name)
|
|
|
|
def get_all(self, component_type: ComponentEnum) -> dict[str, type[BaseComponent]]:
|
|
"""Get all registered components of a given type."""
|
|
return dict(self._registry.get(component_type, {}))
|
|
|
|
def unregister(self, component_type: ComponentEnum, name: str) -> bool:
|
|
"""Remove a component from the registry. Returns True if found."""
|
|
if name in self._registry.get(component_type, {}):
|
|
del self._registry[component_type][name]
|
|
return True
|
|
return False
|
|
|
|
def clear(self) -> None:
|
|
"""Clear all registered components."""
|
|
self._registry.clear()
|
|
|
|
|
|
R = ComponentRegistry()
|