memory_cli.py

#!/usr/bin/env python3
"""
QAI Memory CLI
Author: William McDonald | UNC Chapel Hill | Founder, QAI Lab

Usage:
  memory read "query"
  memory write session "content"
  memory write knowledge "topic" "content"
  memory log "entry"
"""
import argparse
import sys
from memory_engine.memory_manager import MemoryManager


def main() -> None:
    parser = argparse.ArgumentParser(
        prog="memory",
        description="QAI Persistent Memory Layer — CLI",
    )
    sub = parser.add_subparsers(dest="command", required=True)

    # memory read "query"
    r = sub.add_parser("read", help="Search memory for a query")
    r.add_argument("query", help="Search term")

    # memory write session|knowledge ...
    w = sub.add_parser("write", help="Write to session or knowledge memory")
    w.add_argument("type", choices=["session", "knowledge"], help="Memory type")
    w.add_argument("args", nargs="+", help="For session: <content>. For knowledge: <topic> <content>")

    # memory log "entry"
    lg = sub.add_parser("log", help="Append an entry to today's log")
    lg.add_argument("entry", help="Log entry text")

    args = parser.parse_args()
    mm = MemoryManager()

    if args.command == "read":
        print(mm.read_memory(args.query))

    elif args.command == "write":
        if args.type == "session":
            content = " ".join(args.args)
            mm.write_session({"current_state": content, "objective": "", "key_decisions": "", "next_steps": ""})
            print(f"Session memory written for session '{mm.session_id}'.")
        elif args.type == "knowledge":
            if len(args.args) < 2:
                print("Usage: memory write knowledge <topic> <content>", file=sys.stderr)
                sys.exit(1)
            topic, content = args.args[0], " ".join(args.args[1:])
            mm.update_knowledge(topic, content)
            print(f"Knowledge memory updated for topic '{topic}'.")

    elif args.command == "log":
        mm.append_log(args.entry)
        print("Log entry appended.")


if __name__ == "__main__":
    main()