()
| 41 | return args |
| 42 | |
| 43 | async def main(): |
| 44 | args = parse_args() |
| 45 | |
| 46 | config.initialize(config_path = args.config, args = args) |
| 47 | logger.initialize(config = config) |
| 48 | logger.info(f"| Config: {config.pretty_text}") |
| 49 | |
| 50 | # Initialize model manager |
| 51 | logger.info("| 🧠 Initializing model manager...") |
| 52 | await model_manager.initialize() |
| 53 | logger.info(f"| ✅ Model manager initialized: {await model_manager.list()}") |
| 54 | |
| 55 | # Initialize prompt manager |
| 56 | logger.info("| 📁 Initializing prompt manager...") |
| 57 | await prompt_manager.initialize() |
| 58 | logger.info(f"| ✅ Prompt manager initialized: {await prompt_manager.list()}") |
| 59 | |
| 60 | # Initialize memory manager |
| 61 | logger.info("| 📁 Initializing memory manager...") |
| 62 | await memory_manager.initialize(memory_names=config.memory_names) |
| 63 | logger.info(f"| ✅ Memory manager initialized: {await memory_manager.list()}") |
| 64 | |
| 65 | # Initialize tools |
| 66 | logger.info("| 🛠️ Initializing tools...") |
| 67 | await tcp.initialize(tool_names=config.tool_names) |
| 68 | logger.info(f"| ✅ Tools initialized: {await tcp.list()}") |
| 69 | |
| 70 | # Initialize environments |
| 71 | logger.info("| 🎮 Initializing environments...") |
| 72 | await ecp.initialize(env_names=config.env_names) |
| 73 | logger.info(f"| ✅ Environments initialized: {await ecp.list()}") |
| 74 | |
| 75 | # Initialize agents |
| 76 | logger.info("| 🤖 Initializing agents...") |
| 77 | await acp.initialize(agent_names=config.agent_names) |
| 78 | logger.info(f"| ✅ Agents initialized: {await acp.list()}") |
| 79 | |
| 80 | # Initialize version manager, must after tool, agent, environment initialized |
| 81 | logger.info("| 📁 Initializing version manager...") |
| 82 | await version_manager.initialize() |
| 83 | logger.info(f"| ✅ Version manager initialized: {json.dumps(await version_manager.list(), indent=4)}") |
| 84 | |
| 85 | # # Step 1: Get variables from prompt manager |
| 86 | # variables = await prompt_manager.get_variables(prompt_name="tool_calling") |
| 87 | # print("="*100) |
| 88 | # print(variables) |
| 89 | # print(type(variables), len(variables)) |
| 90 | # sub_variables = variables[0].variables |
| 91 | # print(sub_variables) |
| 92 | # print(type(sub_variables), len(sub_variables)) |
| 93 | # print("="*100) |
| 94 | |
| 95 | # print("="*100) |
| 96 | # trainable_variables = await prompt_manager.get_trainable_variables(prompt_name="tool_calling") |
| 97 | # print(trainable_variables) |
| 98 | # print(type(trainable_variables), len(trainable_variables)) |
| 99 | |
| 100 | # sub_variables = trainable_variables[0].variables |
no test coverage detected