"""
Command Line Interface for AGI Memory System
"""
import argparse
import sys
from .core import AGIMemorySystem, AGIMemoryConfig

def main():
    parser = argparse.ArgumentParser(description="AGI Memory System CLI")
    parser.add_argument("--model", type=str, default="Qwen/Qwen3-0.6B", help="Model name")
    parser.add_argument("--device", type=str, default="cuda", help="Device (cuda/cpu)")
    parser.add_argument("--capacity", type=int, default=10000, help="Memory capacity")
    
    args = parser.parse_args()
    
    print("Initializing AGI Memory System...")
    config = AGIMemoryConfig(
        shared_model_name=args.model,
        device=args.device,
        memory_graph_capacity=args.capacity
    )
    
    memory = AGIMemorySystem(config)
    print("AGI Memory System initialized successfully!")
    print(f"Memory stats: {memory.get_stats()}")
    
    return 0

if __name__ == "__main__":
    sys.exit(main())