From b1228edd9c112e5bed21bea6586d3f660528cc17 Mon Sep 17 00:00:00 2001
From: Samuel Sloniker <sam@kj7rrv.com>
Date: Sat, 26 Nov 2022 18:26:52 -0800
Subject: [PATCH] Add CLI for Model.get()

---
 README.md        | 7 +++++++
 gptc/__main__.py | 8 ++++++++
 2 files changed, 15 insertions(+)

diff --git a/README.md b/README.md
index a22febc..798e97b 100644
--- a/README.md
+++ b/README.md
@@ -25,6 +25,13 @@ Alternatively, if you only need the most likely category, you can use this:
 This will prompt for a string and classify it, outputting the category on
 stdout (or "None" if it cannot determine anything).
 
+### Checking individual words or ngrams
+
+    gptc check <compiled model file> <token or ngram>
+
+This is very similar to `gptc classify`, except it takes the input as an
+argument, and it treats the input as a single token or ngram.
+
 ### Compiling models
 
     gptc compile [-n <max_ngram_length>] [-c <min_count>] <raw model file>
diff --git a/gptc/__main__.py b/gptc/__main__.py
index 11cd795..32ec754 100644
--- a/gptc/__main__.py
+++ b/gptc/__main__.py
@@ -53,6 +53,10 @@ def main() -> None:
         action="store_true",
     )
 
+    check_parser = subparsers.add_parser("check", help="check one word or ngram in model")
+    check_parser.add_argument("model", help="compiled model to use")
+    check_parser.add_argument("token", help="token or ngram to check")
+
     pack_parser = subparsers.add_parser("pack", help="pack a model from a directory")
     pack_parser.add_argument("model", help="directory containing model")
 
@@ -81,6 +85,10 @@ def main() -> None:
         else:
             probabilities = model.confidence(text, args.max_ngram_length)
             print(json.dumps(probabilities))
+    elif args.subparser_name == "check":
+        with open(args.model, "rb") as f:
+            model = gptc.deserialize(f.read())
+        print(json.dumps(model.get(args.token)))
     else:
         print(json.dumps(gptc.pack(args.model, True)[0]))