ayjays132's picture
Upload 478 files
101858b verified
Raw History Blame Contribute Delete
965 Bytes
"""
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())