feat(config): add ParallelAgent and SequentialAgent to the loader

PiperOrigin-RevId: 781712004
This commit is contained in:
Liang Wu
2025-07-10 15:42:06 -07:00
committed by Copybara-Service
parent a313c2c1af
commit 68f34135fe
@@ -26,6 +26,10 @@ from .llm_agent import LlmAgent
from .llm_agent import LlmAgentConfig
from .loop_agent import LoopAgent
from .loop_agent import LoopAgentConfig
from .parallel_agent import ParallelAgent
from .parallel_agent import ParallelAgentConfig
from .sequential_agent import SequentialAgent
from .sequential_agent import SequentialAgentConfig
@working_in_progress("from_config is not ready for use.")
@@ -50,6 +54,10 @@ def from_config(config_path: str) -> BaseAgent:
return LlmAgent.from_config(config.root)
elif isinstance(config.root, LoopAgentConfig):
return LoopAgent.from_config(config.root)
elif isinstance(config.root, ParallelAgentConfig):
return ParallelAgent.from_config(config.root)
elif isinstance(config.root, SequentialAgentConfig):
return SequentialAgent.from_config(config.root)
else:
raise ValueError("Unsupported config type")