From 77e1056b43273f33f5d272c1a074e05a3f04cdf0 Mon Sep 17 00:00:00 2001 From: rootflo-hardik Date: Tue, 11 Nov 2025 12:11:52 +0530 Subject: [PATCH] builder -> from_yaml: picking agent from agents if present instead of yaml --- flo_ai/flo_ai/arium/builder.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/flo_ai/flo_ai/arium/builder.py b/flo_ai/flo_ai/arium/builder.py index cc66bdb8..6071f7b6 100644 --- a/flo_ai/flo_ai/arium/builder.py +++ b/flo_ai/flo_ai/arium/builder.py @@ -457,6 +457,9 @@ def from_yaml( f'Either provide the agent in the agents parameter or add configuration fields.' ) + elif agents and agent_name in agents: + agent = agents[agent_name] + # Method 2: Direct agent definition elif ( 'job' in agent_config