diff --git a/src/cleveragents/application/services/acms_service.py b/src/cleveragents/application/services/acms_service.py index a0efd4c03..54bbfbaf7 100644 --- a/src/cleveragents/application/services/acms_service.py +++ b/src/cleveragents/application/services/acms_service.py @@ -915,7 +915,10 @@ class ACMSPipeline: for name in registry.list_all(): if name not in self._strategies: strat_inst = registry.get(name) - self.register_strategy(name=name, strategy=strat_inst) + self.register_strategy( + name=name, + strategy=cast(ContextStrategy, strat_inst), + ) # Sync the enabled list. enabled = registry.list_enabled() if enabled: diff --git a/src/cleveragents/application/services/context_strategies.py b/src/cleveragents/application/services/context_strategies.py index 07187b84f..bc98418a5 100644 --- a/src/cleveragents/application/services/context_strategies.py +++ b/src/cleveragents/application/services/context_strategies.py @@ -53,9 +53,6 @@ from cleveragents.application.services.strategy_registry import ( from cleveragents.domain.models.acms.strategy import ( ContextStrategy, # noqa: F401 ) -from cleveragents.domain.models.acms.strategy import ( - StrategyCapabilities as _DomainStrategyCapabilities, # noqa: F401 -) from cleveragents.domain.models.core.context_fragment import ( ContextBudget, ContextFragment,