@@ -22,7 +22,6 @@ def __init__(self):
2222 """Initialize empty registry."""
2323 self ._agents : dict [str , type [BaseGalaxyAgent ]] = {}
2424 self ._agent_metadata : dict [str , dict ] = {}
25- self ._disabled : set [str ] = set ()
2625
2726 def register (
2827 self ,
@@ -68,8 +67,6 @@ def get_agent(self, agent_type: str, deps: GalaxyAgentDependencies) -> BaseGalax
6867 Raises:
6968 ValueError: If agent type is not registered
7069 """
71- if agent_type in self ._disabled :
72- raise ValueError (f"Agent '{ agent_type } ' is disabled in configuration" )
7370 if agent_type not in self ._agents :
7471 available = list (self ._agents .keys ())
7572 raise ValueError (f"Unknown agent type: { agent_type } . Available: { available } " )
@@ -149,11 +146,10 @@ def _is_enabled(agent_type: str) -> bool:
149146 return agent_cfg .get ("enabled" , True )
150147 return True
151148
152- def _register_or_disable (registry : AgentRegistry , agent_type : str , agent_class : type [BaseGalaxyAgent ]):
149+ def _register_if_enabled (registry : AgentRegistry , agent_type : str , agent_class : type [BaseGalaxyAgent ]):
153150 if _is_enabled (agent_type ):
154151 registry .register (agent_type , agent_class )
155152 else :
156- registry ._disabled .add (agent_type )
157153 log .info (f"Agent '{ agent_type } ' disabled by configuration, skipping registration" )
158154
159155 registry = AgentRegistry ()
@@ -163,8 +159,8 @@ def _register_or_disable(registry: AgentRegistry, agent_type: str, agent_class:
163159 log .warning ("Router agent cannot be disabled — ignoring enabled: false" )
164160 registry .register (AgentType .ROUTER , QueryRouterAgent )
165161
166- _register_or_disable (registry , AgentType .ERROR_ANALYSIS , ErrorAnalysisAgent )
167- _register_or_disable (registry , AgentType .CUSTOM_TOOL , CustomToolAgent )
168- _register_or_disable (registry , AgentType .ORCHESTRATOR , WorkflowOrchestratorAgent )
169- _register_or_disable (registry , AgentType .TOOL_RECOMMENDATION , ToolRecommendationAgent )
162+ _register_if_enabled (registry , AgentType .ERROR_ANALYSIS , ErrorAnalysisAgent )
163+ _register_if_enabled (registry , AgentType .CUSTOM_TOOL , CustomToolAgent )
164+ _register_if_enabled (registry , AgentType .ORCHESTRATOR , WorkflowOrchestratorAgent )
165+ _register_if_enabled (registry , AgentType .TOOL_RECOMMENDATION , ToolRecommendationAgent )
170166 return registry
0 commit comments