Function main

tests/test_memory_system.py:47–74

()

    return args

async def main():
    args = parse_args()
    
    config.init_config(args.config, args)
    logger.init_logger(config)
    logger.info(f"| Config: {config.pretty_text}")
    
    await model_manager.init_models(use_local_proxy=config.use_local_proxy)
    logger.info(f"| Models: {model_manager.list_models()}")
    
    memory_manager = MemoryManager(**config.memory)
    logger.info(f"| Memory Manager: {memory_manager}")
    
    session_id = "test_session"
    await memory_manager.start_session(session_id)
    
    for i in range(10):
        await memory_manager.add_event(step_number=i,
                                       event_type=EventType.ACTION_STEP,
                                       data = dict(
                                           test_data = f"test_data_{i}"
                                       ),
                                       agent_name="test_agent",
                                       task_id="test_task",
                                       session_id=session_id)
        
    state = await memory_manager.get_state(session_id, n=5)
    logger.info(f"| State: {state}")
    
if __name__ == "__main__":
    asyncio.run(main())