ayjays132's picture
Upload 478 files
101858b verified
"""
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())