generic-ml-cache-cli 0.2.0__tar.gz → 0.3.0__tar.gz

{generic_ml_cache_cli-0.2.0 → generic_ml_cache_cli-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: generic-ml-cache-cli
-Version: 0.2.0
+Version: 0.3.0
 Summary: Terminal UI for generic-ml-cache: the gmlcache command. A thin inbound driver over generic-ml-cache-core -- reads config, provides the data source, maps commands onto the core library.
 Project-URL: Homepage, https://github.com/danielslobozian/generic-ml-cache
 Project-URL: Repository, https://github.com/danielslobozian/generic-ml-cache
@@ -24,7 +24,7 @@ Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Utilities
 Requires-Python: >=3.9
 Requires-Dist: argcomplete<4,>=3
-Requires-Dist: generic-ml-cache-core>=0.2.0
+Requires-Dist: generic-ml-cache-core>=0.3.0
 Provides-Extra: dev
 Requires-Dist: coverage>=7; extra == 'dev'
 Requires-Dist: pytest-cov; extra == 'dev'

{generic_ml_cache_cli-0.2.0 → generic_ml_cache_cli-0.3.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "generic-ml-cache-cli"
-version = "0.2.0"
+version = "0.3.0"
 description = "Terminal UI for generic-ml-cache: the gmlcache command. A thin inbound driver over generic-ml-cache-core -- reads config, provides the data source, maps commands onto the core library."
 readme = "README.md"
 requires-python = ">=3.9"
@@ -25,7 +25,7 @@ classifiers = [
   "Programming Language :: Python :: 3.13",
   "Topic :: Utilities",
 ]
-dependencies = ["generic-ml-cache-core>=0.2.0", "argcomplete>=3,<4"]
+dependencies = ["generic-ml-cache-core>=0.3.0", "argcomplete>=3,<4"]
 
 [project.urls]
 Homepage = "https://github.com/danielslobozian/generic-ml-cache"

{generic_ml_cache_cli-0.2.0 → generic_ml_cache_cli-0.3.0}/src/generic_ml_cache_cli/cli.py

@@ -172,6 +172,7 @@ def _cmd_run(args: argparse.Namespace) -> int:
         grants=list(getattr(args, "grant", None) or []),
         cache_mode=cache_mode,
         record_on_error=args.record_on_error,
+        tags=list(getattr(args, "tag", None) or []),
     )
 
     def executable_override(client: str):
@@ -550,11 +551,15 @@ def _cmd_list(args: argparse.Namespace) -> int:
             "kind": summary.kind,
             "key": summary.execution_key,
             "hits": hit_counts.get(summary.execution_key, 0),
+            "tags": wired.repository.tags_for(summary.execution_key),
         }
         for summary in wired.repository.current_execution_summaries()
         if (not args.client or summary.client == args.client)
         and (not args.model or summary.model == args.model)
     ]
+    wanted_tags = set(getattr(args, "tag", None) or [])
+    if wanted_tags:
+        entries = [entry for entry in entries if wanted_tags & set(entry["tags"])]
 
     if args.json:
         print(json.dumps({"executions": entries}, indent=2))
@@ -568,10 +573,13 @@ def _cmd_list(args: argparse.Namespace) -> int:
     for entry in sorted(entries, key=lambda item: (item["client"], item["model"], item["key"])):
         hits = entry["hits"]
         hits_text = _paint(str(hits), _GREEN) if hits else _paint(str(hits), _GREY)
-        print(
+        line = (
             f"  {entry['client']:<8} {entry['model']:<20} {entry['kind']:<18} "
             f"{_paint(entry['key'][:12], _GREY)}  hits:{hits_text}"
         )
+        if entry["tags"]:
+            line += "  tags:" + _paint(",".join(entry["tags"]), _TEAL)
+        print(line)
     return 0
 
 
@@ -709,6 +717,16 @@ def build_parser() -> argparse.ArgumentParser:
         choices=GRANT_CHOICES,
         help=_GRANT_HELP,
     )
+    run.add_argument(
+        "--tag",
+        action="append",
+        dest="tag",
+        metavar="TAG",
+        help=(
+            "label this execution with a tag for later grouping/queries (repeatable; "
+            "metadata only -- never part of the cache key). A relabel on a hit accumulates."
+        ),
+    )
     run.add_argument(
         "--json",
         action="store_true",
@@ -840,6 +858,13 @@ def build_parser() -> argparse.ArgumentParser:
     )
     listp.add_argument("--client", help="only executions recorded for this client")
     listp.add_argument("--model", help="only executions recorded for this model")
+    listp.add_argument(
+        "--tag",
+        action="append",
+        dest="tag",
+        metavar="TAG",
+        help="only executions carrying any of these tags (repeatable; match-any)",
+    )
     listp.add_argument("--json", action="store_true", help="emit machine-readable JSON")
     listp.set_defaults(func=_cmd_list)
 

{generic_ml_cache_cli-0.2.0 → generic_ml_cache_cli-0.3.0}/tests/test_cli.py

@@ -324,3 +324,48 @@ def test_check_reports_hit_after_a_run(tmp_path, monkeypatch, capsys):
     out = capsys.readouterr().out
     assert rc == 0
     assert "status  : hit" in out
+
+
+def test_run_tag_stores_tags_through_the_cli(tmp_path):
+    import glob
+    import sqlite3
+
+    rc = run_cli(
+        [
+            "run",
+            "--client",
+            "fake",
+            "--model",
+            "m1",
+            "--effort",
+            "high",
+            "--prompt",
+            "STDOUT hi",
+            "--tag",
+            "ticket",
+            "--tag",
+            "id-scan",
+        ]
+    )
+    assert rc == 0
+    stores = glob.glob(str(tmp_path / "**" / "executions.sqlite3"), recursive=True)
+    assert stores, "no executions store was written"
+    connection = sqlite3.connect(stores[0])
+    stored = sorted(tag for (tag,) in connection.execute("SELECT tag FROM execution_tags"))
+    connection.close()
+    assert stored == ["id-scan", "ticket"]
+
+
+def test_list_filters_by_tag_and_shows_tags(capsys):
+    import json
+
+    base = ["run", "--client", "fake", "--model", "m1", "--effort", "high"]
+    run_cli(base + ["--prompt", "STDOUT a", "--tag", "alpha"])
+    run_cli(base + ["--prompt", "STDOUT b", "--tag", "beta"])
+    capsys.readouterr()
+
+    rc = main(["list", "--tag", "alpha", "--json"])
+    assert rc == 0
+    listed = json.loads(capsys.readouterr().out)["executions"]
+    assert len(listed) == 1  # match-any filter keeps only the alpha-tagged entry
+    assert listed[0]["tags"] == ["alpha"]