@@ -57,34 +57,37 @@ def load_config(
5757 raw_config = yaml .safe_load (f )
5858
5959 base_system_prompt = ""
60- if "system_prompt" in raw_config :
60+
61+ if raw_config .get ("system_prompt" ):
6162 base_system_prompt = load_prompt (raw_config ["system_prompt" ])
6263
63- if "agents" in raw_config :
64+ if raw_config . get ( "agents" ) :
6465 for agent_config in raw_config ["agents" ].values ():
65- if "prompt" in agent_config :
66+ if agent_config . get ( "prompt" ) :
6667 agent_config ["prompt" ] = load_prompt (agent_config ["prompt" ])
6768
6869 mcp_server_prompts = []
69- if "mcp_servers" in raw_config :
70+
71+ if raw_config .get ("mcp_servers" ):
7072 enabled_servers = {
7173 name : config
7274 for name , config in raw_config ["mcp_servers" ].items ()
7375 if config .get ("enabled" , True )
7476 }
77+
7578 raw_config ["mcp_servers" ] = enabled_servers
7679
7780 for server_config in enabled_servers .values ():
78- if "prompt" in server_config and server_config [ "prompt" ] :
81+ if server_config . get ( "prompt" ) :
7982 loaded_prompt = load_prompt (server_config ["prompt" ])
8083 server_config ["prompt" ] = loaded_prompt
8184 mcp_server_prompts .append (loaded_prompt )
8285
83- if "env" in server_config :
86+ if server_config . get ( "env" ) :
8487 for key , value in server_config ["env" ].items ():
8588 server_config ["env" ][key ] = os .path .expandvars (value )
8689
87- if "args" in server_config :
90+ if server_config . get ( "args" ) :
8891 server_config ["args" ] = [
8992 os .path .expandvars (arg ) for arg in server_config ["args" ]
9093 ]
0 commit comments