File size: 915 Bytes
ddbc0c8 | 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 33 34 35 36 37 38 39 | from __future__ import annotations
import argparse
from pathlib import Path
from legalrag.config import AppConfig
from legalrag.index.registry import IndexRegistry
def parse_args() -> argparse.Namespace:
p = argparse.ArgumentParser(description="Index registry admin")
sub = p.add_subparsers(dest="cmd", required=True)
sub.add_parser("list", help="List index versions")
act = sub.add_parser("activate", help="Activate an index version")
act.add_argument("version", type=str)
return p.parse_args()
def main() -> None:
args = parse_args()
cfg = AppConfig.load()
registry = IndexRegistry(Path(cfg.paths.index_dir))
if args.cmd == "list":
for v in registry.list_versions():
print(v)
return
if args.cmd == "activate":
registry.activate(args.version)
print(f"active={args.version}")
if __name__ == "__main__":
main()
|