okf-generator 0.1.1__py3-none-any.whl

okf/__init__.py

@@ -0,0 +1,34 @@
+"""OKF — Open Knowledge Format toolkit for codebases.
+
+Generate OKF v0.1 knowledge bundles from source code, look up concepts
+for AI agent context injection, and convert bundles into LLM training pairs.
+
+Quick start:
+    from okf.generator import scan_codebase, write_bundle
+    from okf.lookup import load_bundle, search
+    from okf.pairs import process_concept
+
+CLI:
+    okf generate ./my_codebase ./okf_bundle
+    okf lookup WorldBankConnector
+    okf pairs ./okf_bundle ./train.jsonl
+    okf summarize ./okf_bundle
+"""
+
+__version__ = "0.1.0"
+__author__  = "Umair Baig"
+__license__ = "MIT"
+
+# Public API surface
+from okf.generator import scan_codebase, write_bundle, write_summary, Concept
+from okf.lookup    import load_bundle, search
+
+__all__ = [
+    "scan_codebase",
+    "write_bundle",
+    "write_summary",
+    "load_bundle",
+    "search",
+    "Concept",
+    "__version__",
+]

okf/cli.py

@@ -0,0 +1,58 @@
+"""okf — unified CLI for the OKF toolkit.
+
+Commands:
+  okf generate   <source_dir> [output_dir]   Generate OKF bundle from codebase
+  okf lookup     <query> [options]            Look up a concept in a bundle
+  okf pairs      <bundle_dir> [output_file]  Convert bundle to training pairs
+  okf summarize  <bundle_dir>                Regenerate SUMMARY.md only
+
+Run `okf <command> --help` for per-command options.
+"""
+
+import sys
+
+
+def main():
+    if len(sys.argv) < 2 or sys.argv[1] in {"-h", "--help"}:
+        print(__doc__)
+        print("Usage: okf <command> [args]")
+        print("")
+        print("Commands:")
+        print("  generate   Generate OKF bundle from a codebase")
+        print("  lookup     Look up concepts in a bundle")
+        print("  pairs      Convert bundle to JSONL training pairs")
+        print("  summarize  Regenerate SUMMARY.md from existing bundle")
+        sys.exit(0)
+
+    cmd, rest = sys.argv[1], sys.argv[2:]
+    sys.argv = [f"okf {cmd}"] + rest   # rewrite argv for sub-parsers
+
+    if cmd == "generate":
+        # Handle --summarize flag routed here too
+        if rest and rest[0] == "--summarize":
+            sys.argv = ["okf generate", "--summarize"] + rest[1:]
+        from okf.generator import main as _main
+        _main()
+
+    elif cmd == "summarize":
+        # Sugar: `okf summarize <bundle>` → generator --summarize <bundle>
+        sys.argv = ["okf generate", "--summarize"] + rest
+        from okf.generator import main as _main
+        _main()
+
+    elif cmd == "lookup":
+        from okf.lookup import main as _main
+        _main()
+
+    elif cmd == "pairs":
+        from okf.pairs import main as _main
+        _main()
+
+    else:
+        print(f"Unknown command: {cmd!r}", file=sys.stderr)
+        print("Run `okf --help` for available commands.", file=sys.stderr)
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()