small fix in agent_system class

Author: TataKKKL

litemultiagent/core/agent_system.py

@@ -25,7 +25,11 @@ def __init__(self, main_agent_config, system_config: Dict[str, Any]) -> None:
 
         # TODO: add shared_memory
         # TODO: add shared_context
+        # set up agent hierarchy
+        agent_manager = AgentManager()
+        self.main_agent = agent_manager.get_agent(main_agent_config)
 
+        # set up agent system shared config
         # create shared_config for agents to share
         self.shared_config = {
             "task_id": 0,
@@ -36,10 +40,6 @@ def __init__(self, main_agent_config, system_config: Dict[str, Any]) -> None:
             "model_name": system_config["model_name"],
             "tool_choice": system_config["tool_choice"]
         }
-
-
-        agent_manager = AgentManager()
-        self.main_agent = agent_manager.get_agent(main_agent_config)
         self.main_agent.set_shared_config(self.shared_config)
 
     def execute(self, task: str):