File size: 965 Bytes
101858b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 | """
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())
|